File: //usr/lib/python3/dist-packages/mercurial/__pycache__/dirstatemap.cpython-310.pyc
o
�]Lb_ � @ s� d dl mZ d dlZddlmZ ddlmZmZmZm Z m
Z
m Z ddl m
ZmZ e�d�Ze�d�Ze jZedu r=ejZnejZd ZG d
d � d e�ZG d d
� d
e�Zedur`G dd
� d
e�ZdS dS )� )�absolute_importN� )�_)�error�pathutil�policy�pycompat�txnutil�util)�docket�v2�parsers�dirstatei���c @ s� e Zd ZdZdZdZdd� Zdd� Zd2dd�Zd d
� Z d d � Z
d
d� Z dd� Z d2dd�Z
d3dd�Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Z $ $ d4d%d&�Zd'd(� Zd5d*d+�Zed,d-� �Zd.d/� Zd0d1� Z dS )6�_dirstatemapcommonz�
Methods that are identical for both implementations of the dirstatemap
class, with and without Rust extensions enabled.
Nc C sF || _ || _|| _|| _|| _d| _d| _d | _d| _d | _ d | _
d S )Ns dirstate� F) �_use_dirstate_v2�_nodeconstants�_ui�_opener�_root� _filename�_nodelen�_parents�
_dirtyparents�_docket� _pendingmode)�self�ui�opener�root�
nodeconstants�use_dirstate_v2� r" �7/usr/lib/python3/dist-packages/mercurial/dirstatemap.py�__init__2 s
z_dirstatemapcommon.__init__c C s
| j dS )z5Loads the underlying data, if it's not already loadedN��_map�r r" r" r# �preloadA �
z_dirstatemapcommon.preloadc C s | j �||�S �N)r& �get)r �key�defaultr" r" r# r+ E � z_dirstatemapcommon.getc C �
t | j�S r* )�lenr& r' r" r" r# �__len__H �
z_dirstatemapcommon.__len__c C r/ r* ��iterr& r' r" r" r# �__iter__K r2 z_dirstatemapcommon.__iter__c C �
|| j v S r* r% )r r, r" r" r# � __contains__N r2 z _dirstatemapcommon.__contains__c C s
| j | S r* r% )r �itemr" r" r# � __getitem__Q r2 z _dirstatemapcommon.__getitem__c C � dS )z�incremente the dirstate counter if applicable
This might be a no-op for some subclass who deal with directory
tracking in a different way.
Nr" �r �filename� old_entryr" r" r# �
_dirs_incrY � z _dirstatemapcommon._dirs_incrFc C r: )z�decremente the dirstate counter if applicable
This might be a no-op for some subclass who deal with directory
tracking in a different way.
Nr" )r r< r= �remove_variantr" r" r# �
_dirs_decr` r? z _dirstatemapcommon._dirs_decrc C r: )z record updated state of an entryNr" �r �f�entryr" r" r# �_refresh_entryg r? z!_dirstatemapcommon._refresh_entryc C r: )z�add a new dirstate entry (or replace an unrelated one)
The fact it is actually new is the responsability of the caller
Nr" rB r" r" r# �
_insert_entryj r? z _dirstatemapcommon._insert_entryc C r: )z�remove any entry for file f
This should also drop associated copy information
The fact we actually need to drop it is the responsability of the callerNr" �r rC r" r" r# � _drop_entryp r? z _dirstatemapcommon._drop_entryc C s | | }|� � | �||� dS )z<record that the current state of the file on disk is unknownN)�set_possibly_dirtyrE )r r< rD r" r" r# rI y s z%_dirstatemapcommon.set_possibly_dirtyc C s< | | }|t @ }|�|||� | �||� | j�|d� dS )z$mark a file as back to a clean stateN)� rangemask� set_cleanrE �copymap�pop)r r< �mode�size�mtimerD r" r" r# rK s
z _dirstatemapcommon.set_cleanc C s� d}| � |�}|d u r | �|� tdd�}| �||� d}|S |js6| �||� |�� | �||� d}|S |�� | �||� |S )NFT)�
wc_tracked)r+ r> � DirstateItemrF �tracked� set_trackedrE rI )r r< �newrD r" r" r# rT � s&
� � � z _dirstatemapcommon.set_trackedc C sV | � |�}|du r dS | j|||j d� |js | j�|d� |�� | �||� dS )z4Mark a file as no longer tracked in the dirstate mapNF)r= r@ T)r+ rA �added�p2_inforL rM �
set_untrackedrE rB r" r" r# rX � s
z _dirstatemapcommon.set_untrackedTc
C s| | j �|d� |s!|s!|s!| j�|�}| �|� | j||d� dS | j�|�}| �||� t|||||d�} | �|| � dS )ah Set a entry to a given state, diregarding all previous state
This is to be used by the part of the dirstate API dedicated to
adjusting the dirstate after a update/merge.
note: calling this might result to no entry existing at all if the
dirstate map does not see any point at having one for this file
anymore.
N)r= )rQ �
p1_trackedrW �has_meaningful_mtime�parentfiledata) rL rM r& r+ rH rA r> rR rF )
r r< rQ rY rW rZ �has_meaningful_datar[ r= rD r" r" r# � reset_state� s
�z _dirstatemapcommon.reset_statec C sL t �| j| j| j�\}}| jd ur!| j|kr!|�� t�t d���|| _|S )Ns1 working directory state may be changed parallelly)
r �
trypendingr r r r �closer �Abortr )r �fprN r" r" r# �_opendirstatefile� s �z$_dirstatemapcommon._opendirstatefile���c
C sp z | � � �}|�|�W d � W S 1 sw Y W d S ty7 } z|jtjkr,� W Y d }~dS d }~ww )N� )rb �read�IOError�errno�ENOENT)r rO ra �errr" r" r# �_readdirstatefile� s
(� ��z$_dirstatemapcommon._readdirstatefilec C s2 | j s| js t�d��tj�| �� | j�| _ | j S )Ns' dirstate only has a docket in v2 format) r r r �ProgrammingError� docketmod�DirstateDocket�parserj r r' r" r" r# r � s �
�z_dirstatemapcommon.docketc s� �j }tj���� t|�|�}|�� � |r|�� d� �j� � |� |� |�
� � |� � |j rM|�� � d� �fdd� }|rJd|j }|�
||� n|� |�_d S )Nr c s �j �� �S r* )r �unlink)�_tr��
data_filenamer r" r# �<lambda> s z7_dirstatemapcommon.write_v2_no_append.<locals>.<lambda>s dirstate-v2-clean-r* )r rl rm �
with_new_uuid�parentsr0 rr �addr �write� serializer_ �uuid� addpostcloser ) r �tr�st�meta�packed�
old_docket�
new_docketro �categoryr" rq r# �write_v2_no_append� s$ �
z%_dirstatemapcommon.write_v2_no_appendc C s� | j sL| jr| jj| _ | j S | jd }| �|�}t|�}||kr5|d | j� || jd| j � f| _ | j S |dkrE| jj| jjf| _ | j S t �
t d���| j S )N� r s( working directory state appears damaged!) r r r ru r rj r0 r �nullidr r` r )r �read_lenr| �lr" r" r# ru s*
�
��� ��z_dirstatemapcommon.parentsr* �NF)FFFTTN)rc ) �__name__�
__module__� __qualname__�__doc__r& rL r$ r( r+ r1 r5 r7 r9 r> rA rE rF rH rI rK rT rX r] rb rj �propertyr r� ru r" r" r" r# r ' sB
�*
r c @ s� e Zd ZdZedd� �Zedd� �Zdd� Zdd � ZeZ d
d � Z
d d
� Z d.dd�Z dd� Z
dd� Zedd� �Zd/dd�Zd0dd�Zed d � �Zed d � �Zd d!� Zd"d#� Zed$d%� �Zed&d'� �Zd(d)� Zd*d+� Zd,d-� ZdS )1� dirstatemapa� Map encapsulating the dirstate's contents.
The dirstate contains the following state:
- `identity` is the identity of the dirstate file, which can be used to
detect when changes have occurred to the dirstate file.
- `parents` is a pair containing the parents of the working copy. The
parents are updated by calling `setparents`.
- the state map maps filenames to tuples of (state, mode, size, mtime),
where state is a single character representing 'normal', 'added',
'removed', or 'merged'. It is read by treating the dirstate as a
dict. File state is updated by calling various methods (see each
documentation for details):
- `reset_state`,
- `set_tracked`
- `set_untracked`
- `set_clean`
- `set_possibly_dirty`
- `copymap` maps destination filenames to their source filename.
The dirstate also provides the following views onto the state:
- `filefoldmap` is a dict mapping normalized filenames to the denormalized
form that they appear as in the dirstate.
- `dirfoldmap` is a dict mapping normalized directory names to the
denormalized form that they appear as in the dirstate.
c C s i | _ | �� | j S r* )r& re r' r" r" r# r& P s �dirstatemap._mapc C s i | _ | j | j S r* )rL r& r' r" r" r# rL V s �dirstatemap.copymapc C s\ | j �� | j�� | �| jj| jj� t�| d� t�| d� t�| d� t�| d� d S )N� _dirs� _alldirss filefoldmap�
dirfoldmap)r& �clearrL �
setparentsr r� r
�clearcachedpropertyr' r" r" r# r� \ s
�dirstatemap.clearc C � t �| j�S r* )r � iteritemsr& r' r" r" r# �itemse � �dirstatemap.itemsc c s0 � | � � D ]\}}||j|j|j|jfV qdS )z�
Return an iterator of (filename, state, mode, size, mtime) tuples
`all` is unused when Rust is not enabled
N)r� �staterN rO rP )r �allr< r8 r" r" r# �
debug_iterk s ���dirstatemap.debug_iterc C �
| j �� S r* )r& �keysr' r" r" r# r� t r2 �dirstatemap.keysFc C sZ ||f| _ d| _i }|r+t�| j�D ]\}}|jr*| j�|d �}|r&|||< |�� q|S )NT) r r r r� r& rW rL rM �drop_merge_data)r �p1�p2�fold_p2�copiesrC �s�sourcer" r" r# r� y s
��dirstatemap.setparentsc C s� t j�| j�| j��| _| jr | jj sd S | j�
| j� � �}n| � � }|s'd S t �
td�r7t�t|�d �| _| jrR| jj}| jj}t �tj�}|| j| j||� nt �tj�}|| j| j|�}| jsh| j|� | jj| _| jj| _| jj | _ d S )Ns dict_new_presized�G ) r
�filestat�frompathr �joinr �identityr r ry re rr rj � safehasattrr
�dict_new_presizedr0 r& ru �
tree_metadata�nogcr �parse_dirstaterL r r� r7 r9 r+ )r r| �pr} r� r" r" r# re � s0 �
zdirstatemap.readc C s^ | j rt�| j| j�\}}| �||||� nt�| j| j| �� �}|�|� |� � d| _
d S r� ) r r �
pack_dirstater& rL r� r
ru rw r_ r )r r{ r| r~ r} r" r" r# rw � s �
�dirstatemap.writec C � | j | jS r* �r& r� r' r" r" r# r� � � �dirstatemap.identityNc C sN |du s|j rd| jv r| j�|� |du r#d| jv r%| j�|� dS dS dS )z-incremente the dirstate counter if applicableN�_dirs�_alldirs)�removed�__dict__r� �addpathr� r; r" r" r# r> � s
�zdirstatemap._dirs_incrc C s� |dur d| j v r|js| j�|� d| j v r |s | j�|� n
|r-d| j v r-| j�|� d| j v r@t�|�}| j� |d� dS dS )z-decremente the dirstate counter if applicableNr� r� � filefoldmap)
r� r� r� �delpathr� r� r
�normcaser� rM )r r< r= r@ �normedr" r" r# rA � s �
�zdirstatemap._dirs_decrc C sn zt j}W n ty Y n
w || jtjtj�S i }tj}t� | j�D ]
\}}|j
s0||||�< q#d|d< |S )zeReturns a dictionary mapping normalized case paths to their
non-normalized versions.
� .) r
�make_file_foldmap�AttributeErrorr& r
� normcasespec�normcasefallbackr� r r� r� )r �makefilefoldmaprC r� �namer� r" r" r# r� � s
� � ��dirstatemap.filefoldmapc C s& i }t j}| jD ]}||||�< q|S r* )r
r� r� �r rC r� r� r" r" r# �
dirfoldmap� s
�dirstatemap.dirfoldmapc C r6 )zo
Returns True if the dirstate contains a tracked (not removed) file
in this directory.
)r� �r �dr" r" r# �
hastrackeddir �
�dirstatemap.hastrackeddirc C r6 )zn
Returns True if the dirstate contains a file (tracked or removed)
in this directory.
)r� r� r" r" r# �hasdir r� �dirstatemap.hasdirc C s t j| jdd�S )NT)� only_tracked�r �dirsr& r' r" r" r# r� s zdirstatemap._dirsc C r� r* r� r' r" r" r# r� s zdirstatemap._alldirsc C s |j s | j�|d � d S d S r* )� any_trackedr& rM rB r" r" r# rE s ��dirstatemap._refresh_entryc C s || j |< d S r* r% rB r" r" r# rF # r. �dirstatemap._insert_entryc C s | j �|d � | j�|d � d S r* )r& rM rL rG r" r" r# rH & � �dirstatemap._drop_entry�Fr* r� )r� r� r� r� �
propertycacher& rL r� r� r� r� r� r� re rw r� r> rA r� r� r� r� r� r� rE rF rH r" r" r" r# r� , s> #
6
r� c @ s� e Zd Zedd� �Zedd� �Zdd� Zdd� Zd d
� Z e Z
d d � Z d%dd�Z edd� �Z
dd� Zedd� �Zdd� Zdd� Zedd� �Zd d � Zd d � Zd d!� Zd"d#� Zd$S )&r� c C s� t j�| j�| j��| _| jr1| jj r | j�
| j� � �}nd}t j
�|| jj| jj�| _| jj}n t j
�| �� �\| _}|rF| jsF| j|� | jj| _| jj| _| jj| _| jS )z<
Fills the Dirstatemap when called.
rd )r
r� r� r r� r r� r r ry re rr �rustmod� DirstateMap�new_v2� data_sizer� r& ru �new_v1rj r r� r7 r9 r+ )r �dataru r" r" r# r& 1 s( ��
�
r� c C r� r* )r& rL r' r" r" r# rL R r) r� c C � | j �|�S )z�
Return an iterator of (filename, state, mode, size, mtime) tuples
`all`: also include with `state == b' '` dirstate tree nodes that
don't have an associated `DirstateItem`.
)r& r� )r r� r" r" r# r� V s r� c C sF | j �� | �| jj| jj� t�| d� t�| d� t�| d� d S )Nr� r� r� )r& r� r� r r� r
r� r' r" r" r# r� ` s
� r� c C r� r* )r& r� r' r" r" r# r� i r2 r� c C r/ r* r3 r' r" r" r# r� o r2 r� Fc
C sx ||f| _ d| _i }|r:dd� | j�� D �}| j}|D ] }|�|�}| j�|d �} | r/| ||< |�� |�||� q|S )NTc S s g | ] \}}|j r|�qS r" )rW )�.0rC r� r" r" r# �
<listcomp>~ s
��z*dirstatemap.setparents.<locals>.<listcomp>) r r r& r� r+ rL rM r� �set_dirstate_item)
r r� r� r� r� �files_with_p2_info�rust_maprC �er� r" r" r# r� t s
�
r� c C r� r* r� r' r" r" r# r� � r� r� c
C sV | j s | �� \}}| j�||�}|�|� |�� d| _d S | jjd u}| j� |�\}}}|r�| j} | �
� }
|r?|� |
| j � | �
|
d��.} | �| j � | �� | j ksUJ �| �|�} | d url| t|�kslJ | t|�f��W d � n1 svw Y | j t|�7 _ | �� | _|| _|�| �� � |�� n| �||||� t�| d� d| _d S )NFs r+bs _map)r ru r& �write_v1rw r_ r r ry �write_v2rr rv r� r �seek�tellr0 r� rx r� r
r� )
r r{ r| r� r� r~ �
can_appendr} �appendr rr ra �writtenr" r" r# rw � s<
� �
r� c C r� )zmReturns a dictionary mapping normalized case paths to their
non-normalized versions.
)r& �filefoldmapasdictr' r" r" r# r� � r� r� c C r� r* )r& r� r� r" r" r# r� � r� r� c C r� r* )r& r� r� r" r" r# r� � r� r� c C s* i }t j}| j�� D ]}||||�< q
|S r* )r
r� r& � tracked_dirsr� r" r" r# r� � s
r� c C s( |j s | j�|� d S | j�||� d S r* )r� r& �drop_item_and_copy_source�addfilerB r" r" r# rE � s r� c C s | j �||� d S r* )r& r� rB r" r" r# rF � s r� c C s | j �|� d S r* )r& r� rG r" r" r# rH � s r� c C s t |t�sJ �| j�||� d S r* )�
isinstancerR r& r� )r r, �valuer" r" r# � __setitem__� r� zdirstatemap.__setitem__Nr� )r� r� r� r� r& r� rL r� r� r� r� r� r� r� rw r� r� r� r� rE rF rH r� r" r" r" r# r� - s0
%
)�
__future__r rg �i18nr � r r r r r r
�
dirstateutilsr rl r � importmodr
�
importrustr� r� rR rJ �objectr r� r" r" r" r# �<module> s*
�