File: //proc/thread-self/root/lib64/python2.7/site-packages/hgext/fastannotate/protocol.pyo
�
1�3\c @@ s� d d l m 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 d d l
m Z d � Z d � Z d � Z d
� Z d � Z d � Z d
� Z e j d � � Z d d d � Z d � Z d � Z d � Z d S( i ( t absolute_importN( t _( t errort
extensionst hgt utilt wireprotov1peert wireprotov1serveri ( t contextc C@ s4 | j d d � } | s0 t j t d � � � n | S( s) get the mainbranch, and enforce it is sett fastannotatet
mainbranchsF fastannotate.mainbranch is required for both the client and the server( t configR t AbortR ( t uit master( ( sA /usr/lib64/python2.7/site-packages/hgext/fastannotate/protocol.pyt
_getmaster s c C@ s | | | � } | j d � | S( Nt getannotate( t append( t origt repot protot result( ( sA /usr/lib64/python2.7/site-packages/hgext/fastannotate/protocol.pyt
_capabilities# s
c
C@ s� d } | j j d d t � } t j | | � �U} | r� t | j � } zz y& | j | � sp | j | | � n WnM t k
r� | j � y | j | | � Wq� t k
r� | j � q� Xn XWd | j
� Xn | j | k r�x� | j | j
g D]� } t j j | � sq� n d } t | d � � }
|
j � } Wd QXt | j j d � } | | } | d | t | � | f 7} q� Wn Wd QX| S( Nt R t serverbuildondemandt rbt /s %s %d %s( R
t
configboolt TrueR t annotatecontextR t
isuptodatet annotatet Exceptiont rebuildt closet lastnodet
revmappatht linelogpatht ost patht existst opent readt lent vfst base(
R R R' R# R t
buildondemandt actxR t pt contentt ft
vfsbaselent relpath( ( sA /usr/lib64/python2.7/site-packages/hgext/fastannotate/protocol.pyt _getannotate( s8
*c C@ s- d t j k r d St j d d � t � d S( NR s
path lastnode( R t commandst wireprotocommandR5 ( ( ( sA /usr/lib64/python2.7/site-packages/hgext/fastannotate/protocol.pyt _registerwireprotocommandR s c C@ s t � t j t d t � d S( NR ( R8 R t wrapfunctionR R ( R
( ( sA /usr/lib64/python2.7/site-packages/hgext/fastannotate/protocol.pyt
serveruisetupX s c C@ s� i } d } t | � d } d } d } } x� | | k r� | | } | d k r� | d k r� t | | d t | � � | | <| t | � 7} d } d } } q� | d k r� d } q� n2 | d k r� | | 7} n | d k r� | | 7} n | d 7} q/ W| S( Ni i R s ( R+ t buffert int( t payloadR t it lt statet vfspatht sizet ch( ( sA /usr/lib64/python2.7/site-packages/hgext/fastannotate/protocol.pyt _parseresponse^ s*
c @ s, d | j f � f d � � Y} | | _ d S( Nt fastannotatepeerc @ s# e Z e j d � f d � � Z RS( c 3@ sm | j d � s* � j t d � � d Vn? i | d 6| p= d d 6} t j � } | | f Vt | j � Vd S( NR s* remote peer cannot provide annotate cache
R' R R# ( NN( t capablet warnR t NoneR t futureRD t value( t selfR' R# t argsR2 ( R
( sA /usr/lib64/python2.7/site-packages/hgext/fastannotate/protocol.pyR x s N( t __name__t
__module__R t batchableRH R ( ( R
( sA /usr/lib64/python2.7/site-packages/hgext/fastannotate/protocol.pyRE w s ( t __class__( R
t peerRE ( ( R
sA /usr/lib64/python2.7/site-packages/hgext/fastannotate/protocol.pyt peersetupv s c c@ sW | j } | j | j d d d � � } t j | i | � } z | VWd | j � Xd S( NR t
remotepatht default( R
t
expandpathR R RQ R" ( R R
RS RQ ( ( sA /usr/lib64/python2.7/site-packages/hgext/fastannotate/protocol.pyt annotatepeer� s c
C@ s� | s
d S| d k rA t | � � } t | | | | � SWd QXn | d k rV i } n | j } g } | j � �W} | j d t | � � x> | D]6 } | j | j d i | d 6| j | � d 6� � q� Wx� | D]� } | j
� } d � | j � D� } x� t | � D]� }
|
j
d � s.d |
d k rE| j d
|
� qn | |
} | j rx| j d t | � |
f � n | j j t j j |
� � | j |
d � � } | j | � Wd QXqWq� WWd QXd S(
s1 download annotate cache from the server for pathsNs" fastannotate: requesting %d files
R R' R# c S@ s( i | ] \ } } | t j | � � q S( ( R t pconvert( t .0R0 t v( ( sA /usr/lib64/python2.7/site-packages/hgext/fastannotate/protocol.pys
<dictcomp>� s s
fastannotate/s /../R s( fastannotate: ignored malicious path %s
s% fastannotate: writing %d bytes to %s
t wb( RH RV t clientfetchR
t commandexecutort debugR+ R t callcommandt getR t iteritemst sortedt
startswitht debugflagR, t makedirsR& R' t dirnamet write(
R t pathst lastnodemapRQ R
t resultst batcherR0 R t rR' R1 R2 ( ( sA /usr/lib64/python2.7/site-packages/hgext/fastannotate/protocol.pyR[ � s<
%
c C@ s� | j j d d d � } | d k r( | Sg } x^ | D]V } y/ t | j | � � | k ri | j | � n Wq5 t k
r� | j | � q5 Xq5 W| S( s� return a subset of paths whose history is long and need to fetch linelog
from the server. works with remotefilelog and non-remotefilelog repos.
R t clientfetchthresholdi
i ( R
t configintR+ t fileR R ( R Rg t thresholdR R' ( ( sA /usr/lib64/python2.7/site-packages/hgext/fastannotate/protocol.pyt _filterfetchpaths� s
c C@ s& d | j f d � � Y} | | _ d S( Nt fastannotaterepoc B@ s e Z d d � Z RS( c S@ s� t | j � } g } i } y� xd t | | � D]S } t j | | � �9 } | j | d t �s{ | j | � | j | | <n Wd QXq. W| r� t | | | | � n Wn' t
k
r� } | j j d | � n Xd S( Nt stricts" fastannotate: prefetch failed: %r
( R R
Rp R R R t FalseR R# R[ R R] ( RK Rg RQ R t needupdatepathsRh R' R/ t ex( ( sA /usr/lib64/python2.7/site-packages/hgext/fastannotate/protocol.pyt prefetchfastannotate� s
N( RM RN RH Rv ( ( ( sA /usr/lib64/python2.7/site-packages/hgext/fastannotate/protocol.pyRq � s ( RP ( R
R Rq ( ( sA /usr/lib64/python2.7/site-packages/hgext/fastannotate/protocol.pyt localreposetup� s c C@ sI t � | j � r# t | | � n t t j k rE t j j t � n d S( N( R8 t localRw RR R t wirepeersetupfuncsR ( R
R ( ( sA /usr/lib64/python2.7/site-packages/hgext/fastannotate/protocol.pyt clientreposetup� s
( t
__future__R t
contextlibR& t mercurial.i18nR t mercurialR R R R R R R R R R R5 R8 R: RD RR t contextmanagerRV RH R[ Rp Rw Rz ( ( ( sA /usr/lib64/python2.7/site-packages/hgext/fastannotate/protocol.pyt <module> s" .
*
'