File: //kunden/lib/python3/dist-packages/breezy/__pycache__/shelf.cpython-39.pyc
a
�*�^�I � @ s� d dl mZ d dlZd dlZddlmZ ddlmZ ee� d� G dd� dej�Z G d d
� d
ej�Z
G d d � d ej�Z G d
d� de �Z
G dd� de �ZG dd� de �ZdS )� )�absolute_importN� )�errors)� lazy_importzt
from breezy import (
bencode,
merge,
merge3,
transform,
)
from breezy.bzr import (
pack,
)
c @ s e Zd ZdZdS )� ShelfCorruptzShelf corrupt.N)�__name__�
__module__� __qualname__�_fmt� r r �./usr/lib/python3/dist-packages/breezy/shelf.pyr % s r c @ s e Zd ZdZdd� ZdS )�
NoSuchShelfIdz.No changes are shelved with id "%(shelf_id)d".c C s t jj| |d� d S )N)�shelf_id�r �BzrError�__init__��selfr r r r r . s zNoSuchShelfId.__init__N�r r r r
r r r r r r
* s r
c @ s e Zd ZdZdd� ZdS )�InvalidShelfIdz?"%(invalid_id)s" is not a valid shelf id, try a number instead.c C s t jj| |d� d S )N)�
invalid_idr )r r r r r r 6 s zInvalidShelfId.__init__Nr r r r r r 2 s r c @ s� e Zd ZdZd'dd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
e dd� �Z dd� Z
dd� Zdd� Zd(dd�Zdd � Zd d � Zd d � Ze d)d!d"��Zd*d#d$�Zed+d%d&��ZdS ),� ShelfCreatorz5Create a transform to shelve objects and its inverse.Nc C s� || _ |�� | _zZ|| _t�| j�| _z(i | _i | _i | _ |j
| j|d�| _
W n | j� � � Y n0 W n | j� � � Y n0 dS )a� Constructor.
:param work_tree: The working tree to apply changes to. This is not
required to be locked - a tree_write lock will be taken out.
:param target_tree: The tree to make the working tree more similar to.
This is not required to be locked - a read_lock will be taken out.
:param file_list: The files to make more similar to the target.
)Zspecific_filesN) � work_treeZ
get_transform�work_transform� target_tree� transform�TransformPreview�shelf_transform�renames�creation�deletion� iter_changes�finalize)r r r Z file_listr r r r = s$
�
zShelfCreator.__init__c c s | j D �]�}|jd du r*|jd dkr*q|jd du rH|jd dkrHq|jd du sd|jd du r�|jd |jd |jd |jf| j|j< d|j|jd |jd fV q|jd du s�|jd du �r|jd |jd |jd |jf| j|j< d|j|jd |jd fV q|jd |jd k�s:|jd |jd k�r`|j|jf| j |j< d|jf|j V |jd |jd k�r�d |j|jd |jd |jd fV q|jd d
k�r�| j
� |jd �}| j � |jd �}d |j|jd ||fV q|j
rd |jfV qdS )
a� Iterable of tuples describing shelvable changes.
As well as generating the tuples, this updates several members.
Tuples may be::
('add file', file_id, work_kind, work_path)
('delete file', file_id, target_kind, target_path)
('rename', file_id, target_path, work_path)
('change kind', file_id, target_kind, work_kind, target_path)
('modify text', file_id)
('modify target', file_id, target_target, work_target)
r Nr � F�add file� delete file�rename� change kind�symlink�
modify target� modify text)r! �kind�name� versionedZ parent_idr �file_id�pathr r r �get_symlink_targetr Zchanged_content)r �changeZt_targetZw_targetr r r �iter_shelvableX s4
� � ,&�zShelfCreator.iter_shelvablec C s� |d dkr | � |d � n�|d dkr8| �|d � nd|d dkrT| �|d � nH|d dv rp| �|d � n,|d dkr�| �|d � ntd|d ��d S )
z-Shelve a change in the iter_shelvable format.r r& r r% r$ )r' r* r) zUnknown change kind: "%s"N)�
shelve_rename�shelve_deletion�shelve_creation�shelve_content_change�shelve_modify_target�
ValueError�r r1 r r r �
shelve_change� s zShelfCreator.shelve_changec C s$ d}| � � D ]}| �|� q |duS )zeShelve all changes.
:return: ``True`` if changes were shelved, otherwise ``False``.
N)r2 r: r9 r r r �
shelve_all� s zShelfCreator.shelve_allc C sr | j | \}}| j�|�}| j�|d �}| j�|d ||� | j�|�}| j�|d �}| j�|d ||� dS )zjShelve a file rename.
:param file_id: The file id of the file to shelve the renaming of.
r r N)r r �trans_id_file_id� adjust_pathr )r r. �names�parents�
w_trans_idZ work_parent�
s_trans_idZ shelf_parentr r r r3 � s zShelfCreator.shelve_renamec C s� | j �|�}| j �|�}| j�|�}| j�|� | j�||� | j�|�}| j�|�}| j�|�}| j�|� | j�||� dS )z�Shelve a change of symlink target.
:param file_id: The file id of the symlink which changed target.
:param new_target: The target that the symlink should have due
to shelving.
N) r �id2pathr0 r r< �delete_contentsZcreate_symlinkr r )r r. �new_path�
new_targetr@ Zold_pathZ
old_targetrA r r r r7 � s z!ShelfCreator.shelve_modify_targetc C s\ | j �|�}| j �|� | j �||� | j�|�}| j�|� | �||�}| j�||� dS )z�Shelve text changes to a file, using provided lines.
:param file_id: The file id of the file to shelve the text of.
:param new_lines: The lines that the file should have due to shelving.
N)r r< rC � create_filer �_inverse_lines)r r. � new_linesr@ rA Z
inverse_linesr r r � shelve_lines� s zShelfCreator.shelve_linesc C s. | � |�}| �|� t�| |||�|�� d S �N)r< rC r �create_from_treerB )�tt�treer. Ztrans_idr r r �_content_from_tree� s
z ShelfCreator._content_from_treec C s( | � | j| j|� | � | j| j|� dS )z�Shelve a kind change or binary file content change.
:param file_id: The file id of the file to shelve the content change
of.
N)rN r r r r )r r. r r r r6 � s z"ShelfCreator.shelve_content_changec
C s> | j | \}}}}|d }| �| j|| j| j||||� dS )z�Shelve creation of a file.
This handles content and inventory id.
:param file_id: The file_id of the file to shelve creation of.
r N)r �_shelve_creationr r r )r r. r+ r, �parentr- �versionr r r r5 � s
�z ShelfCreator.shelve_creationc C s^ | j | \}}}}| j�|�}| j�|�s.d}|d }| j| j|| j| j|||||d� dS )z�Shelve deletion of a file.
This handles content and inventory id.
:param file_id: The file_id of the file to shelve deletion of.
Nr )�
existing_path)r r rB r Z has_filenamerO r r )r r. r+ r, rP r- rR rQ r r r r4 � s
�z ShelfCreator.shelve_deletionc
C s� |� |�}
|d ur$|d ur$|�|
� |�|
� | d urB|�| �} n
|� |�} |d ur�|� |�} |�|| | � | d u r�|d u r�|�dg| � nt�|| ||�|�� |r�|� || � d S )N� )
r< rC Zunversion_fileZtrans_id_tree_pathr= rF r rK rB Z version_file)
r rM r. Zfrom_transformZ to_transformr+ r, rP rQ rR r@ rA Z s_parent_idr r r rO � s&
�z ShelfCreator._shelve_creationc C sB | j �|�}| j �|�}| j�|�}| j�|�}t�|||��� S )zAProduce a version with only those changes removed from new_lines.)r rB Zget_file_linesr Zmerge3ZMerge3Z merge_lines)r rH r. Z target_pathZ target_linesZ work_pathZ
work_linesr r r rG s
zShelfCreator._inverse_linesc C s | j �� | j�� dS )z0Release all resources used by this ShelfCreator.N)r r" r �r r r r r" s
zShelfCreator.finalizec C s | j �� dS )z!Shelve changes from working tree.N)r ZapplyrT r r r r s zShelfCreator.transformc C s0 d|i}|d ur |� d�|d< | �t�|�d�S )N� revision_id�utf-8� message)�s metadata)�encodeZ bytes_record�bencode)�
serializer� revision_id�message�metadatar r r �metadata_record" s
�z ShelfCreator.metadata_recordc C s( t �| j� | j�� }| �|| j||�S )z�Serialize the shelved changes to a file.
:param shelf_file: A file-like object to write the shelf to.
:param message: An optional message describing the shelved changes.
:return: the filename of the written file.
)r Zresolve_conflictsr r Zget_revision_id� _write_shelf)r �
shelf_filer] r\ r r r � write_shelf* s
�zShelfCreator.write_shelfc C sZ t �� }|�|�� � | �|||�}|�|� |�|�D ]}|�|� q8|�|�� � d S rJ )�packZContainerSerialiser�writeZbeginr_ Z serialize�end)�clsra r r\ r] r[ r^ �bytesr r r r` 6 s
zShelfCreator._write_shelf)N)N)N)N)N)r r r �__doc__r r2 r: r; r3 r7 rI � staticmethodrN r6 r5 r4 rO rG r" r r_ rb � classmethodr` r r r r r : s.
.
�
r c @ sL e Zd ZdZdd� Zedd� �Zedd� �Zedd � �Z d
d � Z
d d
� Z dS )� UnshelverzUnshelve shelved changes.c C s || _ || _|| _|| _dS )z�Constructor.
:param tree: The tree to apply the changes to.
:param base_tree: The basis to apply the tranform to.
:param message: A message from the shelved transform.
N)rM � base_treer r] )r rM rl r r] r r r r D s zUnshelver.__init__c C s" t �� }|�| �� � t|�� �S rJ )rc ZContainerPushParserZ accept_bytes�read�iterZread_pending_records)ra �parserr r r � iter_recordsP s zUnshelver.iter_recordsc C sJ t | �\}}|d dkr t�t�|�}|�d�}|d urF|�d�|d< |S )Nr rX rW rV )�nextr rZ Zbdecode�get�decode)�recordsr> Zmetadata_bytesr^ r] r r r �parse_metadataV s
zUnshelver.parse_metadatac C sv | � |�}| �|�}|d }z|�|�}W n" tjyL |jj�|�}Y n0 t�|�}|� |� | ||||�
d��S )z�Create an Unshelver from a tree and a shelf file.
:param tree: The tree to apply shelved changes to.
:param shelf_file: A file-like object containing shelved changes.
:return: The Unshelver.
rU rW ) rp ru Z
revision_treer ZNoSuchRevisionInTree�branchZ
repositoryr r Z deserializerr )�klassrM ra rt r^ Zbase_revision_idrl rL r r r �from_tree_and_shelfa s
z Unshelver.from_tree_and_shelfc C s* | j �� }tj�| j|| j�}tj|_|S )z.Return a merger that can unshelve the changes.) r Zget_preview_tree�mergeZMergerZfrom_uncommittedrM rl Z Merge3MergerZ
merge_type)r r Zmergerr r r � make_mergert s
�zUnshelver.make_mergerc C s | j �� dS )z-Release all resources held by this Unshelver.N)r r" rT r r r r" | s zUnshelver.finalizeN) r r r rh r ri rp ru rj rx rz r" r r r r rk A s
rk c @ sj e Zd ZdZdd� Zdd� Zdd� Zdd � Zdd d �Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zd
S )� ShelfManagerz#Maintain a list of shelved changes.c C s || _ |�d�| _| j�� d S )NZshelf)rM Zclone� transportZ ensure_base)r rM r| r r r r � s zShelfManager.__init__c C s d| S )Nzshelf-%dr r r r r �get_shelf_filename� s z ShelfManager.get_shelf_filenamec C sB t �d�}g }|D ]*}|�|�}|d ur|�t|�d��� q|S )Nzshelf-([1-9][0-9]*)r )�re�compile�match�append�int�group)r � filenamesZmatcherZ shelf_ids�filenamer� r r r �
get_shelf_ids� s
zShelfManager.get_shelf_idsc C sB | � � }|du rd}n|d }| �|�}t| j�|�d�}||fS )z=Return a file object and id for a new set of shelved changes.Nr �wb)�
last_shelfr} �openr| �
local_abspath)r r� �
next_shelfr� ra r r r � new_shelf� s
zShelfManager.new_shelfNc C s< | � � \}}z|�||� W |�� n
|�� 0 |�� |S )z/Store the changes in a ShelfCreator on a shelf.)r� rb �closer )r Zcreatorr] r� ra r r r �shelve_changes� s zShelfManager.shelve_changesc
C s^ | � |�}zt| j�|�d�W S tyX } z"|jtjkr<� t|��W Y d}~n
d}~0 0 dS )z�Return the file associated with a shelf_id for reading.
:param shelf_id: The id of the shelf to retrive the file for.
�rbN)r} r� r| r� �IOError�errnoZENOENTr
)r r r� �er r r �
read_shelf� s
zShelfManager.read_shelfc C s2 | � |�}zt�| j|�W |�� S |�� 0 dS )zvReturn an unshelver for a given shelf_id.
:param shelf_id: The shelf id to return the unshelver for.
N)r� rk rx rM r� )r r ra r r r �
get_unshelver� s
�zShelfManager.get_unshelverc C s6 | � |�}zt�|�}W |�� n
|�� 0 t�|�S )z5Return the metadata associated with a given shelf_id.)r� rk rp r� ru )r r ra rt r r r � get_metadata� s
zShelfManager.get_metadatac C s | � |�}| j�|� dS )zrDelete the shelved changes for a given id.
:param shelf_id: id of the shelved changes to delete.
N)r} r| �delete)r r r� r r r � delete_shelf� s
zShelfManager.delete_shelfc C s t | �| j�d���}|S )z!Return a list of shelved changes.�.)�sortedr� r| Zlist_dir�r Zactiver r r �active_shelves� s zShelfManager.active_shelvesc C s$ | � � }t|�dkr |d S dS dS )z1Return the id of the last-created shelved change.r ���N)r� �lenr� r r r r� � s zShelfManager.last_shelf)N)r r r rh r r} r� r� r� r� r� r� r� r� r� r r r r r{ � s
r{ )Z
__future__r r� r~ r# r r �globalsr r r
r �objectr rk r{ r r r r �<module> s
@