File: //lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyc
�
1�3\c @@ sN d d l m Z d d l Z d d l Z d d l Z d d l m 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 m Z d d l m Z e j Z d � Z d e f d
� � YZ d � Z d e j j f d
� � YZ e j e j � d e f d � � Y� Z! e j e j" e j# � d e j$ f d � � Y� Z% d S( i ( t absolute_importNi ( t _( t bin( t bundle2t changegroupt encodingt errort pushkeyt pycompatt
repositoryt utilt wireprototypes( t
interfaceutilc @ s6 � f d � } t | d � � t | d � j � | S( sU annotation for batchable methods
Such methods must implement a coroutine as follows:
@batchable
def sample(self, one, two=None):
# Build list of encoded arguments suitable for your wire protocol:
encargs = [('one', encode(one),), ('two', encode(two),)]
# Create future for injection of encoded result:
encresref = future()
# Return encoded arguments and future:
yield encargs, encresref
# Assuming the future to be filled with the result from the batched
# request now. Decode it:
yield decode(encresref.value)
The decorator returns a function which wraps this coroutine as a plain
method, but adds the original method as an attribute called "batchable",
which is used by remotebatch to split the call into separate encoding and
decoding phases.
c @ sj � | | � } t | � \ } } | s+ | S| d } t j � j � } | j | j | | � � t | � S( Ni ( t nextR t bytesurlt __name__t sett
_submitone( t argst optst batchablet encargsorrest encresreft selft cmd( t f( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyt plain9 s
R R ( t setattrR ( R R ( ( R s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyR # s t futurec B@ s e Z d Z d � Z RS( s' placeholder for a value to be set laterc C@ s1 t j | d � r$ t j d � � n | | _ d S( Nt values future is already set( R
t safehasattrR t RepoErrorR ( R R ( ( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyR H s ( R t
__module__t __doc__R ( ( ( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyR F s c @ s� t j � g } xr | D]j \ } } t � f d � | D� � sD t � d j � f d � | j � D� � } | j d | | f � q Wd j | � S( s; Return a ``cmds`` argument value for the ``batch`` command.c 3@ s! | ] } � | � | k Vq d S( N( ( t .0t k( t escapearg( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pys <genexpr>V s t ,c 3@ s1 | ]' \ } } d � | � � | � f Vq d S( s %s=%sN( ( R" R# t v( R$ ( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pys <genexpr>X s s %s %st ;( R t escapebatchargt allt AssertionErrort joint iteritemst append( t reqt cmdst opt argsdictR ( ( R$ s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyt encodebatchcmdsM s "t unsentfuturec B@ s e Z d Z d d � Z RS( s0 A Future variation to represent an unsent command.
Because we buffer commands and don't submit them immediately, calling
``result()`` on an unsent future could deadlock. Futures for buffered
commands are represented by this type, which wraps ``result()`` to
call ``sendcommands()``.
c C@ s< | j � r" t j j j | | � S| j j � | j | � S( N( t doneR t futurest Futuret resultt
_peerexecutort sendcommands( R t timeout( ( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyR7 g s
N( R R R! t NoneR7 ( ( ( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyR3 ^ s t peerexecutorc B@ sG e Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z RS( c C@ sI | | _ t | _ t | _ g | _ t j � | _ d | _ d | _
d S( N( t _peert Falset _sentt _closedt _callst weakreft WeakSett _futuresR; t _responseexecutort
_responsef( R t peer( ( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyt __init__t s c C@ s | S( N( ( R ( ( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyt __enter__} s c C@ s | j � d S( N( t close( R t exctypet excvaleet exctb( ( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyt __exit__� s c @ s� � j r t j d � � n � j r6 t j d � � n t � j t j � � d � � � sp t j d � � � n � � � � f d � } t � d t � r� | � } t
| _ � | _ n2 � j
r� t j d � � � n | � } � j � | S( Ns4 callcommand() cannot be used after commands are sents* callcommand() cannot be used after close()sA cannot call command %s: method of same name not available on peerc @ s? t j j � } � j j | � � j j � � � | f � | S( N( R R5 R6 RD t addRA R- ( R ( R t commandt fnR ( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyt addcall� s R sX %s is not batchable and cannot be called on a command executor along with other commands( R? R t ProgrammingErrorR@ t getattrR= R t sysstrR; R> R3 t __class__R8 RA R9 ( R RP R RR R ( ( R RP RQ R s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyt callcommand� s*
c
C@ sS | j r
d S| j s d St | _ x; | j D]0 } t | t � r- t j j | _ d | _ q- q- W| j } d | _ t | � d k r| d \ } } } } | j
� s� d Sy | t j | � � } Wn+ t k
r� t j | t j � d � n X| j | � d Sg } g } x� | D]� \ } } } } | j
� s:qn y" | j | j t j | � � } Wn, t k
r�t j | t j � d � d SXy t | � \ }
} Wn, t k
r�t j | t j � d � d SX| j | |
f � | j | | | | f � qW| s
d S| j j | � } t j j d � | _ | j j | j | | � | _ d S( Ni i ( R? RA t TrueRD t
isinstanceR3 R R5 R6 RV R; R8 t lent set_running_or_notify_cancelt strkwargst Exceptiont future_set_exception_infot syst exc_infot
set_resultR t __self__R
R- R= t _submitbatcht ThreadPoolExecutorRE t submitt _readbatchresponseRF (
R R t callsRP R RQ R7 t requestst statesR t encodedargst fremotet wireresults( ( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyR9 � sZ
c C@ s� | j � | j r d St | _ | j s- d Sz | j j � Wd | j j d t � d | _ d | _ x< | j D]1 } | j � sp | j
t j t
d � � � qp qp Wd | _ Xd S( Nt waits" unfulfilled batch command response( R9 R@ RX RF R7 RE t shutdownR; RD R4 t
set_exceptionR t
ResponseErrorR ( R R ( ( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyRJ
s
c C@ s� x~ | D]v \ } } } } t | � } | j | � y t | � } Wn+ t k
ro t j | t j � d � q X| j | � q Wd S( Ni ( R
R R] R R^ R_ R` Ra ( R Ri Rl RP R R Rk t remoteresultR7 ( ( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyRf ' s
( R R RH RI RN RW R9 RJ Rf ( ( ( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyR< r s 3 T t wirepeerc B@ s e Z d Z d � Z d � Z e d � � Z e d � � Z e d � � Z e d � � Z e d � � Z
e d � � Z d � Z d
� Z
d � Z d � Z d
� Z d � Z d � Z d � Z d � Z d d d d � Z d � Z d � Z d � Z d � Z d � Z d � Z RS( s� Client-side interface for communicating with a peer repository.
Methods commonly call wire protocol commands of the same name.
See also httppeer.py and sshpeer.py for protocol-specific
implementations of this interface.
c C@ s
t | � S( N( R< ( R ( ( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyt commandexecutorB s c C@ s# | j d t d � � | j d � S( Nt clonebundless
clone bundles( t
requirecapR t _call( R ( ( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyRt G s c c@ s� | j d t d � � t � } i t j | � d 6| f V| j } | d j d d � \ } } t | � ry t | � Vn | j t
j | � � d S( Nt lookups look up remote revisiont keyi����t i ( Ru R R R t fromlocalR t splitt intR t _abortR R ( R Rx R t dt successt data( ( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyRw K s c c@ sj t � } i | f V| j } y t j | d � VWn0 t k
re | j t j t d � | � � n Xd S( Ni����s unexpected response:( R R R t
decodelistt
ValueErrorR} R Rp R ( R R R~ ( ( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyt headsW s
c c@ s� t � } i t j | � d 6| f V| j } y1 g t j | � D] } t t | � � ^ q@ VWn0 t k
r� | j t
j t d � | � � n Xd S( Nt nodess unexpected response:(
R R t
encodelistR R t iterbytestrt boolR| R� R} R Rp R ( R R� R R~ t b( ( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyt knowna s 1
c c@ s� t � } i | f V| j } yo i } x] | j � D]O } | j d d � \ } } t j t j | � � } t j | � } | | | <q3 W| VWn0 t
k
r� | j t j
t d � | � � n Xd S( NRy i s unexpected response:( R R t
splitlinesR{ R t tolocalt urlreqt unquoteR R� t TypeErrorR} R Rp R ( R R R~ t branchmapt
branchpartt
branchnamet branchheads( ( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyR� k s
c c@ s� | j d � s i d f Vn t � } | j j d | � i t j | � d 6| f V| j } | j j d | t | � f � t j
| � Vd S( NR s preparing listkeys for "%s"
t namespaces$ received listkey for "%s": %i bytes
( t capableR; R t uit debugR Rz R RZ t
pushkeymodt
decodekeys( R R� R R~ ( ( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyt listkeys{ s c c@ s* | j d � s t d f Vn t � } | j j d | | f � i t j | � d 6t j | � d 6t j | � d 6t j | � d 6| f V| j } | j d d � \ } } y t
t | � � } Wn) t k
r� t
j t d � | � � n Xx0 | j t � D] } | j j t d
� | � q� W| Vd S( NR s preparing pushkey for "%s:%s"
R� Rx t oldt news
i s" push failed (unexpected response):s remote: ( R� R> R; R R� R� R Rz R R{ R� R| R� R Rp R R� RX t status( R R� Rx R� R� R R~ t outputt l( ( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyR � s$
c C@ s
| j d � S( Nt
stream_out( t _callstream( R ( ( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyR� � s c K@ s� t j | � } | j d t d � � i } | j d � p@ t � } x| j � D]� \ } } | d k rn qP n t j j | � } | d k r� t
j d | � � n� | d k r� t j | � } n� | d k r� d j
| � } nb | d k rd j
t | � � } n> | d k r!d
t | � } n | d k r@t d | � � n | | | <qP W| j d t j | � � } t d
� | D� � r�t j | j | � St j | d � Sd S( Nt getbundles look up remote changest
bundlecapss$ Unexpectedly None keytype for key %sR� t csvR% t scsvt booleans %iR s unknown getbundle option type %sc s@ s | ] } | j d � Vq d S( t HG2N( t
startswith( R" t cap( ( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pys <genexpr>� s t UN( R t byteskwargsRu R t getR R, R; R t GETBUNDLE_ARGUMENTSR RS R� R+ t sortedR� t KeyErrort _callcompressableR\ t anyR t getunbundlerR� t changegroupmodt cg1unpacker( R t sourcet kwargsR R� Rx R t keytypeR ( ( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyR� � s6
c C@ s] | d g k rT | j d � rT t j d t j d j t | � � � j � g � } n t j | � } t j | d � r,| j
d | d | �\ } } | d k r� t j t
d � | � � n y t | � } Wn) t k
r� t j t
d � | � � n Xx` | j t � D] } | j j t
d
� | � qWn- | j d | d | �} t j | j | � } | S( s� Send cg (a readable file-like object representing the
changegroup to push, typically a chunkbuffer object) to the
remote server as a bundle.
When pushing a bundle10 stream, return an integer indicating the
result of the push (see changegroup.apply()).
When pushing a bundle20 stream, return a bundle20 stream.
`url` is the url the client thinks it's pushing to, which is
visible to hooks.
t forcet unbundlehasht hashedt t deltaheadert unbundleR� s push failed:s" push failed (unexpected response):s remote: ( R� R R� t hashlibt sha1R+ R� t digestR
R t _callpushR Rp R R| R� R� RX R� R� t _calltwowaystreamR R� ( R t bundleR� t urlt retR� R� t stream( ( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyR� � s& 0
c C@ s� t j | � } | j d d | �} y6 g | j � D] } t t j | � � ^ q4 } | SWn0 t k
r� | j t j t
d � | � � n Xd S( Nt branchesR� s unexpected response:( R R� Rv R� t tupleR� R� R} R Rp R ( R R� t nR~ R� t br( ( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyR� � s .
c C@ s� d } g } x� t j d t | � | � D]� } d j g | | | | !D] } t j | d � ^ qF � } | j d d | �} y! | j d � | j � D� � Wq( t k
r� | j
t j t
d � | � � q( Xq( W| S( Ni i Ry t -t betweent pairsc s@ s* | ] } | r t j | � p! g Vq d S( N( R R� ( R" R� ( ( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pys <genexpr>� s s unexpected response:( R t xrangeRZ R+ R R� Rv t extendR� R� R} R Rp R ( R R� t batcht rt it pR� R~ ( ( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyR� � s " 0
'c C@ s4 t j | � } | j d d | �} t j | d � S( NR t rootsR� ( R R� R� R� R� ( R R� R� R� R ( ( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyR s c C@ s_ | j d t d � � t j | � } t j | � } | j d d | d | �} t j | d � S( Nt changegroupsubsets look up remote changest basesR� R� ( Ru R R R� R� R� R� ( R R� R� R� R ( ( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyR� s c c@ sd | j } | j rn | j d d � rn | j d � x: | D]/ \ } } d } | j | | t | � f � q8 Wn t j } | j d d t | � �} | j d � } | g } x� | rKx2 d | k r� | r� | j d � } | j
| � q� Wd j | � }
x3 d |
k r,|
j d d
� \ } }
| | � Vq� W| j d � } |
| g } q� W| d j | � � Vd S( sq run batch request <req> on the server
Returns an iterator of the raw responses from the server.
t devels debug.peer-requests$ devel-peer-request: batched-content
s+ devel-peer-request: - %s (%d arguments)
R� R/ i R' R� i N(
R� t debugflagt
configboolR� RZ R t unescapebatchargR� R2 t readR- R+ R{ ( R R. R� R0 R t msgt unescapeargt rspt chunkt workt mergedt one( ( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyRc s*
$ c C@ s | j | t j | � � S( N( Rv R R\ ( R R0 R ( ( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyR , s c C@ sT i } | d k r | | d <n | d k r8 | | d <n | j d d | d | | �S( Nt threet fourt
debugwireargsR� t two( R; Rv ( R R� R� R� R� t fiveR ( ( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyR� / s
c K@ s
t � � d S( s� execute <cmd> on the server
The command is expected to return a simple string.
returns the server reply as a string.N( t NotImplementedError( R R R ( ( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyRv 8 s c K@ s
t � � d S( s execute <cmd> on the server
The command is expected to return a stream. Note that if the
command doesn't return a stream, _callstream behaves
differently for ssh and http peers.
returns the server reply as a file like object.
N( R� ( R R R ( ( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyR� @ s c K@ s
t � � d S( sE execute <cmd> on the server
The command is expected to return a stream.
The stream may have been compressed in some implementations. This
function takes care of the decompression. This is the only difference
with _callstream.
returns the server reply as a file like object.
N( R� ( R R R ( ( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyR� K s c K@ s
t � � d S( s� execute a <cmd> on server
The command is expected to be related to a push. Push has a special
return method.
returns the server reply as a (ret, output) tuple. ret is either
empty (error) or a stringified int.
N( R� ( R R t fpR ( ( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyR� X s c K@ s
t � � d S( sq execute <cmd> on server
The command will send a stream to the server and get a stream in reply.
N( R� ( R R R� R ( ( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyR� c s c C@ s
t � � d S( sK clearly abort the wire protocol connection and raise the exception
N( R� ( R t exception( ( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyR} j s N( R R R! Rs Rt R Rw R� R� R� R� R R� R� R� R� R� R R� Rc R R; R� Rv R� R� R� R� R} ( ( ( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyRr 8 s2
,
(& t
__future__R R� R_ RB t i18nR t nodeR R� R R R� R R R R� R R R
R t utilsR R� R t objectR R2 R5 R6 R3 t implementert ipeercommandexecutorR< t
ipeercommandst ipeerlegacycommandsRG Rr ( ( ( s? /usr/lib64/python2.7/site-packages/mercurial/wireprotov1peer.pyt <module> s" @ # �