a
D gp @ s d dl Z d dlmZ d dlmZ d dlmZ d dlmZm Z m
Z
d dl m Z d dl
mZ d dlmZ d d lmZ d d
lmZ d d lmZmZ d d lmZ d d
lm Z e Z d:ddZ G dd dZ G dd de Z e ej!e d e ej"e d G dd de Z#e ee#d dd Z$G dd dZ%G d d d e%Z&G d d d e%Z'G d d! d!e%Z(G d"d# d#e%Z)G d$d% d%e)Z*G d&d' d'e)Z+G d(d) d)e Z,e ee,d G d*d+ d+e Z-G d,d- d-e Z.G d.d/ d/e.Z/e ee/d G d0d1 d1e Z0G d2d3 d3Z1G d4d5 d5Z2G d6d7 d7Z3d8d9 Z4dS ); N)ImproperlyConfigured)models) force_str)escape format_htmlformat_html_join) mark_safe)capfirst) gettext_lazy)TaggableManager)blocks)
RichTextField StreamField)ModelFieldRegistry)text_from_htmlFc C s< |du rt d|r&| tjkr&t dtj| |||d dS )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)r r Z
ForeignKeycomparison_class_registryregister)Z field_classr comparison_classr r T/var/www/lab.imftr.de/x/nb_venv/lib/python3.9/site-packages/wagtail/admin/compare.pyregister_comparison_class s r c @ s4 e Zd ZdZdZdd Zdd Zdd Zd d
Zd S ) FieldComparisonTFc C s" || _ ||| _||| _d S N)fieldZvalue_from_objectval_aval_b)selfr obj_aobj_br r r __init__/ s zFieldComparison.__init__c C s. t | jdd}|du r&| jjdd}t|S zL
Returns a label for this field to be displayed to the user
verbose_nameN_ )getattrr namereplacer r r$ r r r field_label4 s zFieldComparison.field_labelc C s6 | j | jkr(td| j fd| jfg S t| j S d S )Ndeletionaddition)r r TextDiffto_htmlr r r r r htmldiff@ s
zFieldComparison.htmldiffc C s | j | jkS )z7
Returns True if the field has changed
r r r0 r r r has_changedH s zFieldComparison.has_changedN) __name__
__module__ __qualname__is_fieldis_child_relationr" r+ r1 r3 r r r r r + s r c @ s e Zd Zdd ZdS )TextFieldComparisonc C s t | j| j S r ) diff_textr r r/ r0 r r r r1 P s z TextFieldComparison.htmldiffNr4 r5 r6 r1 r r r r r9 O s r9 )r c @ s e Zd Zdd ZdS )RichTextFieldComparisonc C s t t| jt| j S r )r: r r r r/ r0 r r r r1 Y s z RichTextFieldComparison.htmldiffNr; r r r r r< X s r< c C s t | dr| S t| tjtjfr(tS t| tjr8tS t| tjrHt S t| tj
rXt S t| tj rht
S t| tjrxtS t S d S )Nget_comparison_class)hasattrr=
isinstancer Z CharBlockZ TextBlockCharBlockComparisonZ RawHTMLBlockZ
RichTextBlockRichTextBlockComparisonZ StructBlockStructBlockComparisonZ StreamBlockStreamBlockComparisonZ ListBlockListBlockComparison)blockr r r get_comparison_class_for_blockb s
rF c @ s< e Zd Zdd Zdd Zdd Zdd Zd d
Zd d Zd
S )BlockComparisonc C s" || _ || _|| _|| _|| _d S r )rE exists_aexists_br r )r rE rH rI r r r r r r" x s
zBlockComparison.__init__c C s | j o | j S r )rI rH r0 r r r is_new s zBlockComparison.is_newc C s | j o | j S r )rH rI r0 r r r
is_deleted s zBlockComparison.is_deletedc C s | j | jkS r r2 r0 r r r r3 s zBlockComparison.has_changedc C s t t| j|S )zu
Return an HTML representation of this block that is safe to be included
in comparison views
)r r rE render_basicr valr r r htmlvalue s zBlockComparison.htmlvaluec C s2 | j | j}| j | j}tt|t| S r )rE rL r r r: r r/ )r Z
html_val_aZ
html_val_br r r r1 s
zBlockComparison.htmldiffN) r4 r5 r6 r" rJ rK r3 rO r1 r r r r rG w s rG c @ s e Zd Zdd Zdd ZdS )r@ c C s t t| jt| j S r )r: r r r r/ r0 r r r r1 s z CharBlockComparison.htmldiffc C s t |S r )r rM r r r rO s z CharBlockComparison.htmlvalueN)r4 r5 r6 r1 rO r r r r r@ s r@ c @ s e Zd ZdS )rA N)r4 r5 r6 r r r r rA s rA c @ s e Zd Zdd Zdd ZdS )rB c C sn g }| j j D ]J\}}| j j| j}t|}||||dd|| || || f qtdtdd|S )NT
{}
{}
{}) rE child_blocksitemslabelrF appendrO r r )r rN Z
htmlvaluesr( rE rU r r r r rO s
z StructBlockComparison.htmlvaluec C sp g }| j j D ]L\}}| j j| j}t|}||||| j| j| j| | j |
f qt dt dd|S )NrP rQ rR )
rE rS rT rU rF rV rH rI r r r1 r r )r Z htmldiffsr( rE rU r r r r r1 s&
z StructBlockComparison.htmldiffN)r4 r5 r6 rO r1 r r r r rB s rB c @ s4 e Zd Zedd Zedd Zdd Zdd Zd S )
BaseSequenceBlockComparisonc C s t d S r )NotImplementedError)rN r r r get_blocks_from_value s z1BaseSequenceBlockComparison.get_blocks_from_valuec C s$ dd | D }dd |D }||fS )Nc S s i | ] }|j |qS r id.0rE r r r
z@BaseSequenceBlockComparison.get_blocks_by_id..c S s i | ] }|j |qS r rZ r\ r r r r^ r_ r )a_blocksb_blocksa_blocks_by_idb_blocks_by_idr r r get_blocks_by_id s z,BaseSequenceBlockComparison.get_blocks_by_idc s8 | | j}| | j}| ||\}}| | g }|D ]V}t|j}|j|v r||||jdd||j j |j q@|||jddd |j q@ fddt
|D }|D ]\}} t|j}||jdd|j d }
d} d} t
|D ]:\}
}| rq| | kr| |
|
d} q$| d7 } q| s||
q|S )NTFc s" g | ]\}}|j v r||fqS r rZ )r] irE Z deleted_idsr r
s zKBaseSequenceBlockComparison.get_block_comparisons_by_id..r )
rY r r rd keysrF rE r[ rV r enumeraterJ insert)r r` ra rb rc comparisonsrE r Zdeleted_block_indicesindexZcomparison_to_insertZ
current_indexZblock_insertedre
comparisonr rf r get_block_comparisons_by_id sR
z7BaseSequenceBlockComparison.get_block_comparisons_by_idc C s g }| D ]}dg}| r6|d ||j}n>| rV|d ||j}n | rh| }n ||j}d |}|d| d| d q t
d |S ) Nzcomparison__child-objectr- r, r& z z
rQ ) get_block_comparisonsrJ rV rO r rK r r3 r1 joinr )r Zcomparisons_htmlrn classesZblock_renderedr r r r1 s
z$BaseSequenceBlockComparison.htmldiffN)r4 r5 r6 staticmethodrY rd ro r1 r r r r rW s
E r_ z.TF) rY r r anyro rF rE Z child_blocklenrj rV r )
r r` ra Zboth_in_new_formatrl r Za_lengthZb_lengthrm rE r r r rp > s> z)ListBlockComparison.get_block_comparisonsNrv r r r r rD 8 s
rD c @ s e Zd Zdd Zdd ZdS )StreamFieldComparisonc C s |sdS t |d jS )NTr )boolr[ rM r r r
has_block_idst s z#StreamFieldComparison.has_block_idsc C sR | | jr4| | jr4t| jjdd| j| j S tt| jt| j S d S )NT)
r| r r rC r Z stream_blockr1 r: r r/ r0 r r r r1 z s z StreamFieldComparison.htmldiffN)r4 r5 r6 r| r1 r r r r rz s s rz c @ s e Zd Zdd ZdS )ChoiceFieldComparisonc C s t t| jj| j| jdd}t t| jj| j| jdd}| j| jkrg }|rb|d|fg7 }|rt|d|fg7 }t| S t |S d S )NT)Z strings_onlyr, r- )
r dictr Z flatchoicesgetr r r. r/ r )r r r Zdiffsr r r r1 s z ChoiceFieldComparison.htmldiffNr; r r r r r} s r} c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
M2MFieldComparisonc C s t | jt | jfS r )rt r r r0 r r r get_items s z M2MFieldComparison.get_itemsc C s t |S r )str)r itemr r r get_item_display s z#M2MFieldComparison.get_item_displayc C s, | \}}td||}g }| D ]\}}}}} |dkr||| D ]}
|d| |
f qH||| D ]}
|d| |
f qnq&|dkr||| D ]}
|d| |
f qq&|dkr||| D ]}
|d| |
f qq&|dkr&||| D ]}
|d| |
f qq&t|dd S )
Nr r) r, r- deleterk equalz, ) separator)r difflibSequenceMatcher get_opcodesrV r r. r/ ) r items_aitems_bsmchangesopi1i2j1j2r r r r r1 s& zM2MFieldComparison.htmldiffc C s | \}}||kS r )r )r r r r r r r3 s z M2MFieldComparison.has_changedN)r4 r5 r6 r r r1 r3 r r r r r s r c @ s e Zd Zdd ZdS )TagsFieldComparisonc C s |j S r )Zslug)r tagr r r r s z$TagsFieldComparison.get_item_displayN)r4 r5 r6 r r r r r r s r c @ s e Zd Zdd Zdd ZdS )ForeignObjectComparisonc C s8 | j j}|jj| jd }|jj| jd }||fS )N)pk)r
related_modelZobjectsfilterr firstr )r modelr r! r r r get_objects s z#ForeignObjectComparison.get_objectsc C s | \}}||krr|r<|r|| q>t |D ] \} }
| |vrf|| qf||||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
Nc S s | d S )N r )matchr r r T r_ z5ChildRelationComparison.get_mapping..)key)rj r get_child_comparisonget_num_differencesrV sort)r objs_aobjs_b map_forwards
map_backwardsaddeddeleteda_idxa_childb_idxb_childmatchesrn num_differencesr r r get_mapping sR &
z#ChildRelationComparison.get_mappingc C s t | jj| j||S r )ChildObjectComparisonr r r r r r r r i s z,ChildRelationComparison.get_child_comparisonc C s t | j }t | j }| ||\}}}}tt|}tt|}g }| D ]>\}} ||v rx|| d| qT|| ||| | qT| D ]"\}
} |
|v r|| | d q|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)
rt r allr r r~ rj rT rV r ) r r r r r r r rl r r r r r r r get_child_comparisonsn s
z-ChildRelationComparison.get_child_comparisonsc
C st t | j }t | j }| ||\}}}}|s8|r| jp | j}| jD ]}|||| q&n | jD ]}||| j| j qD|S )z
Returns a list of comparisons for all the fields in this object.
Fields that haven't changed are included as well.
)r r r r! r rV )r rl objZfield_comparisonr r r get_field_comparisons s
z+ChildObjectComparison.get_field_comparisonsc C s | D ]}| r dS qdS )NTFr r3 )r rn r r r r3 s z!ChildObjectComparison.has_changedc C s& d}| D ]}| r |d7 }q |S )z[
Returns the number of fields that differ between the two
objects.
r rh r )r r rn r r r r s
z)ChildObjectComparison.get_num_differencesN)
r4 r5 r6 r" r r r r r3 r r r r r r s
r c @ s e Zd Zd
ddZd ddZd S ) r. r c C s || _ || _d S r )r r )r r r r r r r" s zTextDiff.__init__spanr- r, c C s g }| j D ]f\}}|dkr*|t| q
|dkrN|dj||t|d q
|dkr
|dj||t|d q
t| j|S )Nr r- z*<{tag} class="{classname}">{value})r classnamer r, )r rV r formatr r rq )r r Zaddition_classZdeletion_classhtml change_typer r r r r/ s"
zTextDiff.to_htmlN)r )r r- r, )r4 r5 r6 r" r/ r r r r r. s
r. c C s dd }|| }||}t dd ||}g }| D ]\}}} }
} |dkr||| D ]} |d| f qX||
| D ]} |d| f qxq6|dkr||| D ]} |d| f qq6|d kr||
| D ]} |d| f qq6|d
kr6||| D ]} |d
| f qq6g }
g }d }|D ]D\}}||krT|d urP|
|d |f g }|}|| q |r||
|d |f t|
S )
z
Performs a diffing algorithm on two pieces of text. Returns
a string of HTML containing the content of both texts with
tags inserted indicating where the differences are.
c S sV g }d}| pdD ]2}| r&||7 }q|r8|| d}|| q|rR|| |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 )isalnumrV )texttokens
current_tokencr r r tokenise s
zdiff_text..tokenisec S s t | dkS )N )ry )tr r r r / r_ zdiff_text..r) r, r- r rk r Nr )r r r rV rq r. )abr Za_tokZb_tokr r r r r r r tokenZmerged_changesZ
current_valueZcurrent_change_typer r r r r r: sB
r: )NNF)5r Zdjango.core.exceptionsr Z django.dbr Zdjango.utils.encodingr Zdjango.utils.htmlr r r Zdjango.utils.safestringr Zdjango.utils.textr Zdjango.utils.translationr
r% Ztaggit.managersr Zwagtailr Zwagtail.fieldsr
r Zwagtail.utils.registryr Zwagtail.utils.textr r r r r9 Z CharFieldZ TextFieldr< rF rG r@ rA rB rW rC rD rz r} r r r r r r. r: r r r r sR
$ .]
; & 9H