File: //kunden/lib/python3/dist-packages/breezy/__pycache__/shelf_ui.cpython-39.pyc
a
�*�^iO � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlmZ ddlm Z m
Z
m Z m Z m
Z
mZmZmZmZmZmZmZ ddlmZ G dd� de�ZG d d
� d
e�ZG d d � d e�ZG d
d� de�Z G dd� de�Z dS )� )�absolute_importN)�BytesIO� ) �builtins�cleanup�delta�diff�errors�osutils�patches�shelf�textfile�trace�ui� workingtree)�gettextc @ s e Zd ZdZdS )� UseEditorz)Use an editor instead of selecting hunks.N)�__name__�
__module__� __qualname__�__doc__� r r �1/usr/lib/python3/dist-packages/breezy/shelf_ui.pyr + s r c @ sv e Zd Zed�ed�ed�ed�ed�ed�ed�ed�d �Zd
Zd d � Zd
d� Zdd� Zdd� Z dd� Z
dd� Z dS )�
ShelfReporterz Shelve adding file "%(path)s"?zShelve binary changes?z0Shelve changing "%s" from %(other)s to %(this)s?z Shelve removing file "%(path)s"?zShelve %d change(s)?zShelve?zDShelve changing target of "%(path)s" from "%(other)s" to "%(this)s"?z*Shelve renaming "%(other)s" => "%(this)s"?�zadd file�binary� change kindz delete file�final�hunk�
modify target�renameFc C s t �� | _d S �N)r �_ChangeReporter�delta_reporter��selfr r r �__init__@ s zShelfReporter.__init__c C s t �d� dS )z.Report that no changes were selected to apply.zNo changes to shelve.N)r Zwarningr$ r r r �
no_changesC s zShelfReporter.no_changesc C s t �td�| � dS )z&Report the id changes were shelved to.z Changes shelved with id "%d".N�r �noter )r% �shelf_idr r r �
shelved_idG s zShelfReporter.shelved_idc C s t �td�� dS )z/Report that changes were made without shelving.zSelected changes destroyed.Nr( r$ r r r �changes_destroyedK s z ShelfReporter.changes_destroyedc C s( t �td�� |�� }t�|| j� dS )z&Report the changes that were selected.zSelected changes:N)r r) r Z iter_changesr Zreport_changesr# )r% � transformZchangesr r r �selected_changesO s z ShelfReporter.selected_changesc C s� |d dkr |d |d d�}nX|d dkrF|d |d |d d�}n2|d d krl|d |d |d d�}n d
|d i}| j |d | }|S ) z+Determine the prompt for a change to apply.r r � � )�this�otherr � )�pathr2 r1 r r4 )�vocab)r% �changeZvals�promptr r r �
prompt_changeU s zShelfReporter.prompt_changeN) r r r r r5 � invert_diffr& r' r+ r, r. r8 r r r r r / s �
r c @ sN e Zd Zed�ed�ed�ed�ed�ed�ed�ed�d �Zd
Zd d � Zd
S )�
ApplyReporterzDelete file "%(path)s"?zApply binary changes?z-Change "%(path)s" from %(this)s to %(other)s?zAdd file "%(path)s"?zApply %d change(s)?z
Apply change?z;Change target of "%(path)s" from "%(this)s" to "%(other)s"?z!Rename "%(this)s" => "%(other)s"?r Tc C s d S r! r r$ r r r r, s s z ApplyReporter.changes_destroyedN)r r r r r5 r9 r, r r r r r: c s � r: c @ sx e Zd ZdZddd�Zeddd��Zdd � Zd
d � Zd d d
�Z dd� Z
d dd�Z dd� Z dd� Z
dd� Zedd� �ZdS ) �Shelverz3Interactively shelve the changes in a working tree.NFc C s� || _ || _|| _| jdu r$tj| _| du r4|�� } | | _|| _|| _|| _ || _
|| _ |
du rft � }
|
| _
| j j�� } | �||�| _| j �� dS )a� Constructor.
:param work_tree: The working tree to shelve changes from.
:param target_tree: The "unchanged" / old tree to compare the
work_tree to.
:param auto: If True, shelve each possible change.
:param auto_apply: If True, shelve changes with no final prompt.
:param file_list: If supplied, only files in this list may be shelved.
:param message: The message to associate with the shelved changes.
:param destroy: Change the working tree without storing the shelved
changes.
:param manager: The shelf manager to use.
:param reporter: Object for reporting changes to user.
N)� work_tree� target_tree� diff_writer�sys�stdout�get_shelf_manager�manager�auto�
auto_apply� file_list�message�destroyr �reporter�branchZ
get_configZget_change_editor�
change_editor�lock_tree_write) r% r< r= r> rC rD rE rF rG rB rH Zconfigr r r r& z s&
zShelver.__init__c
s� � du rd� n|r$� fdd�|D �}t j�� �\}} |�� �B t�d||j|�}
|�|�} | ||
|||| ||�W d � S 1 s�0 Y dS )aa Create a shelver from commandline arguments.
The returned shelver wil have a work_tree that is locked and should
be unlocked.
:param revision: RevisionSpec of the revision to compare to.
:param all: If True, shelve all changes without prompting.
:param file_list: If supplied, only files in this list may be shelved.
:param message: The message to associate with the shelved changes.
:param directory: The directory containing the working tree.
:param destroy: Change the working tree without storing the shelved
changes.
N�.c s g | ]}t �� |��qS r )r
Zpathjoin)�.0�f�� directoryr r �
<listcomp>� � z%Shelver.from_args.<locals>.<listcomp>Zshelf2)r � WorkingTree�open_containingrK r Z_get_one_revision_treerI Zsafe_relpath_files) �klassr> Zrevision�allrE rF rP rG �treer4 r= �filesr rO r � from_args� s
�
�zShelver.from_argsc C sh t �| j| j| j�}t�� | _d}�z(|�� D ]�}|d dkr�z|| � ||d �7 }W q� t
j y� | � | j
jd �r�|d7 }|�|d � Y q�0 q.| � | j
�|��r.|�|� |d7 }q.|dk�r,| j
�|j� | js�| � | j
jd | ��r6| j�r|�� | j
�� n | j�|| j�}| j
�|� n
| j
� � W t � | j� |� � nt � | j� |� � 0 dS )z!Interactively shelve the changes.r z modify textr r r N) r Z ShelfCreatorr< r= rE �tempfileZmkdtempZtempdirZiter_shelvable�handle_modify_textr Z
BinaryFile� prompt_boolrH r5 Zshelve_content_changer8 Z
shelve_changer. Zwork_transformrD rG r- r, rB Zshelve_changesrF r+ r' �shutilZrmtree�finalize)r% �creatorZchanges_shelvedr6 r* r r r �run� sH �
�
� �
� z Shelver.runc C s" | j d ur| j �� | j�� d S r! )rJ Zfinishr< �unlockr$ r r r r^ � s
zShelver.finalizec C sv t � }|r| j}| j}n | j}| j}|�|�}|�|�}t�� }tj||||d�} | �||dd�}
|�d� t �
|�S )a& Return a parsed version of a file's patch.
:param file_id: The id of the file to generate a patch for.
:param invert: If True, provide an inverted patch (insertions displayed
as removals, removals displayed as insertions).
:return: A patches.Patch.
��
path_encoding�filer ) r r< r= �id2pathr
Zget_terminal_encodingr ZDiffText�seekr Z parse_patch) r% �file_id�invertZ diff_file�old_tree�new_treeZold_path�new_pathrc Z text_differZpatchr r r �get_parsed_patch� s
�
zShelver.get_parsed_patchc C s t jj|||d�S )N)�default)r �
ui_factoryZchoose)r% rF �choicesrm r r r r7 � s zShelver.promptc C s� | j r
dS d}d}|r&|d7 }|d7 }|d7 }|d7 }| �||d�}|d u rRd
}n|| }|d krfdS |dkrx|rxt�n|d kr�d| _ dS |d
kr�t�� �ndS d S )z�Prompt the user with a yes/no question.
This may be overridden by self.auto. It may also *set* self.auto. It
may also raise UserAbort.
:param question: The question to ask the user.
:return: True or False
TZynz&yes
&No�ez
&edit manuallyZfqz
&finish
&quitr N�n�yrN �qF)rC r7 r r Z UserAbort)r% Zquestion� allow_editorZalternatives_charsZ alternativesZchoice�charr r r r\ s.
zShelver.prompt_boolc C sl | j �|�}| j �||�}z| �|||�\}}W n" tyR | �||�\}}Y n0 |dkrh|�||� |S )z�Handle modified text, by using hunk selection or file editing.
:param creator: A ShelfCreator.
:param file_id: The id of the file that was modified.
:return: The number of changes.
r )r< re �get_file_lines�
_select_hunksr �
_edit_fileZ shelve_lines)r% r_ rg r4 �work_tree_lines�lines� change_countr r r r[ % s � zShelver.handle_modify_textc C s& | j jr|}n| j�|�}| j�|�}t�|� t�|� | �|| j j�}g }| js�d}| j �
|� � � |j D ]l} | j �
| �
� � | j| j jd | jdud�}
| j js�|
}
|
r�| j|7 _|�| � qn|| j| j 8 }qntj�� | j jr�t|�} nt|j �t|� } t�||�} t| �}
|
| fS )a� Provide diff hunk selection for modified text.
If self.reporter.invert_diff is True, the diff is inverted so that
insertions are displayed as removals and vice versa.
:param creator: a ShelfCreator
:param file_id: The id of the file to shelve.
:param work_tree_lines: Line contents of the file in the working tree.
:return: number of shelved hunks.
r r N)rt ) rH r9 r= re rv r
Zcheck_text_linesrl rC r> �writeZ
get_headerZhunksZas_bytesr\ r5 rJ Zmod_pos�appendZ mod_rangeZ
orig_ranger? r@ �flush�lenr Ziter_patched_from_hunks�list)r% r_ rg ry Z target_linesr4 ZparsedZ final_hunks�offsetr Zselectedr{ Zpatchedrz r r r rw 7 sB
��
�zShelver._select_hunksc C s8 t �| j�| jj�|�| jj�|���}|| �||�fS )a:
:param file_id: id of the file to edit.
:param work_tree_lines: Line contents of the file in the working tree.
:return: (lines, change_region_count), where lines is the new line
content of the file, and change_region_count is the number of
changed regions.
)r
Z split_linesrJ Z edit_fileri re rj �_count_changed_regions)r% rg ry rz r r r rx d s
�zShelver._edit_filec C s" t �d | |�}|�� }t|�d S )Nr0 )� patiencediffZPatienceSequenceMatcherZget_matching_blocksr )Z old_linesZ new_linesZmatcherZblocksr r r r� q s
�z Shelver._count_changed_regions)NFFNNFNN)NFNNNF)F)F)r r r r r& � classmethodrY r` r^ rl r7 r\ r[ rw rx � staticmethodr� r r r r r; w s& �
% � %
#-
r; c @ s@ e Zd ZdZeddd��Zdd d
�Zd d � Zd
d� Zdd� Z dS )� Unshelverz%Unshelve changes into a working tree.N�applyrL c C s� t j�|�\}}|�� z�|�� }|durVz t|�}W qt tyR t�|��Y qt0 n |� � }|du rtt
� t d���d}d} d}
d} |dkr�d}d} n8|dkr�d}d} d} n"|dkr�d}d}
n|dkr�d}d} W n |�
� � Y n0 | ||||| |
| |�S ) a� Create an unshelver from commandline arguments.
The returned shelver will have a tree that is locked and should
be unlocked.
:param shelf_id: Integer id of the shelf, as a string.
:param action: action to perform. May be 'apply', 'dry-run',
'delete', 'preview'.
:param directory: The directory to unshelve changes into.
:param write_diff_to: See Unshelver.__init__().
NzNo changes are shelved.TFzdry-runZpreviewz delete-onlyZkeep)r rS rT rK rA �int�
ValueErrorr ZInvalidShelfIdZ
last_shelfr ZBzrCommandErrorr ra ) rU r* �actionrP �
write_diff_torW r4 rB �
apply_changes� delete_shelf�
read_shelf� show_diffr r r rY | sJ � �zUnshelver.from_argsTFc C s< || _ |�� }|| _|| _|| _|| _|| _|| _|| _dS )a� Constructor.
:param tree: The working tree to unshelve into.
:param manager: The ShelveManager containing the shelved changes.
:param shelf_id:
:param apply_changes: If True, apply the shelved changes to the
working tree.
:param delete_shelf: If True, delete the changes from the shelf.
:param read_shelf: If True, read the changes from the shelf.
:param show_diff: If True, show the diff that would result from
unshelving the changes.
:param write_diff_to: A file-like object where the diff will be
written to. If None, ui.ui_factory.make_output_stream() will
be used.
N) rW rA rB r* r� r� r� r� r� ) r% rW rB r* r� r� r� r� r� r r r r& � s zUnshelver.__init__c C s� t �� ��}|�| j�� � | jr�t�td�| j � | j
� | j �}|� |j
� |jdurlt�td�|j � t�� }|�� }||_| jr�|�� n | jr�| �|� n
| �|� | jr�| j
�| j � t�td�| j � W d � n1 s�0 Y dS )z!Perform the unshelving operation.zUsing changes with id "%d".Nz Message: %sz Deleted changes with id "%d".)r � ExitStack�
enter_contextrW rK r� r r) r r* rB Z
get_unshelver�callbackr^ rF r r" � make_merger�change_reporterr� Zdo_merger� �
write_diff� show_changesr� )r% Z
exit_stackZ unshelverr� �mergerr r r r` � s.
�
�z
Unshelver.runc C s\ |� � }|�� }|�� }| jdu r2tjjdd�| _t�� }t j
|j || j|d� |� � dS )z2Write this operation's diff to self.write_diff_to.N�exact)Z
encoding_typerb )
r� �make_preview_transformZget_preview_treer� r rn Zmake_output_streamr
Zget_diff_header_encodingr Zshow_diff_treesZ this_treer^ )r% r� � tree_merger�ttrj rc r r r r� � s
��zUnshelver.write_diffc C s |� � }|�� }|�� dS )z/Show the changes that this operation specifies.N)r� r� r^ )r% r� r� r� r r r r� � s zUnshelver.show_changes)Nr� rL N)TTTFN)
r r r r r� rY r& r` r� r� r r r r r� y s � 2 �
r� ) Z
__future__r r� r] r? rZ �ior � r r r r r r
r r r
r r r Zi18nr � Exceptionr �objectr r: r; r� r r r r �<module> s 8 4