o hp@s ddlZddlmZddlmZddlmZddlmZm Z m Z ddl m Z ddl mZddlmZdd lmZdd lmZdd lmZmZdd lmZdd lmZeZ d:ddZGdddZGdddeZ eej!e deej"e dGddde Z#eee#dddZ$GdddZ%Gddde%Z&Gddde%Z'Gd d!d!e%Z(Gd"d#d#e%Z)Gd$d%d%e)Z*Gd&d'd'e)Z+Gd(d)d)eZ,eee,dGd*d+d+eZ-Gd,d-d-eZ.Gd.d/d/e.Z/eee/dGd0d1d1eZ0Gd2d3d3Z1Gd4d5d5Z2Gd6d7d7Z3d8d9Z4dS);N)ImproperlyConfigured)models) force_str)escape format_htmlformat_html_join) mark_safe)capfirst) gettext_lazy)TaggableManager)blocks) RichTextField StreamField)ModelFieldRegistry)text_from_htmlFcCs<|durtd|r|tjkrtdtj||||ddS)zo Define parameters for form fields to be used by WagtailAdminModelForm for a given database field. NzNregister_comparison_class must be passed a 'comparison_class' keyword argumentzRThe 'to' argument on register_comparison_class is only valid for ForeignKey fields)tovalue exact_class)rr ForeignKeycomparison_class_registryregister) field_classrcomparison_classrrU/var/www/html/ndineBlogger/venv/lib/python3.10/site-packages/wagtail/admin/compare.pyregister_comparison_classs rc@s4eZdZdZdZddZddZddZd d Zd S) FieldComparisonTFcCs"||_|||_|||_dSN)fieldvalue_from_objectval_aval_b)selfrobj_aobj_brrr__init__/s zFieldComparison.__init__cCs.t|jdd}|dur|jjdd}t|SzL Returns a label for this field to be displayed to the user verbose_nameN_ )getattrrnamereplacer r"r'rrr field_label4szFieldComparison.field_labelcCs2|j|jkrtd|jfd|jfgSt|jS)Ndeletionaddition)r r!TextDiffto_htmlrr"rrrhtmldiff@s  zFieldComparison.htmldiffcC |j|jkS)z7 Returns True if the field has changed r r!r3rrr has_changedHs zFieldComparison.has_changedN) __name__ __module__ __qualname__is_fieldis_child_relationr%r.r4r7rrrrr+s rc@eZdZddZdS)TextFieldComparisoncCst|j|jSr) diff_textr r!r2r3rrrr4PszTextFieldComparison.htmldiffNr8r9r:r4rrrrr>O r>)rc@r=)RichTextFieldComparisoncCtt|jt|jSr)r?rr r!r2r3rrrr4Ys z RichTextFieldComparison.htmldiffNr@rrrrrBXrArBcCs|t|dr |St|tjtjfrtSt|tjrtSt|tjr$t St|tj r,t St|tj r4t St|tjr {}  
{}
{}
) rS child_blocksitemslabelrTappendr_rr)r"r^ htmlvaluesr+rSrjrrrrr_s  zStructBlockComparison.htmlvaluec Cspg}|jjD]&\}}|jj|j}t|}|||||j|j|j||j | fqt dt dd|S)Nrerfrg) rSrhrirjrTrkrVrWr r!r4rr)r" htmldiffsr+rSrjrrrrr4s(  zStructBlockComparison.htmldiffN)r8r9r:r_r4rrrrrNs rNc@s4eZdZeddZeddZddZddZd S) BaseSequenceBlockComparisoncCstr)NotImplementedError)r^rrrget_blocks_from_valuesz1BaseSequenceBlockComparison.get_blocks_from_valuecCs$dd|D}dd|D}||fS)NcSi|]}|j|qSrid.0rSrrr z@BaseSequenceBlockComparison.get_blocks_by_id..cSrqrrrrtrrrrvrwr)a_blocksb_blocksa_blocks_by_idb_blocks_by_idrrrget_blocks_by_idsz,BaseSequenceBlockComparison.get_blocks_by_idc s4||j}||j}|||\}}||g}|D]+}t|j}|j|vr>|||jdd||jj |j q |||jddd|j q fddt |D}|D]>\}} t|j}||jdd|j d} d} d} t |D]\} }| r}qt| | kr| | | d} n| d7} qt| s|| qY|S)NTFcs"g|] \}}|jvr||fqSrrr)ruirS deleted_idsrr szKBaseSequenceBlockComparison.get_block_comparisons_by_id..r) rpr r!r|keysrTrSrsrkr enumeraterYinsert)r"rxryrzr{ comparisonsrSrdeleted_block_indicesindexcomparison_to_insert current_indexblock_insertedr} comparisonrr~rget_block_comparisons_by_idsT            z7BaseSequenceBlockComparison.get_block_comparisons_by_idcCsg}|D]E}dg}|r|d||j}n|r+|d||j}n|r4|}n||j}d |}|d|d|dqt d |S) Nzcomparison__child-objectr0r/r)z
z
rf) get_block_comparisonsrYrkr_r!r[r r7r4joinr)r"comparisons_htmlrclassesblock_renderedrrrr4s      z$BaseSequenceBlockComparison.htmldiffN)r8r9r: staticmethodrpr|rr4rrrrrns   Esz.TF) rpr r!anyrrTrS child_blocklenrrkr) r"rxryboth_in_new_formatrra_lengthb_lengthrrSrrrr>s>    z)ListBlockComparison.get_block_comparisonsNrrrrrrR8rrRc@rb)StreamFieldComparisoncCs|sdSt|djS)NTr)boolrsr]rrr has_block_idstsz#StreamFieldComparison.has_block_idscCsN||jr||jrt|jjdd|j|jStt|jt|j S)NT) rr r!rPr stream_blockr4r?rr2r3rrrr4zszStreamFieldComparison.htmldiffN)r8r9r:rr4rrrrrs rc@r=)ChoiceFieldComparisoncCstt|jj|j|jdd}tt|jj|j|jdd}|j|jkr@g}|r1|d|fg7}|r:|d|fg7}t|St |S)NT) strings_onlyr/r0) rdictr flatchoicesgetr r!r1r2r)r"r r!diffsrrrr4s  zChoiceFieldComparison.htmldiffNr@rrrrrrArc@s,eZdZddZddZddZddZd S) M2MFieldComparisoncCst|jt|jfSr)rr r!r3rrr get_itemsszM2MFieldComparison.get_itemscCrcr)str)r"itemrrrget_item_displayrdz#M2MFieldComparison.get_item_displayc Cs*|\}}td||}g}|D]y\}}}}} |dkrE|||D] } |d|| fq$||| D] } |d|| fq7q|dkr]|||D] } |d|| fqOq|dkru||| D] } |d|| fqgq|dkr|||D] } |d|| fqqt|dd S) Nrr,r/r0deleterequalz, ) separator)rdifflibSequenceMatcher get_opcodesrkrr1r2) r"items_aitems_bsmchangesopi1i2j1j2rrrrr4s. zM2MFieldComparison.htmldiffcCs|\}}||kSr)r)r"rrrrrr7s zM2MFieldComparison.has_changedN)r8r9r:rrr4r7rrrrrs  rc@r=)TagsFieldComparisoncCs|jSr)slug)r"tagrrrrsz$TagsFieldComparison.get_item_displayN)r8r9r:rrrrrrrArc@rb)ForeignObjectComparisoncCs8|jj}|jj|jd}|jj|jd}||fS)N)pk)r related_modelobjectsfilterr firstr!)r"modelr#r$rrr get_objectssz#ForeignObjectComparison.get_objectscCs|\}}||kr:|r|rtdt|fdt|fgS|r+tdt|fgS|r8tdt|fgSdS|rBtt|StdS)Nr/r0None)rr1rr2rr(r"r#r$rrrr4s  z ForeignObjectComparison.htmldiffN)r8r9r:rr4rrrrrrrc@sFeZdZdZdZdddZddZdd Zd d Zd d Z ddZ dS)ChildRelationComparisonFTcCs2||_||_t||j|_t||j|_||_dSr)rfield_comparisonsr* related_namer r!rj)r"rrr#r$rjrrrr%s  z ChildRelationComparison.__init__cCsBt|jdd}|dur|jr|j}t|S|jjdd}t|Sr&)r*rrjr+r,r r-rrrr.sz#ChildRelationComparison.field_labelcCsi}i}g}g}t|D]*\}}t|D]!\} } | |vrq|jdur5| jdur5|j| jkr5| ||<||| <qq g} t|D]8\}}||vrut|D]+\} } | |vrt|jr^| jr^|j| jkr^qI||||| } | } | || | fqIq=| jddd| D]\}} } ||vs| |vrq| ||<||| <qt|D] \}}||vr||qt|D] \} } | |vr|| q||||fS)a This bit of code attempts to match the objects in the A revision with their counterpart in the B revision. A match is firstly attempted by PK (where a matching ID indicates they're the same). We compare remaining the objects by their field data; the objects with the fewest fields changed are matched until there are no more possible matches left. This returns 4 values: - map_forwards => a mapping of object indexes from the B version to the A version - map_backwards => a mapping of object indexes from the A version to the B version - added => a list of indices for objects that didn't exist in the B version - deleted => a list of indices for objects that didn't exist in the A version Note the indices are 0-based array indices indicating the location of the object in either the objs_a or objs_b arrays. For example: objs_a => A, B, C, D objs_b => B, C, D, E Will return the following: map_forwards = { 1: 0, # B (objs_a: objs_b) 2: 1, # C (objs_a: objs_b) 3: 2, # D (objs_a: objs_b) } map_backwards = { 0: 1, # B (objs_b: objs_a) 1: 2, # C (objs_b: objs_a) 2: 3, # D (objs_b: objs_a) } added = [4] # D in objs_b deleted = [0] # A in objs_a NcSs|dS)Nr)matchrrrTsz5ChildRelationComparison.get_mapping..)key)rrget_child_comparisonget_num_differencesrksort)r"objs_aobjs_b map_forwards map_backwardsaddeddeleteda_idxa_childb_idxb_childmatchesrnum_differencesrrr get_mapping sV&         z#ChildRelationComparison.get_mappingcCst|jj|j||Sr)ChildObjectComparisonrrrrrrrrisz,ChildRelationComparison.get_child_comparisonc Cst|j}t|j}|||\}}}}tt|}tt|}g}|D]\}} ||vr<|| d| q*|| |||| q*|D]\} } | |vr_|| | dqN|S)aU Returns a list of ChildObjectComparison objects. Representing all child objects that existed in either version. They are returned in the order they appear in the B version with deletions appended at the end. All child objects are returned, regardless of whether they were actually changed. N) rr allr!rrrrirkr) r"rrrrrrrrrrrrrrget_child_comparisonsns"   z-ChildRelationComparison.get_child_comparisonsc Cstt|j}t|j}|||\}}}}|s|rdS|D]\}}|||||} | r7dSq"dS)z Returns true if any changes were made to any of the child objects. This includes adding, deleting and reordering. TF)rr rr!rrirr7) r"rrrrrrrrrrrrr7sz#ChildRelationComparison.has_changedNr) r8r9r:r;r<r%r.rrrr7rrrrrs ` !rc@sDeZdZddZddZddZddZd d Zd d Zd dZ dS)rcCs||_||_||_||_dSr)rrr#r$)r"rrr#r$rrrr%s zChildObjectComparison.__init__cCrX)zK Returns True if this child object was created since obj_a )r$r#r3rrr is_additionz!ChildObjectComparison.is_additioncCrX)zH Returns True if this child object was deleted in obj_b )r#r$r3rrr is_deletionrz!ChildObjectComparison.is_deletioncCsD|s|s t|jddpd}t|jddpd}||SdSdS)a Returns the change in position as an integer. Positive if the object was moved down, negative if it moved up. For example: '3' indicates the object moved down three spaces. '-1' indicates the object moved up one space. sort_orderrN)rrr*r#r$)r"sort_asort_brrrget_position_changes z)ChildObjectComparison.get_position_changecCsdg}|s |r |jp|j}|jD] }||||q|S|jD] }|||j|jq#|S)z Returns a list of comparisons for all the fields in this object. Fields that haven't changed are included as well. )rrr#r$rrk)r"robjfield_comparisonrrrget_field_comparisonss   z+ChildObjectComparison.get_field_comparisonscCs |D] }|r dSqdS)NTFrr7)r"rrrrr7s z!ChildObjectComparison.has_changedcCs&d}|D] }|r|d7}q|S)z[ Returns the number of fields that differ between the two objects. rrr)r"rrrrrrs  z)ChildObjectComparison.get_num_differencesN) r8r9r:r%rrrrr7rrrrrrs  rc@s eZdZd ddZd ddZd S) r1rcCs||_||_dSr)rr)r"rrrrrr%s zTextDiff.__init__spanr0r/cCsg}|jD]3\}}|dkr|t|q|dkr'|dj||t|dq|dkr8|dj||t|dqt|j|S)Nrr0z*<{tag} class="{classname}">{value})r classnamerr/)rrkrformatrrr)r"raddition_classdeletion_classhtml change_typerrrrr2s$  zTextDiff.to_htmlNr)rr0r/)r8r9r:r%r2rrrrr1s r1cCs|dd}||}||}tdd||}g}|D]j\}}} } } |dkrG||| D] } |d| fq,|| | D] } |d| fq tags inserted indicating where the differences are. cSsVg}d}|pdD]}|r||7}q|r||d}||q|r)|||S)a Tokenises a string by splitting it into individual characters and grouping the alphanumeric ones together. This means that punctuation, whitespace, CJK characters, etc become separate tokens and words/numbers are merged together to form bigger tokens. This makes the output of the diff easier to read as words are not broken up. r)isalnumrk)texttokens current_tokencrrrtokenises      zdiff_text..tokenisecSs t|dkS)N)r)trrrr/s zdiff_text..r,r/r0rrrNr)rrrrkrr1)abra_tokb_tokrrrrrrrtokenmerged_changes current_valuecurrent_change_typerrrrrr?sJ  r?)NNF)5rdjango.core.exceptionsr django.dbrdjango.utils.encodingrdjango.utils.htmlrrrdjango.utils.safestringrdjango.utils.textr django.utils.translationr r(taggit.managersr wagtailr wagtail.fieldsr rwagtail.utils.registryrwagtail.utils.textrrrrr> CharField TextFieldrBrTrUrIrLrNrnrPrRrrrrrrrr1r?rrrrsT           $  .] ; & 9H