3
d�GZ�_ �
@ s� d Z ddlmZ ddlZddlZddlZddlZddlZddlZddl Z ddl
m Z m Z dZ
dZyddlmZmZ W n( ek
r� ddlmZmZ eZY nX dZdZd Zd
Ze jd0kr�eefZG d d
� d
e �Z G dd� de �Z G dd� de �Z G dd� de e �Z!dd� Z"ej#ej$e"d�Z%d1dd�Z&dd� 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.G d(d)� d)e*�Z/G d*d+� d+e*�Z0G d,d-� d-e(�Z1d.d/� Z2dS )2z Apply JSON-Patches (RFC 6902) � )�unicode_literalsN)� JsonPointer�JsonPointerException� )�MutableMapping�MutableSequenceu Stefan Kögl <stefan@skoegl.net>z1.21z0https://github.com/stefankoegl/python-json-patchzModified BSD License� c @ s e Zd ZdZdS )�JsonPatchExceptionzBase Json Patch exceptionN)�__name__�
__module__� __qualname__�__doc__� r r � /usr/lib/python3.6/jsonpatch.pyr G s r c @ s e Zd ZdZdS )�InvalidJsonPatchz, Raised if an invalid JSON Patch is created N)r
r r r
r r r r r K s r c @ s e Zd ZdZdS )�JsonPatchConflicta Raised if patch could not be applied due to conflict situation such as:
- attempt to add object key then it already exists;
- attempt to operate with nonexistence object key;
- attempt to insert value to array at position beyond of it size;
- etc.
N)r
r r r
r r r r r O s r c @ s e Zd ZdZdS )�JsonPatchTestFailedz A Test operation failed N)r
r r r
r r r r r X s r c C s@ t jt�}x | D ]\}}|| j|� qW tdd� |j� D ��S )z'Convert duplicate keys values to lists.c s s. | ]&\}}|t |�d kr |d n|fV qdS )r r N)�len)�.0�key�valuesr r r � <genexpr>e s z multidict.<locals>.<genexpr>)� collections� defaultdict�list�append�dict�items)Z
ordered_pairsZmdictr �valuer r r � multidict\ s
r )Zobject_pairs_hookFc C s* t |t�rtj|�}nt|�}|j| |�S )aO Apply list of patches to specified json document.
:param doc: Document object.
:type doc: dict
:param patch: JSON patch as list of dicts or raw JSON-encoded string.
:type patch: list or str
:param in_place: While :const:`True` patch will modify target document.
By default patch will be applied to document copy.
:type in_place: bool
:return: Patched document object.
:rtype: dict
>>> doc = {'foo': 'bar'}
>>> patch = [{'op': 'add', 'path': '/baz', 'value': 'qux'}]
>>> other = apply_patch(doc, patch)
>>> doc is not other
True
>>> other == {'foo': 'bar', 'baz': 'qux'}
True
>>> patch = [{'op': 'add', 'path': '/baz', 'value': 'qux'}]
>>> apply_patch(doc, patch, in_place=True) == {'foo': 'bar', 'baz': 'qux'}
True
>>> doc == other
True
)�
isinstance�
basestring� JsonPatch� from_string�apply)�doc�patch�in_placer r r � apply_patcho s
r( c C s t j| |�S )a� Generates patch by comparing of two document objects. Actually is
a proxy to :meth:`JsonPatch.from_diff` method.
:param src: Data source document object.
:type src: dict
:param dst: Data source document object.
:type dst: dict
>>> src = {'foo': 'bar', 'numbers': [1, 3, 4, 8]}
>>> dst = {'baz': 'qux', 'numbers': [1, 4, 7]}
>>> patch = make_patch(src, dst)
>>> new = patch.apply(src)
>>> new == dst
True
)r" � from_diff)�src�dstr r r �
make_patch� s r, c @ s� e Zd ZdZdd� Zdd� Zdd� ZeZdd � Zd
d � Z d d
� Z
dd� Z e dd� �Z
e d dd��Zdd� Zedd� �Zd dd�Zd d � Zd S )!r" ag A JSON Patch is a list of Patch Operations.
>>> patch = JsonPatch([
... {'op': 'add', 'path': '/foo', 'value': 'bar'},
... {'op': 'add', 'path': '/baz', 'value': [1, 2, 3]},
... {'op': 'remove', 'path': '/baz/1'},
... {'op': 'test', 'path': '/baz', 'value': [1, 3]},
... {'op': 'replace', 'path': '/baz/0', 'value': 42},
... {'op': 'remove', 'path': '/baz/1'},
... ])
>>> doc = {}
>>> result = patch.apply(doc)
>>> expected = {'foo': 'bar', 'baz': [42]}
>>> result == expected
True
JsonPatch object is iterable, so you could easily access to each patch
statement in loop:
>>> lpatch = list(patch)
>>> expected = {'op': 'add', 'path': '/foo', 'value': 'bar'}
>>> lpatch[0] == expected
True
>>> lpatch == patch.patch
True
Also JsonPatch could be converted directly to :class:`bool` if it contains
any operation statements:
>>> bool(patch)
True
>>> bool(JsonPatch([]))
False
This behavior is very handy with :func:`make_patch` to write more readable
code:
>>> old = {'foo': 'bar', 'numbers': [1, 3, 4, 8]}
>>> new = {'baz': 'qux', 'numbers': [1, 4, 7]}
>>> patch = make_patch(old, new)
>>> if patch:
... # document have changed, do something useful
... patch.apply(old) #doctest: +ELLIPSIS
{...}
c C s || _ ttttttd�| _d S )N)�remove�add�replace�moveZtest�copy)r& �RemoveOperation� AddOperation�ReplaceOperation�
MoveOperation�
TestOperation�
CopyOperation�
operations)�selfr& r r r �__init__� s zJsonPatch.__init__c C s | j � S )z str(self) -> self.to_string())� to_string)r9 r r r �__str__� s zJsonPatch.__str__c C s
t | j�S )N)�boolr& )r9 r r r �__bool__� s zJsonPatch.__bool__c C s
t | j�S )N)�iterr& )r9 r r r �__iter__� s zJsonPatch.__iter__c C s t t| j��S )N)�hash�tuple�_ops)r9 r r r �__hash__� s zJsonPatch.__hash__c C s t |t�sdS | j|jkS )NF)r r" rC )r9 �otherr r r �__eq__� s
zJsonPatch.__eq__c C s
| |k S )Nr )r9 rE r r r �__ne__� s zJsonPatch.__ne__c C s t |�}| |�S )z�Creates JsonPatch instance from string source.
:param patch_str: JSON patch as raw string.
:type patch_str: str
:return: :class:`JsonPatch` instance.
)�
_jsonloads)�clsZ patch_strr& r r r r# � s zJsonPatch.from_stringTc C s* t � }|jdd||� t|j� �}| |�S )aO Creates JsonPatch instance based on comparing of two document
objects. Json patch would be created for `src` argument against `dst`
one.
:param src: Data source document object.
:type src: dict
:param dst: Data source document object.
:type dst: dict
:return: :class:`JsonPatch` instance.
>>> src = {'foo': 'bar', 'numbers': [1, 3, 4, 8]}
>>> dst = {'baz': 'qux', 'numbers': [1, 4, 7]}
>>> patch = JsonPatch.from_diff(src, dst)
>>> new = patch.apply(src)
>>> new == dst
True
� N)� DiffBuilder�_compare_valuesr �execute)rI r* r+ � optimizationZbuilderZopsr r r r) s zJsonPatch.from_diffc C s t j| j�S )z!Returns patch set as JSON string.)�json�dumpsr& )r9 r r r r; ! s zJsonPatch.to_stringc C s t t| j| j��S )N)rB �map�_get_operationr& )r9 r r r rC % s zJsonPatch._opsFc C s, |st j|�}x| jD ]}|j|�}qW |S )a/ Applies the patch to given object.
:param obj: Document object.
:type obj: dict
:param in_place: Tweaks way how patch would be applied - directly to
specified `obj` or to his copy.
:type in_place: bool
:return: Modified `obj`.
)r1 �deepcopyrC r$ )r9 �objr' � operationr r r r$ ) s
zJsonPatch.applyc C sT d|krt d��|d }t|t�s*t d��|| jkrBt dj|���| j| }||�S )N�opz&Operation does not contain 'op' memberzOperation must be a stringzUnknown operation {0!r})r r r! r8 �format)r9 rU rV rI r r r rR > s
zJsonPatch._get_operationN)T)F)r
r r r
r: r< r> Z __nonzero__r@ rD rF rG � classmethodr# r) r; �propertyrC r$ rR r r r r r" � s -
r" c @ s^ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Ze d d
� �Z
e dd� �Z e j dd� �Z dS )�PatchOperationz'A single operation inside a JSON Patch.c C s |d | _ t| j �| _|| _d S )N�path)�locationr �pointerrU )r9 rU r r r r: Q s
zPatchOperation.__init__c C s t d��dS )zAAbstract method that applies patch operation to specified object.z!should implement patch operation.N)�NotImplementedError)r9 rT r r r r$ V s zPatchOperation.applyc C s t t| jj� ��S )N)rA � frozensetrU r )r9 r r r rD Z s zPatchOperation.__hash__c C s t |t�sdS | j|jkS )NF)r rZ rU )r9 rE r r r rF ] s
zPatchOperation.__eq__c C s
| |k S )Nr )r9 rE r r r rG b s zPatchOperation.__ne__c C s dj | jjd d� �S )N�/r ���)�joinr] �parts)r9 r r r r[ e s zPatchOperation.pathc C s2 yt | jjd �S tk
r, | jjd S X d S )Nr ra ra )�intr] rc �
ValueError)r9 r r r r i s zPatchOperation.keyc C s* t |�| jjd< | jj| _| j| jd< d S )Nr r[ ra )�strr] rc r[ r\ rU )r9 r r r r r p s
N)
r
r r r
r: r$ rD rF rG rY r[ r �setterr r r r rZ N s rZ c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) r2 z/Removes an object property or an array element.c C sX | j j|�\}}y
||= W n8 ttfk
rR } zdj|�}t|��W Y d d }~X nX |S )Nz&can't remove non-existent object '{0}')r] �to_last�KeyError�
IndexErrorrW r )r9 rT �subobj�part�ex�msgr r r r$ z s
zRemoveOperation.applyc C s0 | j |kr,| j|kr$| jd7 _n|d8 }|S )Nr )r[ r )r9 r[ r r r r �_on_undo_remove� s
z RemoveOperation._on_undo_removec C s0 | j |kr,| j|kr$| jd8 _n|d8 }|S )Nr )r[ r )r9 r[ r r r r � _on_undo_add� s
z RemoveOperation._on_undo_addN)r
r r r
r$ ro rp r r r r r2 w s
r2 c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) r3 z,Adds an object property or an array element.c C s� y| j d }W n* tk
r8 } ztd��W Y d d }~X nX | jj|�\}}t|t�r�|dkrh|j|� q�|t|�ks||dk r�t d��q�|j
||� n4t|t �r�|d kr�|}q�|||< nt dj
t|����|S )Nr z/The operation does not contain a 'value' member�-r z can't insert outside of listzinvalid document type {0})rU ri r r] rh r r r r r �insertr � TypeErrorrW �type)r9 rT r rm rk rl r r r r$ � s$
zAddOperation.applyc C s0 | j |kr,| j|kr$| jd7 _n|d7 }|S )Nr )r[ r )r9 r[ r r r r ro � s
z AddOperation._on_undo_removec C s0 | j |kr,| j|kr$| jd8 _n|d7 }|S )Nr )r[ r )r9 r[ r r r r rp � s
zAddOperation._on_undo_addN)r
r r r
r$ ro rp r r r r r3 � s r3 c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) r4 z=Replaces an object property or an array element by new value.c C s� y| j d }W n* tk
r8 } ztd��W Y d d }~X nX | jj|�\}}|d krV|S t|t�r~|t|�kst|dk r�td��n8t|t �r�||kr�dj
|�}t|��nt dj
t |����|||< |S )Nr z/The operation does not contain a 'value' memberr z can't replace outside of listz'can't replace non-existent object '{0}'zinvalid document type {0})
rU ri r r] rh r r r r r rW rs rt )r9 rT r rm rk rl rn r r r r$ � s$
zReplaceOperation.applyc C s |S )Nr )r9 r[ r r r r ro � s z ReplaceOperation._on_undo_removec C s |S )Nr )r9 r[ r r r r rp � s z ReplaceOperation._on_undo_addN)r
r r r
r$ ro rp r r r r r4 � s r4 c @ sN e Zd ZdZdd� Zedd� �Zedd� �Zejdd� �Zd d
� Z d d � Z
d
S )r5 z=Moves an object property or an array element to new location.c ! C s� yt | jd �}W n* tk
r< } ztd��W Y d d }~X nX |j|�\}}y || }W n2 ttfk
r� } ztt|���W Y d d }~X nX | j|kr�|S t |t
�r�| jj |�r�td��t d| jd d��j
|�}td| j|d��j
|�}|S )N�fromz.The operation does not contain a 'from' memberz(Cannot move values into its own childrenr- )rV r[ r. )rV r[ r )r rU ri r rh rj r rf r] r r �containsr2 r$ r3 r\ )r9 rT �from_ptrrm rk rl r r r r r$ � s2
zMoveOperation.applyc C s" t | jd �}dj|jd d� �S )Nru r` r ra )r rU rb rc )r9 rw r r r � from_path s zMoveOperation.from_pathc C s< t | jd �}yt|jd �S tk
r6 |jd S X d S )Nru r ra ra )r rU rd rc rs )r9 rw r r r �from_key s
zMoveOperation.from_keyc C s, t | jd �}t|�|jd< |j| jd< d S )Nru r ra )r rU rf rc r[ )r9 r rw r r r ry s c C s\ | j |kr,| j|kr$| jd7 _n|d8 }| j|krX| j|krP| jd7 _n|d7 }|S )Nr )rx ry r[ r )r9 r[ r r r r ro # s
z MoveOperation._on_undo_removec C s\ | j |kr,| j|kr$| jd8 _n|d8 }| j|krX| j|krP| jd8 _n|d7 }|S )Nr )rx ry r[ r )r9 r[ r r r r rp 0 s
zMoveOperation._on_undo_addN) r
r r r
r$ rY rx ry rg ro rp r r r r r5 � s "
r5 c @ s e Zd ZdZdd� ZdS )r6 z!Test value by specified location.c # C s� y0| j j|�\}}|d kr |}n| j j||�}W n. tk
r^ } ztt|���W Y d d }~X nX y| jd }W n* tk
r� } ztd��W Y d d }~X nX ||kr�d}t|j |t
|�|t
|����|S )Nr z/The operation does not contain a 'value' memberz0{0} ({1}) is not equal to tested value {2} ({3})) r] rh �walkr r rf rU ri r rW rt )r9 rT rk rl �valrm r rn r r r r$ A s" zTestOperation.applyN)r
r r r
r$ r r r r r6 > s r6 c @ s e Zd ZdZdd� ZdS )r7 zA Copies an object property or an array element to a new location c ! C s� yt | jd �}W n* tk
r< } ztd��W Y d d }~X nX |j|�\}}ytj|| �}W n2 ttfk
r� } ztt |���W Y d d }~X nX t
d| j |d��j |�}|S )Nru z.The operation does not contain a 'from' memberr. )rV r[ r )
r rU ri r rh r1 rS rj r rf r3 r\ r$ )r9 rT rw rm rk rl r r r r r$ \ s
zCopyOperation.applyN)r
r r r
r$ r r r r r7 Y s r7 c @ s| e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd S ) rK c C s4 i i g| _ g g g| _g | _}||d g|d d �< d S )N)�
index_storage�index_storage2�_DiffBuilder__root)r9 �rootr r r r: t s
zDiffBuilder.__init__c C sh y:| j | }|j|�}|d kr*|g||< n|| j|� W n( tk
rb | j| j||f� Y nX d S )N)r| �getr rs r} )r9 r �index�st�storage�storedr r r � store_indexz s
zDiffBuilder.store_indexc C s� y | j | j|�}|r |j� S W nZ tk
rz | j| }x:tt|�d dd�D ]"}|| d |krP|j|�d S qPW Y nX d S )Nr r ra ra )r| r� �poprs r} �ranger )r9 r r� r� r� �ir r r �
take_index� s
zDiffBuilder.take_indexc C s, | j }|d }|||g |d< |d< |d S )Nr r )r~ )r9 rV r Zlastr r r rr � s zDiffBuilder.insertc C s* |\}}}||d< ||d< g |d d �< d S )Nr r r )r9 r� Z link_prevZ link_next�_r r r r- � s
zDiffBuilder.removec c s2 | j }|d }x ||k r,|d V |d }qW d S )Nr � )r~ )r9 �startr �currr r r � iter_from� s
zDiffBuilder.iter_fromc c s2 | j }|d }x ||k r,|d V |d }qW d S )Nr r� )r~ )r9 r r� r r r r@ � s
zDiffBuilder.__iter__c c s� | j }|d }x�||k r�|d |k r�|d |d d }}|j|jkr�t|�tkr�t|�tkr�td|j|jd d��jV |d d }q|d jV |d }qW d S )Nr r� r/ r )rV r[ r )r~ r\ rt r2 r3 r4 rU )r9 r r� Zop_firstZ op_secondr r r rM � s
zDiffBuilder.executec C s� | j |t�}|d k r�|d }t|j�tkrPx$| j|�D ]}|j|j|j�|_q6W | j|� |j t
||�kr�t d|j t
||�d��}| j |� n.t
dt
||�|d��}| j |�}| j||t� d S )Nr� r0 )rV ru r[ r. )rV r[ r )r� �
_ST_REMOVErt r rd r� ro r[ r- r\ �
_path_joinr5 rr r3 r� �_ST_ADD) r9 r[ r �itemr� rV �v�new_op� new_indexr r r � _item_added� s&
zDiffBuilder._item_addedc C s� t dt||�d��}| j|t�}| j|�}|d k r�|d }t|j�tkrnx$| j|�D ]}|j |j
|j�|_qTW | j |� |j |j kr�t
d|j |j d��}||d< q�| j |� n| j||t� d S )Nr- )rV r[ r� r0 )rV ru r[ )r2 r� r� r� rr rt r rd r� rp r[ r- r\ r5 r� r� ) r9 r[ r r� r� r� r� rV r� r r r �
_item_removed� s&
zDiffBuilder._item_removedc C s | j tdt||�|d��� d S )Nr/ )rV r[ r )rr r4 r� )r9 r[ r r� r r r �_item_replaced� s zDiffBuilder._item_replacedc C s� t |j� �}t |j� �}|| }|| }x"|D ]}| j|t|�|| � q.W x"|D ]}| j|t|�|| � qRW x(||@ D ] }| j|||| || � qzW d S )N)�set�keysr� rf r� rL ) r9 r[ r* r+ Zsrc_keysZdst_keysZ
added_keysZ removed_keysr r r r �_compare_dicts� s
zDiffBuilder._compare_dictsc C s t |�t |� }}t||�}t||�}x�t|�D ]�}||k r�|| || } }
| |
krZq0q�t| t�r�t|
t�r�| jt||�| |
� q�t| t�r�t|
t�r�| j t||�| |
� q�| j
||| � | j |||
� q0||kr�| j
|||| � q0| j |||| � q0W d S )N) r �max�minr� r r r� r� r �_compare_listsr� r� ) r9 r[ r* r+ Zlen_srcZlen_dstZmax_lenZmin_lenr �old�newr r r r� s&
zDiffBuilder._compare_listsc C sr ||kr d S t |t�r6t |t�r6| jt||�||� n8t |t�r`t |t�r`| jt||�||� n| j|||� d S )N)r r r� r� r r� r� )r9 r[ r r* r+ r r r rL ' s
zDiffBuilder._compare_valuesN)r
r r r: r� r� rr r- r� r@ rM r� r� r� r� r� rL r r r r rK r s rK c C s, |d kr | S | d t |�jdd�jdd� S )Nr` �~z~0z~1)rf r/ )r[ r r r r r� 7 s r� )r r )F)3r
Z
__future__r r r1 � functools�inspect� itertoolsrO �sysZ jsonpointerr r r� r� �collections.abcr r � ImportErrorZunicoderf �
__author__� __version__Z __website__Z __license__� version_info�bytesr! � Exceptionr r r �AssertionErrorr r �partial�loadsrH r( r, �objectr" rZ r2 r3 r4 r5 r6 r7 rK r� r r r r �<module>! sT
% &) 2$S F