File: //lib64/python2.7/site-packages/hgext/fastannotate/context.pyc
�
1�3\c @@ s� d d l m Z d d l Z d d l Z d d l Z d d l Z d d l m Z d d l m Z m
Z m Z
m Z m Z m Z m Z m Z d d l m Z d d l m Z m Z e j d � � Z e d � Z d
� Z d � Z e j e j � Z e! e" d � Z# d
� Z$ d � Z% e% e j& � Z' d e( f d � � YZ) e) � Z& d e( f d � � YZ* d � Z+ d e( f d � � YZ, e j- e& e! d � � Z. e e" e! d � Z/ d S( i ( t absolute_importN( t _( t errort linelogt lockt mdifft nodet pycompatt scmutilt util( t
stringutili ( R t revmapc C@ s
| j | � S( N( t file( t repot path( ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyt _getflog$ s c C@ s� | j | j � } | sM g | D]$ } | j � | j � k r | ^ q } n x; | D]3 } d | j k rT t | j | j � � | _ qT qT W| S( Nt _filelog( t _changeidt parentsR t __dict__R t _repoR ( t ft followt plt p( ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyt _parents) s 4
"c C@ sj | j � } | j d � } | r> | j d � r> | d 7} n g t j | � D] } | | f ^ qN | f S( Ns
i ( t datat countt endswithR t xrange( t fctxt textt linecountt i( ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyt _decorate? s
c C@ sR xK | D]C \ \ } } } } } | d k r | d | | !| d | | +q q W| S( Nt =i ( ( t parentt childt blockst a1t a2t b1t b2t t( ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyt _pairI s #c C@ s� | r4 t | t � r4 | d k r4 t | | � } n
| | } | d k r� | j � d k ri | | } q� | j | d | j � �} n\ | | } | d k r� | j � } n | j � } | | j � k r� | | _ | | | _ n | S( s2 (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.
t changeidt linkrevN(
t
isinstancet intt Nonet
_revsinglet revt filectxR. t introrevR t
_changectx( R
R3 R t
resolverevt adjustctxt ctxR R5 ( ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyt resolvefctxU s "
c C@ s4 | j d d � j d d � j d d � j d d � S( Ns .hg/s .hg.hg/s .l/s .l.hg/s .m/s .m.hg/s .lock/s .lock.hg/( t replace( R ( ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyt encodedir� s
c @ sB t j t � f d � t j j D� � � } t j | � j � d S( Nc 3@ s$ | ] } | t � | � f Vq d S( N( t getattr( t .0t k( t diffopts( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pys <genexpr>� s i ( R
t pprintt sortedR R@ t defaultst hashlibt sha1t hexdigest( R@ t
diffoptstr( ( R@ s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyt hashdiffopts� s t annotateoptsc B@ sD e Z d Z i d d 6e d 6e d 6Z d � Z e j d � � Z RS( s� 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
R@ t followrenamet followmergec K@ s@ x9 | j j � D]( \ } } t | | | j | | � � q Wd S( N( RC t iteritemst setattrt get( t selft optsR? t v( ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyt __init__� s c C@ s� d } | j s | d 7} n | j s2 | d 7} n | j d k r� t | j t j � s\ t � t | j � } | t k r� | d | 7} q� n | p� d S( s? represent opts in a short string, suitable for a directory namet t r0t m0R! t defaultN( RJ RK R@ R1 R/ R t AssertionErrorRH t _defaultdiffopthash( RO t resultt diffopthash( ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyt shortstr� s
N(
t __name__t
__module__t __doc__R1 t TrueRC RR R t
propertycacheR[ ( ( ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyRI � s
t _annotatecontextc B@ s e Z d Z d � Z e d � � Z e d � � Z d � Z e Z d � Z e d � � Z
e d � Z d e e d � Z d � Z e e d
� Z d � Z d � Z d
� Z d d � Z d � Z e d d � � Z d d � Z d � Z e j d � � Z e j d � � Z d d � Z RS( s{ do not use this class directly as it does not use lock to protect
writes. use "with annotatecontext(...)" instead.
c C@ sX | | _ | j | _ | | _ | | _ | | _ | | _ d | _ d | _ i | _ d S( N(
R
t uiR RP t linelogpatht
revmappathR1 t _linelogt _revmapt
_node2path( RO R
R Rc Rd RP ( ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyRR � s c C@ s� | j d k r� t j j | j � r� t | j d � �L } y t j j | j
� � | _ Wn# t j k
r} t j � | _ n XWd QXq� t j � | _ n | j S( Nt rb( Re R1 t osR t existsRc t opent
linelogmodR t fromdatat readt LineLogError( RO R ( ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyR � s c C@ s. | j d k r' t j | j � | _ n | j S( N( Rf R1 t revmapmodR Rd ( RO ( ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyR � s c C@ sx | j d k r( | j j � d | _ n | j d k rt t | j d � � } | j | j j � � Wd QXd | _ n d S( Nt wb( Rf R1 t flushRe Rk Rc t writet encode( RO R ( ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyt close� s
c C@ s1 | j � | j j � t | j | j g � d S( s0 delete linelog and revmap, useful for rebuildingN( Ru Rg t cleart _unlinkpathsRd Rc ( RO ( ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyt rebuild� s
c C@ s9 | j d k r t j | j � S| j j | j j � Sd S( s6 return last node in revmap, or None if revmap is emptyN( Rf R1 Rp t getlastnodeRd t rev2hsht maxrev( RO ( ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyt lastnode� s c C@ s� | j } ym | j | d t �} | j j j | j � � } | rk | rk | | k rk | j j | � d k S| | k SWn t
k
r� t SXd S( s 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.
R7 N( R| t _resolvefctxR_ R
t changelogR R. R t hsh2revR1 t LookupError( RO t mastert strictR| R t linknode( ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyt
isuptodate� s
c C@ s" t | t � r0 t j | j j j | � � } n | j | � \ } } | r� | j j r� | j j d | j
t j t
j | d � � f � n | j | | | � Sd } | r� y | j | d t d t �} Wn t k
r� q� X| | j k r� d } q� n | g } | rN| j � d k r>t j t d � d t d � �� n | j | � n | } i }
i d | 6} i } x� | rc| j � }
|
|
k sq|
| k r�qqn |
| j k r | j j |
j � � } | j j | � | j j } | |
j � f | |
<qqn | j |
� } | |
|
<xD | D]< } | j! | d � d | | <| |
k r | j | � q q WqqWt" � } | }
xJ |
r�|
| j k r�| j# |
� |
|
} | r�| d }
qvd }
PqvW| d k r�| j$ |
� n | j j r7| r| j j d
| j
t% | � f � q7| s7| j j d | j
� q7n | j j | j j& � | } d } xE| r�| d }
|
| k r�| j � q]n t } |
|
} x0 | D]( } | | k r�t' } | j | � q�q�W| s�q]n | j � d } |
| k } |
| k st( � t) |
� } x� t* | � D]� \ } } t+ | j, | | d | d � � } | d k rj| rj| } n t- | | | | � } | | d k r�| | =| | =q | | c d 8<q W| | |
<|
|
=| r|| j j. s
| d 7} | j j/ t d
� | d t% | � �n d } t% | � d k r;| j0 j1 r;| d } n | d k rft+ | j, d | d � � } n | j2 |
| | � q]| r]|
j
� | j3 |
j � <q]q]W| r�| j j4 � n g | | d D]? \ } } t | t � r�| j j5 | � n | j � | f ^ q�} | j6 | | | | � S( s� 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)
R R7 R8 s cannot update linelog to wdir()t hints set fastannotate.mainbranchi i s7 fastannotate: %s: %d new changesets in the main branch
sB fastannotate: %s: linelog cannot help in annotating this revision
i����s building cachet totali RS N(7 R/ R0 R t hexR
R~ t canannotatedirectlyRb t debugflagt debugR R
RA R t safehasattrt annotatedirectlyR1 R} R_ R� R R3 R t AbortR t appendt popR R t annotatet annotateresultR t _parentfuncRN t sett addt _checklastmasterheadt lenR{ t FalseRW R" t enumeratet listt _diffblocksR, t quiett progressRP RK t
_appendrevRg Rs Rz t _refineannotateresult( RO R3 R� t showpatht showlinest directlyt revfctxt
masterfctxt initvisitt visitt pcachet neededt histR t llrevRY R R t
newmainbranchR� t readyR& t ismainbrancht currR! t bst
bannotatedt frt l( ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyR� s� !
$
Qc C@ s t } d } t | t � r� | d k r� i t d 6t j d 6j t | � d � � | � } | d k r� | | j f | j
k r� | } q� n | d k r| j r� d n t } | j | d | d t �} | | j
k } | r| j r| j | d t d t �} qn | | f S( s� (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.
i i( c S@ s d S( N( R1 ( t x( ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyt <lambda>� RS R. R8 R7 N(
R_ R1 R/ R0 t bytesR t binRN R� R R t _perfhackR} ( RO R3 RY R t hshR8 ( ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyR� � s 2$ !c C@ s� | j | d t d t �} | | j k rC| g } d } xe t r� | d } | j | � } | se Pn | d | j k r� | d j � } Pn | j | d � q< Wt j � } | j | j � | j | j
� t j � }
|
j | j � xi t | � D]F } | j � } t
| j | | � � } | j | |
| | � | } q� Wn | j } | j }
| j � }
| j � } |
j | � } g |
D]- \ } } | | k r�|
j | � | f ^ q�} | r�| j | |
� } n | r�| j | |
| � } | | f } n | S( s
(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.
R7 R8 RS i����i ( R} R_ R R� R R� Rl R t copyfromR� R{ Rp t reversedR� R� t _doappendrevt getalllinesR R Rz t _addpathtoresultt
_resolvelines( RO R3 R� R� R� t chaint aR R R R t bR& t linesR� R� t rR� RY t linecontents( ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyt annotatealllines� sF
:c C@ s� g | D] } | j | d � ^ q } d g t | � } t j t � } x@ t j t | � � D]) } | | | | | d f j | � q^ Wxl| r�d } yV xO | j � D]A \ \ }
} } | j
|
� t j @r� q� n | | d d } Pq� WWn t
k
r| SX| d k r
i }
x� | j � D]� \ \ }
} } |
|
k r�| | d d } | j j r�| j j d t j | � | | f � n | j | | j |
� � } t j | j � � } | |
|
<n x | D] } |
|
| | | <q�Wq$Wt d � | D� � s t � | S| j j d t j | � � | j |
� | j | | j |
� � } | j } t j | j � � } t | � t | � k r�t j d � � n xY t | � D]K \ } } | | } | | k r�x | | D] } | | | <q�W| | =q�q�Wq� W| S( sh (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.
i i s6 fastannotate: reading %s line #%d to resolve lines %r
c s@ s | ] } | d k Vq d S( N( R1 ( R> R� ( ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pys <genexpr>D s s+ fastannotate: annotate %s to resolve lines
s unexpected annotated linesN( R R1 R� t collectionst defaultdictR� R R R� RL t rev2flagRp t sidebranchflagt
StopIterationRb R� R� R t shortR} t rev2pathR t
splitnewlinesR t allRW R� R� t faerrort CorruptedFileErrorR� ( RO R� R R R� t revsRY t key2idxsR! R� R3 t _linenumt idxst revlinest linenumR R� t idxt annotatedt lineR? ( ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyR� sZ & '
c C@ s� t | t � r | } n | j � } | j j | � } | s[ t j d t j | � � � n | j j | � t j
@d k r� t j d t j | � � � n | j j | � g | j j
D]$ \ } } | j j | � | f ^ q� } | j | | | | � S( s 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 revmapi s %s is not in revmap mainbranch( R/ t strR R R R� R� R� R� Rp R� R R� R� Rz R� ( RO R R� R� R� R� R� R� RY ( ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyR� Y s 4c C@ s� | r | j | � } n | r� t | t � r` | j j | � } | j | | j j | � � } n | } t j | j � � } t
| � t
| � k r� t j � � n | | f } n | S( sn add 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� ( RO RY R R� R� R� R R� ( ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyR� o s !c C@ s# | j | j | j | | | � d S( N( R� R R ( RO R R& R� ( ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyR� � s c C@ s t j | | | j j � S( N( R t allblocksRP R@ ( RO R� R� ( ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyR� � s c @ s� � f d � } g } i } | d k r� xk | D]` \ \ } } }
} } | d k r. | g | |
| !D]! \ }
} t |
t � sc |
^ qc 7} q. q. Wn t | � } | | k r� | j | � n x | D] }
| |
� | |
<q� W� j | j � d | j � �} | | | <x� t | � D]� \ \ } } }
} } | d k r@qn | d k rh| j | | | |
| � qg | |
| !D]1 \ } } t | t � r�| n | | | f ^ qv} | j
| | | | � qWd S( s' append a revision to linelog and revmapc @ sp t | t � r! t j d � � n | j � } � j | � } | d k rl � j | d t d | j � �} n | S( s
(fctx) -> ints f should not be an intt
sidebranchR N(
R/ R0 R t ProgrammingErrorR R R1 R� R_ R ( R R� R3 ( R ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyt getllrev� s $R# R N( R1 R/ R0 R� t removeR� R R R� t replacelinest replacelines_vec( R R R R& R� R� t siderevst
siderevmapR' R( R) R* t opR R� R� R� t blines( ( R s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyR� � s.
&
!
% Bc @ sV � d k r � j � n � � f d � } g | D]! \ } } | | | | � f ^ q1 S( s6 (revmap, [(node, linenum)]) -> [(node, linenum, path)]c @ sJ � j j | � } | d k rF � j � j | � � } | � j | <n | S( N( Rg RN R1 R� R ( t nodeidR ( R RO ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyt _getpath� s
N( R1 R ( RO R� R R� t nR� ( ( R RO s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyR� � s c C@ sg | d k r d } n- | j j | j � � } | sB t j � � n | j j | k rc t j � � n d S( s: check if fctx is the master's head last time, raise if noti N( R1 R R R R� t CannotReuseErrorR R{ ( RO R R� ( ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyR� � s c @ s. | j j � | j j � � � f d � } | S( s -> (fctx) -> [fctx]c @ s) t | d � �} � s% | d } n | S( NR i ( R ( R R ( RK RJ ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyR � s
( RP RJ RK ( RO R ( ( RK RJ s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyR� � s c C@ s | j j d d � S( Nt fastannotatet perfhack( Rb t
configbool( RO ( ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyR� � s c K@ s t | j | | p | j | � S( N( R: R
R ( RO R3 R t kwds( ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyR} � s N( R\ R] R^ RR t propertyR R Ru t __del__Rx R| R_ R� R1 R� R� R� R� R� R� R� R� R� t staticmethodR� R� R� R R` R� R� R} ( ( ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyRa � s0
#� 8 > , c C@ s: x3 | D]+ } y t j | � Wq t k
r1 q Xq Wd S( s silent, best-effort unlinkN( R t unlinkt OSError( t pathsR ( ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyRw � s
t
pathhelperc B@ sb e Z d Z e d � Z e d � � Z e d � � Z d � Z e j
d � � Z e d � � Z RS( s9 helper for getting paths for lockfile, linelog and revmapc C@ s1 t j j d | j t | � � | _ | | _ d S( NR� ( Ri R t joinR[ R< t _vfspathR ( RO R
R RP ( ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyRR � s c C@ s" t j j | j j j | j � � S( N( Ri R t dirnameR t vfsR� R� ( RO ( ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyR� � s c C@ s | j j j | j d � S( Ns .l( R R� R� R� ( RO ( ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyRc � s c C@ s t j | j j | j d � S( Ns .lock( t lockmodR R R� R� ( RO ( ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyR � s c c@ s� d d l } | j } t j t j j | � � t j | t j t j Bd � } | j
| | j � z d VWd | j
| | j � t j
| � Xd S( sg the same as 'lock' but use flock instead of lockmod.lock, to avoid
creating temporary symlinks.i Ni� ( t fcntlRc R t makedirsRi R R� Rk t O_RDONLYt O_CREATt flockt LOCK_EXt LOCK_UNRu ( RO R� t lockpatht lockfd( ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyt
_lockflock� s c C@ s | j j j | j d � S( Ns .m( R R� R� R� ( RO ( ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyRd s (
R\ R] R^ t defaultoptsRR R� R� Rc R t
contextlibt contextmanagerR Rd ( ( ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyR� � s c
c@ s� t | | | � } t j | j � | j } | j } d } z� yG | j � �5 t | | | | | � } | rx | j � n | VWd QXWnA t
k
r� | d k r� | j � n | j j d | � � n XWd | d k r� | j
� n Xd S( s 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 R� R� Rd Rc R1 R Ra Rx t ExceptionRb R� Ru ( R
R RP Rx t helperRd Rc t actx( ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyt annotatecontext s&
c C@ s[ | j } | j } | j j d d t � r3 t } n t d | d | � } t | | | | � S( sd like annotatecontext but get the context from a fctx. convenient when
used in fctx.annotate
R� t forcefollowR@ RJ ( R t _pathRb R� R_ RI R ( R R R@ Rx R
R t aopts( ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyt fctxannotatecontext3 s (0 t
__future__R R� R RD Ri t mercurial.i18nR t mercurialR R Rl R R� R R R R R t mercurial.utilsR
RS R� R Rp t lrucachefuncR R_ R R" R, t revsingleR2 R� R1 R: R< RH R RX t objectRI Ra Rw R� R R R
( ( ( s@ /usr/lib64/python2.7/site-packages/hgext/fastannotate/context.pyt <module> s6 :
, � � / '#