File: //usr/lib/python3/dist-packages/mercurial/upgrade_utils/__pycache__/engine.cpython-310.pyc
o
�]LbvW � @ s
d dl mZ d dlZd dlZddlmZ ddlmZ ddlm Z m
Z
m Z m Z m
Z
mZmZmZmZmZmZ ddlmZmZmZmZ dd lmZ d
d � Z d d
� Z dd� Z dZ dZ!dZ"e#e e!e"g�Z$dd� Z%dd� Z&dd� Z'dd� Z(dd � Z)d d � Z*d d � Z+d!d"� Z,d#d$� Z-dS )%� )�absolute_importN� )�_)�getattr) � changelog�error�filelog�manifest�metadata�pycompat� requirements�scmutil�store�util�vfs)� constants�flagutil�nodemap�sidedata� )�actionsc C sT | j �dd�}tjp | }|s#| jtjtjtjft �
| |�t j dd� t�
| |j�S )Ns experimentals worker.repository-upgradeT)�replace)�ui�
configboolr � iswindows�register_sidedata_computer� revlogconst�KIND_CHANGELOG� sidedatamod�SD_FILESr
�_get_worker_sidedata_adderr �REVIDX_HASCOPIESINFO�get_sidedata_helpers�_wanted_sidedata)�srcrepo�dstrepo�use_w�
sequential� r( �@/usr/lib/python3/dist-packages/mercurial/upgrade_utils/engine.pyr" % s
�r" c C s� |t j@ r t�| j�S |t j@ r(d}d|v r |�dd�d }tj| j| j|d�S |�dd�d � dd�}t
|�dk rGt d�}||; }t �
|��|d }t�| j|�S ) z]Obtain a revlog from a repo path.
An instance of the appropriate class is returned.
� � /r r )�tree� .r s) cannot recognize revlog from filename: %s)r �FILEFLAGS_CHANGELOGr �svfs�FILEFLAGS_MANIFESTLOG�rsplitr �manifestrevlog�
nodeconstants�split�lenr r �Abortr )�repo�rl_type�path�mandir� path_part�msgr( r( r) �_revlogfrompath4 s
�
r= c
C s� t |d|�}t|||�}t |d|�}|j}|j}|�|j�}|�|j�} |�|j�}
|�|j�} ||jd�� W d � n1 sAw Y t�|| � |j�|j�} | r[t�|
| � |t j
@ rx|j j �
|� | rz|j j �
|dd� d � dS dS dS )a copy all relevant files for `oldrl` into `destrepo` store
Files are copied "as is" without any transformation. The copy is performed
without extra checks. Callers are responsible for making sure the copied
content is compatible with format of the destination repository.
�_revlog� wN���s .d)r r= �opener�join�
_indexfile� _datafiler �copyfile�existsr �FILEFLAGS_FILELOGr/ �fncache�add)
�tr�destrepo�oldrlr8 �
unencodedname�newrl�oldvfs�newvfs�oldindex�newindex�olddata�newdata�copydatar( r( r) � _copyrevlogM s, �
�rV s changelogs manifests all-filelogsc C s: |t j@ r t| v S |t j@ rt| v S |t j@ sJ �t| v S )z�check if a revlog is selected for cloning.
In other words, are there any updates which need to be done on revlog
or it can be blindly copied.
The store entry is checked against the passed filter)r r. �UPGRADE_CHANGELOGr0 �UPGRADE_MANIFESTrG �UPGRADE_FILELOGS)� revlogfilterr8 r( r( r) � matchrevlogv s
r[ c C s� d} t |j|�r*| �td�t|�|f � t|||�} |j|| ||j|j|d� | S td�}
| �|
|t|�f � t |||||� t|||�} | S )z%returns the new revlog object createdNs cloning %d revisions from %s
)�
addrevisioncb�
deltareuse�forcedeltabothparents�sidedata_helperss+ blindly copying %s containing %i revisions
)
r[ �revlogs_to_process�noter r5 r= �clone�delta_reuse_mode�force_re_delta_both_parentsrV ) r r% rJ �
old_revlogr8 � unencoded�
upgrade_opr_ �oncopiedrevisionrN r< r( r( r) �_perform_clone� s( � �� ri c ' sP d}d}d}d}d} d}
d} d} d}
d}d}d}d}d}d}d}d}d}t |j�� �}i }i }i }|D ]�\}} } |tj@ s>q3d} | | v rEq3t||| �} | jddddd�} || d pZd7 }| d pad}!| d pgd}"||!7 }||"7 }|tj@ r�|| f|| < |t| �7 }||!7 }||"7 }q3|tj@ r�|| f|| < |d7 }|t| �7 }||!7 }||"7 }q3|tj @ r�|| f|| < | d7 } |
t| �7 }
| |!7 } | |"7 } q3t
� d � q3|s�d
S | � t
d �||
||f � | � t
d �t�|�t�|�f � d
� � fd
d�}#t||�}$| � t
d�| |
t�| �t�| �f � |jjt
d�|
d�� t|�� �D ]"\} \}}%t| |||%|| ||$|#� }&|&jdd�} |
| d �p=d7 }
�q | � t
d�|
| t�|
| �f � | � t
d�||t�|�t�|�f � � �rl� �� |jjt
d�|d�� t|�� �D ]"\} \}}%t| |||%|| ||$|#� }&|&jdd�} || d �p�d7 }�q|| � t
d�||t�|| �f � | � t
d�|t�|�t�|�f � � �rɈ �� |jjt
d�|d�� t|�� �D ]"\} \}}%t| |||%|| ||$|#� }&|&jdd�} || d �p�d7 }�qو �� | � t
d�|t�|| �f � |
| | }| � t
d�|t�|| �f � d
S )z Copy revlogs between 2 repos.r )s undo.backup.00changelog.is undo.backup.00manifest.iT)�exclusivefiles�revisionscount� trackedsize�
storedsizes revisionscounts
storedsizes trackedsizer s unknown revlog typeNsP migrating %d total revisions (%d in filelogs, %d in manifests, %d in changelog)
s' migrating %s in store; %s tracked data
c s � � � d S )N)� increment)�rl�rev�node��progressr( r) rh s z'_clonerevlogs.<locals>.oncopiedrevisionsM migrating %d filelogs containing %d revisions (%s in store; %s tracked data)
s file revisions)�total)rm sO finished migrating %d filelog revisions across %d filelogs; change in size: %s
sN migrating %d manifests containing %d revisions (%s in store; %s tracked data)
s manifest revisionssQ finished migrating %d manifest revisions across %d manifests; change in size: %s
sK migrating changelog containing %d revisions (%s in store; %s tracked data)
s changelog revisionss> finished migrating %d changelog revisions; change in size: %s
sF finished migrating %d total revisions; total change in store size: %s
)�listr �walk�FILEFLAGS_REVLOG_MAINr= � storageinfor. r5 r0 rG r �ProgrammingError�statusr r � bytecountr" r � makeprogress�sorted�itemsri �complete)'r r$ r% rJ rg �revcount�srcsize�
srcrawsize�dstsize�fcount� frevcount�fsrcsize�frawsize�fdstsize�mcount� mrevcount�msrcsize�mrawsize�mdstsize� crevcount�csrcsize�crawsize�cdstsize� alldatafiles� manifests�
changelogs�filelogsr8 rf �size� skip_undoro �info�datasize�rawsizerh r_ rL rN r( rr r) �
_clonerevlogs� s|
�
�
����
���� � ��� ���� �� ��� ���� �� ��� ���r� c c s` � t | jjjddd��D ]!\}}}t�|�durq |�d�r q |tjkr%q |dv r*q |V q dS )zOyields files which should be copied to destination after revlogs
are clonedr* T)�statNs undo)s locks fncache)r} r r �readdir� revlog_type�
startswithr� �S_IFREG)r$ r9 �kind�str( r( r) � _files_to_copy_post_revlog_clone� s �
�r� c C s@ |j rt�| j|�d�� n| jjddd� t�|j| j� dS )a� Replace the stores after current repository is upgraded
Creates a backup of current repository store at backup path
Replaces upgraded store files in current repo from upgraded one
Arguments:
currentrepo: repo object of current repository
upgradedrepo: repo object of the upgraded data
backupvfs: vfs object for the backup path
upgrade_op: upgrade operation object
to be used to decide what all is upgraded
s storeT)�forciblyN)� backup_storer �rename�spathrB r �rmtree)� currentrepo� upgradedrepo� backupvfsrg r( r( r) �_replacestores� s r� c C s dS )z�Hook point for extensions to perform additional actions during upgrade.
This function is called after revlogs and store files have been copied but
before the new store is swapped into the original location.
Nr( )r r$ r% r r( r( r) �finishdatamigration� s r� c C s8 |� � sJ �|� � s J �d}d}| �td�� tj|jv r3| �td�� t| ||dd� |j�tj� tj|jv rO| �td�� t| ||dd� |j�tj� tj|jv rl| �td�� t | ||dd � |j�tj� n tj|jv r�| �td
�� t | ||d d � |j�tj� |js�|js�dS |j
r�| �td �� t � ||j
� |S t|j�d
kr�d|jv r�|js�| �td�� |�d��"}|�� }|j}tj||dd� tj||jjjdd� W d � n1 s�w Y t � ||j
� |S t|j�d
k�rDdd� |jD ��rD|j�sD| �td�� |�d�� }|�� }|j}t�|||� t�|||jjj� W d � n 1 �s6w Y t � ||j
� |S |�d��}t| ||||� W d � n 1 �s]w Y t|�D ]#} |j�td�| � |j j � | �}
|j j � | �} t j |
| dd� �qft!| ||t"� | �td�� |j#�r�t$j%d|j&d�}t'�(|�}t � |j(� d�|� d�� | �td�� t � ||j"d hB � | �td �� |j#�r�| �td �| � | �td �� t �)� } t*||||� t �)� | }
| �td �|
� | �td!�� t � ||j
� |j#�r|du�sJ �|�+d"� |S )#ay Do the low-level work of upgrading a repository.
The upgrade is effectively performed as a copy between a source
repository and a temporary destination repository.
The source repository is unmodified for as long as possible so the
upgrade can abort at any time without causing loss of service for
readers and without corrupting the source repository.
NsP (it is safe to interrupt this process any time before data migration completes)
s! upgrading to dirstate-v2 from v1
s v1� v2s# downgrading from dirstate-v2 to v1
s" create dirstate-tracked-hint file
T)rI s" remove dirstate-tracked-hint file
Fs" upgrading repository requirements
r � persistent-nodemaps7 upgrading repository to use persistent nodemap feature
s upgrade)�forcec S s g | ] }|j d kr|�qS )r� )�name)�.0�xr( r( r) �
<listcomp> s
�zupgrade.<locals>.<listcomp>s= downgrading repository to not use persistent nodemap feature
s copying %s
)�copystats. data fully upgraded in a temporary repository
� upgradebackup.��prefix�dir� requiress\ marking source repository as being upgraded; clients will be unable to read from repository
s upgradeinprogresss* starting in-place swap of repository data
�' replaced files will be backed up at %s
s replacing store...
sC store replacement complete; repository was inconsistent for %0.1fs
sB finalizing requirements file and making repository readable again
s
store/lock),� currentwlockrz r �upgrade_actions�
dirstatev2�upgrade_dirstate�remove�removed_actions�dirstatetrackedkey�upgrade_tracked_hint�requirements_onlyr
�writereporequirements�new_requirementsr5 �upgrade_actions_names� transaction�
unfilteredr r �persist_nodemap� manifestlog�
_rootstorer> �delete_nodemapr� r� r r �rawvfsrB r rE r� r r� r �mkdtempr9 �vfsmodr �timerr� �unlink)r r$ r% rg �
backuppathr� rJ �unfi�cl�p�src�dst�tstart�elapsedr( r( r) �upgrade� s�
�� �
�� � � k����� � �V �� �
�
��� �
� ���
��
r� c
C sJ |j rQtjd|jd�}| �td�| � t�|�}t� |j�
d�|�
d�� zt� |j�
d�|�
d�� W n t t fyP } z
|j
t
jkrF� W Y d }~nd }~ww |jj|dkks[J �|jj�� |dk|j_|jj|jj_d|j_z|j�d� W n t t fy� } z
|j
t
jkr�� W Y d }~nd }~ww |j�d � t�||j� d S )Nr� r� r� r� s dirstater� T)r� r r� r9 rz r r� r r rE rB �IOError�OSError�errno�ENOENT�dirstate�_use_dirstate_v2�_map�preload�_dirtyr� �writer
r� r� )r r$ rg �old�newr� r� �er( r( r) r� | sD �
�� ��� ��� r� c C sH |rd|j _d|j _d|j _|j �d � |s|j �� t�||j� d S )NT) r� �_use_tracked_hintr� �_dirty_tracked_setr� �delete_tracked_hintr
r� r� )r r$ rg rI r( r( r) r� � s
r� ).�
__future__r r� r� �i18nr r r � r r r r r
r r
r r r r� � revlogutilsr r r r r r r r� r" r= rV rW rX rY � frozenset�UPGRADE_ALL_REVLOGSr[ ri r� r� r� r� r� r� r� r( r( r( r) �<module> s8 4
�# o 4 &