/
lib64
/
python3.9
/
site-packages
/
hgext
/
fastannotate
/
__pycache__
/
Upload File
HOME
a �+�b/{ � @ sT d dl mZ d dlZd dlZd dlZd dlmZ d dlmZm Z m Z d dl m Z m Z mZ d dlmZmZmZmZmZmZmZ d dlmZmZ dd l mZ m Z ej d d � �Z!d&d d�Z"dd� Z#dd� Z$e� ej%�Z&d'dd�Z'dd� Z(dd� Z)e)ej*�Z+G dd� de,�Z-e-� Z*G d d � d e,�Z.d d � Z/G d d!� d!e,�Z0ej1e*dfd"d#��Z2d(d$d%�Z3dS ))� )�absolute_importN)�_)�getattr�open�setattr)�bin�hex�short)�error�linelog�lock�mdiff�pycompat�scmutil�util)�hashutil� stringutil� )r �revmapc C s | � |�S �N)�file)�repo�path� r �@/usr/lib64/python3.9/site-packages/hgext/fastannotate/context.py�_getflog- s r Tc sN � j � �� }|s$� fdd�|D �}|D ] }d|jvr(t� j|�� �|_q(|S )Nc s g | ]}|� � � � � kr|�qS r �r )�.0�p��fr r � <listcomp>= � z _parents.<locals>.<listcomp>�_filelog)� _changeid�parents�__dict__r �_repor r# )r �follow�plr r r r �_parents3 s r* c sD � � � }|�d�}|r(|�d�s(|d7 }� fdd�t�|�D �|fS )N� r c s g | ] }� |f�qS r r )r �i��fctxr r r! O r" z _decorate.<locals>.<listcomp>)�data�count�endswithr �xrange)r. �textZ linecountr r- r � _decorateJ s r4 c C s> |D ]4\\}}}}}|dkr| d ||� |d ||�< q|S )N� =r r )�parent�child�blocks�a1�a2�b1�b2�tr r r �_pairU s r> Fc C s� |r"t |t�s"|dur"t| |�}n| | }|du r\|�� du rH|| }q�| j||�� d�}n>|| }|dkrv|�� }n|�� }||�� kr�||_| | |_|S )a2 (repo, str, str) -> fctx get the filectx object from repo, rev, path, in an efficient way. if resolverev is True, "rev" is a revision specified by the revset language, otherwise "rev" is a nodeid, or a revision number that can be consumed by repo.__getitem__. if adjustctx is not None, the returned fctx will point to a changeset that introduces the change (last modified the file). if adjustctx is 'linkrev', trust the linkrev and do not adjust it. this is noticeably faster for big repos but is incorrect for some cases. N)Zchangeid� linkrev) � isinstance�int� _revsingle�revZfilectx�linkrev�introrevr$ Z _changectx)r rC r � resolverev� adjustctx�ctxr. rE r r r � resolvefctxc s rI c C s$ | � dd�� dd�� dd�� dd�S ) Ns .hg/s .hg.hg/s .l/s .l.hg/s .m/s .m.hg/s .lock/s .lock.hg/)�replacer r r r � encodedir� s ����rK c s: t �t� fdd�tjjD ���}tt�|�� � �d d� S )Nc 3 s | ]}|t � |�fV qd S r )r )r �k��diffoptsr r � <genexpr>� r" z hashdiffopts.<locals>.<genexpr>� ) r �pprint�sortedr rN �defaultsr r Zsha1�digest)rN Z diffoptstrr rM r � hashdiffopts� s �rU c @ s2 e Zd ZdZdddd�Zdd� Zejdd� �ZdS ) � annotateoptsz�like mercurial.mdiff.diffopts, but is for annotate followrename: follow renames, like "hg annotate -f" followmerge: follow p2 of a merge changeset, otherwise p2 is ignored NT)s diffoptss followrenames followmergec K s8 t �|�}t �| j�D ] \}}t| ||�||�� qd S r )r Z byteskwargs� iteritemsrS r �get)�self�optsrL �vr r r �__init__� s zannotateopts.__init__c C sP d}| j s|d7 }| js |d7 }| jdurHt| j�}|tkrH|d| 7 }|pNdS )z?represent opts in a short string, suitable for a directory namer" s r0s m0N� is default)� followrename� followmergerN rU �_defaultdiffopthash)rY �resultZ diffopthashr r r �shortstr� s zannotateopts.shortstr) �__name__� __module__� __qualname__�__doc__rS r\ r � propertycacherb r r r r rV � s �rV c @ s� e Zd ZdZdd� Zedd� �Zedd� �Zdd � ZeZ d d � Z ed d � �Z d/dd�Z d0dd�Z dd� Zd1dd�Zdd� Zdd � Zd d � Zd2d d �Zd!d"� Zed3d#d$��Zd4d%d&�Zd'd(� Zejd)d*� �Zejd+d,� �Z d5d-d.�Z dS )6�_annotatecontextz{do not use this class directly as it does not use lock to protect writes. use "with annotatecontext(...)" instead. c C s< || _ |j| _|| _|| _|| _|| _d | _d | _i | _d S r ) r �uir rZ � linelogpath� revmappath�_linelog�_revmap� _node2path)rY r r rj rk rZ r r r r\ � s z_annotatecontext.__init__c C s� | j d u r�tj�| j�r|t| jd��F}ztj�|� � �| _ W n tj yZ t�� | _ Y n0 W d � q�1 sp0 Y n t�� | _ | j S )Ns rb) rl �osr �existsrj r � linelogmodr Zfromdata�readZ LineLogError�rY r r r r r � s 0 z_annotatecontext.linelogc C s | j d u rt�| j�| _ | j S r )rm � revmapmodr rk �rY r r r r � s z_annotatecontext.revmapc C sj | j d ur| j �� d | _ | jd urft| jd�� }|�| j�� � W d � n1 sV0 Y d | _d S )Ns wb)rm �flushrl r rj �write�encoders r r r �close� s .z_annotatecontext.closec C s& | � � | j�� t| j| jg� dS )z0delete linelog and revmap, useful for rebuildingN)ry rn �clear� _unlinkpathsrk rj ru r r r �rebuild� s z_annotatecontext.rebuildc C s* | j du rt�| j�S | j �| j j�S dS )z6return last node in revmap, or None if revmap is emptyN)rm rt Z getlastnoderk �rev2hsh�maxrevru r r r �lastnode� s z_annotatecontext.lastnodeTc C sl | j }zL| j|dd�}| jj�|�� �}|sJ|rJ||krJ| j�|�duW S ||kW S tyf Y dS 0 dS )a return True if the revmap / linelog is up-to-date, or the file does not exist in the master revision. False otherwise. it tries to be fast and could return false negatives, because of the use of linkrev instead of introrev. useful for both server and client to decide whether to update fastannotate cache or not. if strict is True, even if fctx exists in the revmap, but is not the last node, isuptodate will return False. it's good for performance - no expensive check was done. if strict is False, if fctx exists in the revmap, this function may return True. this is useful for the client to skip downloading the cache if the client's master is behind the server's. T)rF N) r � _resolvefctxr � changelog�noderD r �hsh2rev� LookupError)rY �master�strictr r Zlinknoder r r � isuptodate s z_annotatecontext.isuptodateNFc s, t |t�r t� jj�|��}� �|�\}}|rh� jjrZ� j� d� j t � t �|d��f � � �|||�S d}|r�z� j|ddd�}W n ty� Y n0 |� jv r�d}|g}|r�|�� du r�tjtd�td�d��|�|� |dd� } i } |d i} i } | �r�| �� } | | v s�| | v �rq�| � jv �r\� j�| �� �}� j�|� � jj }|| � � f| | < q�� � | �}|| | < |D ].}| � |d �d | |< || v�rr| �|� �qrq�t � }|} | �r�| � jv�r�|�!| � | | }|�r�|d } nd} �q�q�|du�r� �"| � � jj�rF|�r.� j� d � j t#|�f � n| �sF� j� d � j � � j�� jj$� |dd� } � jj%d t#|�d�}| �r�| d } | | v �r�| �� �qvd}| | }|D ] }|| v�r�d}| �|� �q�|�sԐqv| �� d}| |v }t&| �}t'|�D ]v\}}t(� �)| | d |d ��}|d k�r0|�r0|}t*| | ||�}| | d k�r\| |= | |= n| | d 8 < �q�|| | < | | = |�r�|�+� d}t#|�dk�r�� j,j-�r�|d }|du �r�t(� �)d|d ��}� �.| ||� n|�rv| � � � j/| �� < �qv|�0� � fdd�| | d D �}� �1||||�S )a� incrementally update the cache so it includes revisions in the main branch till 'master'. and run annotate on 'rev', which may or may not be included in the main branch. if master is None, do not update linelog. the first value returned is the annotate result, it is [(node, linenum)] by default. [(node, linenum, path)] if showpath is True. if showlines is True, a second value will be returned, it is a list of corresponding line contents. s6 fastannotate: %s: using fast path (resolved fctx: %s) s nodeNT�rF rG s cannot update linelog to wdir()s set fastannotate.mainbranch)�hintr r s7 fastannotate: %s: %d new changesets in the main branch sB fastannotate: %s: linelog cannot help in annotating this revision s building cache)�total���F� r" c s2 g | ]*\}}t |t�r"� j�|�n|�� |f�qS r )r@ rA r r} r� )r �fr�lru r r r! � s �z-_annotatecontext.annotate.<locals>.<listcomp>)2r@ rA r r r� r� �canannotatedirectlyri � debugflag�debugr r rQ r Z safehasattr�annotatedirectlyr� r� r rC r ZAbortr �append�popr� r �annotate�annotateresultr/ � _parentfuncrX �set�add�_checklastmasterhead�lenr~ Z makeprogressr4 � enumerate�list� _diffblocksr> Z incrementrZ r_ � _appendrevrn Zcomplete�_refineannotateresult)rY rC r� �showpath� showlinesZdirectly�revfctxZ masterfctxZ initvisitZvisitZpcacheZneededZhistr �llrevra r) r Z newmainbranchZprogressZreadyr8 Z ismainbranchZcurrr, �bs� bannotatedr ru r r� ( s� ���� � ���� � �z_annotatecontext.annotatec C s� d}d}t |t�sT|durTttd��t|�dd� �|�}|durT|| jf| jv rT|}|du r�| jrfdnd}| j ||dd�}|| jv }|s�| jr�| j |ddd�}||fS )z�(str) -> bool, fctx or node. return (True, f) if we can annotate without updating the linelog, pass f to annotatedirectly. return (False, f) if we need extra calculation. f is the fctx resolved from rev. TN)� �( c S s d S r r )�xr r r �<lambda>� r" z6_annotatecontext.canannotatedirectly.<locals>.<lambda>r? )rG rF ) r@ rA �bytesr rX r� r r � _perfhackr� )rY rC ra r �hshrG r r r r� � s z$_annotatecontext.canannotatedirectlyc sD | j |ddd�}|| jvr�|g}d}|d }| �|�}|s<qh|d | jv rX|d �� }qh|�|d � q$t�� } | �| j� | �| j � t �� ���| j� t |�D ]0}|�� } t | � || ��} | �| �|| � | }q�n | j} | j�| �� } |�� } ��| �� � �fdd�| D �}|�r$| �|��}|�r@| �|�| �}||f}|S )a (rev : str) -> [(node : str, linenum : int, path : str)] the result has the same format with annotate, but include all (including deleted) lines up to rev. call this after calling annotate(rev, ...) for better performance and accuracy. Tr� r" r� r c s& g | ] \}}|� kr�� |�|f�qS r )r} �r �rr� �r� r r r r! / r" z5_annotatecontext.annotatealllines.<locals>.<listcomp>)r� r r� r/ r� rq r Zcopyfromr� r~ rt �reversedr� r� � _doappendrevZ getalllinesr� r� �_addpathtoresult� _resolvelines)rY rC r� r� r� �chain�ar r) r �br8 �linesr� ra Z linecontentsr r� r �annotatealllines sD z!_annotatecontext.annotatealllinesc s � fdd�|D �}dgt |� }t�t�}t�t |��D ]"}||| || d f �|� q8|�rd}zBt�|�D ]2\\} } } � �| �t j @ r�qr|| d d } q�qrW n t y� | Y S 0 |du �rli } t�|�D ]�\\} } } | | v�rJ|| d d }| j j �r"| j �dt|�| | f � | �|� �| ��}t�|�� �}|| | < | D ]}| | | ||< �qNq�|S | j �dt|� � |�| � | �|� �| ��}|j}t�|�� �}t |�t |�k�r�t�d��t|�D ]:\}}|| }||v �r�|| D ]}|||< �q�||= �q�q\|S ) ah (annotateresult) -> [line]. designed for annotatealllines. this is probably the most inefficient code in the whole fastannotate directory. but we have made a decision that the linelog does not store line contents. so getting them requires random accesses to the revlog data, since they can be many, it can be very slow. c s g | ]}� � |d ��qS )r )r� )r r� �r r r r! A r" z2_annotatecontext._resolvelines.<locals>.<listcomp>Nr r s6 fastannotate: reading %s line #%d to resolve lines %r s+ fastannotate: annotate %s to resolve lines s unexpected annotated lines)r� � collections� defaultdictr� r r2 r� rW �rev2flagrt �sidebranchflag� StopIterationri r� r� r r� �rev2pathr � splitnewlinesr/ r� r� �faerror�CorruptedFileErrorr� )rY r� r r Zrevsra Zkey2idxsr, r� rC Z_linenumZidxsZrevlinesZlinenumr. r� �idxZ annotated�linerL r r� r r� 9 s` �� � z _annotatecontext._resolvelinesc s� t |t�r|}n|�� }� j�|�}|s:t�dt|� ��� j�|�t j @ dkrbt�dt|� ��� j � |� � fdd�� j j D �}� �||||�S )a like annotate, but when we know that f is in linelog. f can be either a 20-char str (node) or a fctx. this is for perf - in the best case, the user provides a node and we don't need to read the filelog or construct any filecontext. s %s is not in revmapr s %s is not in revmap mainbranchc s g | ]\}}� j �|�|f�qS r )r r} r� ru r r r! � s z5_annotatecontext.annotatedirectly.<locals>.<listcomp>)r@ r� r� r r� r� r� r r� rt r� r r� r� r� )rY r r� r� r� r� ra r ru r r� z s � �z!_annotatecontext.annotatedirectlyc C st |r| � |�}|rpt|t�r>| j�|�}| �|| j�|��}n|}t�|� � �}t |�t |�krht � � �||f}|S )znadd the missing path or line contents, they can be expensive. f could be either node or fctx. ) r� r@ r� r r� r� r� r r� r/ r� r� r� )rY ra r r� r� r� r. r� r r r r� � s z&_annotatecontext._refineannotateresultc C s | � | j| j|||� d S r )r� r r )rY r. r8 r� r r r r� � s z_annotatecontext._appendrevc C s t �||| jj�S r )r Z allblocksrZ rN )rY r� r� r r r r� � s z _annotatecontext._diffblocksc s � fdd�}g }i �|durT|D ]2\\}}} } } | dkr |dd� || | � D �7 }q t |�}||v rn|�|� |D ]} || ��| < qr� j|�� |�� d�} | �|< t|�D ]`\\}}} } } | dkr�q�|du r�| �| ||| | � q��fdd�|| | � D �}| �| |||� q�dS ) z'append a revision to linelog and revmapc sF t | t�rt�d��| �� }� �|�}|du rB� j|d| �� d�}|S )z (fctx) -> ints f should not be an intNT)Z sidebranchr )r@ rA r ZProgrammingErrorr� r� r� r )r r� rC r� r r �getllrev� s z/_annotatecontext._doappendrev.<locals>.getllrevNr5 c S s g | ]\}}t |t�s|�qS r �r@ rA )r r r� r r r r! � s �z1_annotatecontext._doappendrev.<locals>.<listcomp>r c s* g | ]"\}}t |t�r|n� | |f�qS r r� r� )� siderevmapr r r! � s �)r� �remover� r� r r� Z replacelinesZreplacelines_vec)r r r. r8 r� r� Zsiderevsr9 r: r; r< �opr r� Zblinesr )r r� r r� � s2 � �z _annotatecontext._doappendrevc s. �du r�j ���fdd�� � fdd�|D �S )z6(revmap, [(node, linenum)]) -> [(node, linenum, path)]Nc s2 �j �| �}|d u r.� �� �| ��}|�j | < |S r )rn rX r� r� )Znodeidr )r rY r r �_getpath� s z3_annotatecontext._addpathtoresult.<locals>._getpathc s g | ]\}}||� |�f�qS r r )r �nr� )r� r r r! � r" z5_annotatecontext._addpathtoresult.<locals>.<listcomp>r� )rY r� r r )r� r rY r r� � s z!_annotatecontext._addpathtoresultc C sB |du rd}n | j �|�� �}|s*t�� �| jj|kr>t�� �dS )z:check if fctx is the master's head last time, raise if notNr )r r� r� r� ZCannotReuseErrorr r~ )rY r. r� r r r r� � s z%_annotatecontext._checklastmasterheadc s" | j j�| j j� � �fdd�}|S )z-> (fctx) -> [fctx]c s t | �d�}� s |d d� }|S )N)r( r )r* )r r) �r_ r^ r r r% � s z-_annotatecontext._parentfunc.<locals>.parents)rZ r^ r_ )rY r% r r� r r� � s z _annotatecontext._parentfuncc C s | j �dd�S )N� fastannotates perfhack)ri � configboolru r r r r� s z_annotatecontext._perfhackc K s t | j||p| jfi |��S r )rI r r )rY rC r �kwdsr r r r� s z _annotatecontext._resolvefctx)T)NFF)FF)N)N)N)N) rc rd re rf r\ �propertyr r ry �__del__r| r r� r� r� r� r� r� r� r� r� � staticmethodr� r� r� r rg r� r� r� r r r r rh � s< # C 8A 1 rh c C s0 | D ]&}zt �|� W q ty( Y q0 qdS )zsilent, best-effort unlinkN)r �unlink�OSError)�pathsr r r r r{ s r{ c @ sH e Zd ZdZefdd�Zedd� �Zedd� �Zdd � Z ed d � �Z d S ) � pathhelperz9helper for getting paths for lockfile, linelog and revmapc C s" t j�d|jt|��| _|| _d S )Nr� )ro r �joinrb rK �_vfspathr' )rY r r rZ r r r r\ s �zpathhelper.__init__c C s t j�| jj�| j��S r )ro r �dirnamer' �vfsr� r� ru r r r r� s zpathhelper.dirnamec C s | j j�| jd �S )Ns .l�r' r� r� r� ru r r r rj ! s zpathhelper.linelogpathc C s t �| jj| jd �S )Ns .lock)�lockmodr r' r� r� ru r r r r % s zpathhelper.lockc C s | j j�| jd �S )Ns .mr� ru r r r rk ( s zpathhelper.revmappathN) rc rd re rf � defaultoptsr\ r� r� rj r rk r r r r r� s r� c c s� t | ||�}t�|j� |j}|j}d}z�zN|�� �2 t| ||||�}|rR|�� |V W d � n1 sl0 Y W n4 t y� |dur�|�� | j � d| � � Y n0 W |dur�|� � n|dur�|� � 0 dS )a context needed to perform (fast) annotate on a file an annotatecontext of a single file consists of two structures: the linelog and the revmap. this function takes care of locking. only 1 process is allowed to write that file's linelog and revmap at a time. when something goes wrong, this function will assume the linelog and the revmap are in a bad state, and remove them from disk. use this function in the following way: with annotatecontext(...) as actx: actx. .... Ns+ fastannotate: %s: cache broken and deleted ) r� r �makedirsr� rk rj r rh r| � Exceptionri r� ry )r r rZ r| �helperrk rj Zactxr r r �annotatecontext- s( ( �r� c C s: | j }| j}|j�ddd�r d}t||d�}t||||�S )zdlike annotatecontext but get the context from a fctx. convenient when used in fctx.annotate r� s forcefollowT)rN r^ )r' �_pathri r� rV r� )r. r( rN r| r r Zaoptsr r r �fctxannotatecontextR s r� )T)FN)TNF)4Z __future__r r� � contextlibro Zmercurial.i18nr Zmercurial.pycompatr r r Zmercurial.noder r r Z mercurialr r rq r r� r r r r Zmercurial.utilsr r � r� r rt Z lrucachefuncr r* r4 r> Z revsinglerB rI rK rU r� r` �objectrV rh r{ r� �contextmanagerr� r� r r r r �<module> s> $ - " F $