/
lib64
/
python3.9
/
site-packages
/
mercurial
/
__pycache__
/
New Directory
Upload File
HOME
a �+�bV � @ s� d dl mZ d dlZd dlZddlmZ ddlmZ ddlm Z m Z ddl m Z m ZmZmZmZmZmZmZ ddlmZmZ dd lmZ ejZd d � Zd d � Z G dd� dej j �Z e� ej!�G dd� de"��Z#e� ej$ej%�G dd� dej&��Z'dS )� )�absolute_importN� )�_)�bin)�getattr�setattr)�bundle2� changegroup�encoding�error�pushkey�pycompat�util�wireprototypes)� repositoryr )�hashutilc s* � fdd�}t |d� � t |d� j� |S )a� 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: encoded_args = [('one', encode(one),), ('two', encode(two),)] # Return it, along with a function that will receive the result # from the batched request. return encoded_args, decode 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 sB � | i |��\}}|s|S | d }t �� j�}|�||�}||�S )Nr )r Zbytesurl�__name__� _submitone)�args�opts�encoded_args_or_res�decode�self�cmdZ encoded_res��f� �?/usr/lib64/python3.9/site-packages/mercurial/wireprotov1peer.py�plain9 s zbatchable.<locals>.plain� batchabler )r r )r r r r r r & s r c sl t j� g }| D ]R\}}t� fdd�|D ��s0J �d�� fdd�t�|�D ��}|�d||f � qd�|�S )z;Return a ``cmds`` argument value for the ``batch`` command.c 3 s | ]}� |�|kV qd S �Nr )�.0�k�Z escapeargr r � <genexpr>P � z"encodebatchcmds.<locals>.<genexpr>� ,c 3 s& | ] \}}d � |�� |�f V qdS )s %s=%sNr )r! r" �vr# r r r$ R s �s %s %s� ;)r Zescapebatcharg�all�joinr � iteritems�append)�req�cmds�opZargsdictr r r# r �encodebatchcmdsG s �r0 c @ s e Zd ZdZddd�ZdS )� unsentfuturea0 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()``. Nc C s, | � � rtjj�| |�S | j�� | �|�S r )�doner �futures�Future�result� _peerexecutor� sendcommands)r �timeoutr r r r5 d s zunsentfuture.result)N)r � __module__� __qualname__�__doc__r5 r r r r r1 [ s r1 c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dS )� peerexecutorc C s2 || _ d| _d| _g | _t�� | _d | _d | _d S )NF) �_peer�_sent�_closed�_calls�weakref�WeakSet�_futures�_responseexecutor� _responsef)r �peerr r r �__init__r s zpeerexecutor.__init__c C s | S r r �r r r r � __enter__{ s zpeerexecutor.__enter__c C s | � � d S r )�close)r �exctypeZexcvalee�exctbr r r �__exit__~ s zpeerexecutor.__exit__c s� �j rt�d���jr t�d��t�jt���d ���sFt�d� ��� ���fdd�}t�dd�rx|� }t|_ �|_ n"�j r�t�d� ��|� }�� � |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��� �| f� | S r )r r3 r4 rC �addr@ r, r �r �command�fnr r r �addcall� s z)peerexecutor.callcommand.<locals>.addcallr FsX %s is not batchable and cannot be called on a command executor along with other commands) r> r �ProgrammingErrorr? r r= r Zsysstrr1 � __class__r6 r@ r7 )r rP r rR r r rO r � callcommand� s: ���� ��zpeerexecutor.callcommandc C s� | j r d S | jsd S d| _ | jD ] }t|t�r tjj|_d |_ q | j}d | _t |�dkr�|d \}}}}|� � std S z|f i t� |���}W n* t y� t�|t�� dd � � Y n 0 |�|� d S g }g }|D ]�\}}}}|� � s�q�z"|j|jfi t� |���\} } W n0 t �y: t�|t�� dd � � Y d S 0 | �sN|�| � q�|�|| f� |�||t| f� q�|�szd S | j�|�} tj�d�| _| j�| j|| �| _d S )NTr r ) r> r@ rC � isinstancer1 r r3 r4 rT r6 �lenZ set_running_or_notify_cancel� strkwargs� Exception�future_set_exception_info�sys�exc_info� set_resultr �__self__r, r= � _submitbatchZThreadPoolExecutorrD Zsubmit�_readbatchresponserE ) r r ZcallsrP r rQ r5 Zrequests�statesr r � wireresultsr r r r7 � s\ �� �zpeerexecutor.sendcommandsc C s� | � � | jrd S d| _| js"d S zX| j�� W | jjdd� d | _d | _| jD ]"}|�� sP|�t � t d�d �� qPd | _nL| jjdd� d | _d | _| jD ]"}|�� s�|�t � t d�d �� q�d | _0 d S )NT)�wait�" unfulfilled batch command response) r7 r? rE r5 rD ZshutdownrC r2 � set_exceptionr � ResponseErrorr )r r r r r rJ s< ��� ��zpeerexecutor.closec C s� |D ]�\}}}}z t |�}W n( tyD |�t�td�d �� Y q0 z ||�}W n* ty| t�|t � � dd � � Y q0 |� |� qd S )Nrd r ) �next� StopIterationre r rf r rY r rZ r[ r\ r] ) r ra rb rP r r r Z remoteresultr5 r r r r` * s �� z peerexecutor._readbatchresponseN) r r9 r: rG rI rM rU r7 rJ r` r r r r r< p s 7R r< c @ s� e Zd ZdZdd� Zdd� Zedd� �Zedd � �Zed d � �Z ed d � �Z edd� �Z edd� �Z dd� Z dd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd3d%d&�Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Z d$S )4�wirepeerz�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 r )r< rH r r r �commandexecutorP s zwirepeer.commandexecutorc C s | � dtd�� | �d�S )Ns clonebundless clone bundles)� requirecapr �_callrH r r r � clonebundlesU s zwirepeer.clonebundlesc s. � � dtd�� � fdd�}dt�|�i|fS )Ns lookups look up remote revisionc s< | d d� � dd�\}}t|�r(t|�S � �t�|�� d S )N���� r )�split�intr �_abortr Z RepoError)�d�success�datarH r r r ] s z wirepeer.lookup.<locals>.decode� key)rk r r � fromlocal)r �keyr r rH r �lookupY s zwirepeer.lookupc s � fdd�}i |fS )Nc sB zt �| d d� �W S ty< � �t�td�| �� Y n0 d S )Nrn � unexpected response:)r � decodelist� ValueErrorrr r rf r �rs rH r r r h s z wirepeer.heads.<locals>.decoder �r r r rH r �headsf s zwirepeer.headsc s � fdd�}dt �|�i|fS )Nc sD zdd� t �| �D �W S ty> � �t�td�| �� Y n0 d S )Nc S s g | ]}t t|���qS r )�boolrq �r! �br r r � <listcomp>t r% z2wirepeer.known.<locals>.decode.<locals>.<listcomp>rz )r Z iterbytestrr| rr r rf r r} rH r r r r s z wirepeer.known.<locals>.decode� nodes�r � encodelist)r �nodesr r rH r �knownp s zwirepeer.knownc s � fdd�}i |fS )Nc sx zJi }| � � D ]6}|�dd�\}}t�t�|��}t�|�}|||< q|W S tyr � � t � t d�| �� Y n0 d S )Nro r rz ) � splitlinesrp r Ztolocal�urlreq�unquoter r{ � TypeErrorrr r rf r )rs � branchmapZ branchpartZ branchnameZ branchheadsrH r r r | s z"wirepeer.branchmap.<locals>.decoder r~ r rH r r� z s zwirepeer.branchmapc sB �� d�si d fS �j�d� � � �fdd�}dt�� �i|fS )N� pushkeys preparing listkeys for "%s" c s" �j �d� t| �f � t�| �S )Ns$ received listkey for "%s": %i bytes )�ui�debugrW � pushkeymodZ decodekeysr} �� namespacer r r r � s �z!wirepeer.listkeys.<locals>.decode� namespace��capabler� r� r rw )r r� r r r� r �listkeys� s zwirepeer.listkeysc sX � � d�sdS � j�d||f � � fdd�}t�|�t�|�t�|�t�|�d�|fS )Nr� )FNs preparing pushkey for "%s:%s" c sj | � dd�\} }ztt| ��} W n" tyB t�td�| ��Y n0 |�d�D ]}� j� td�|� qN| S )N� r �" push failed (unexpected response):T� remote: ) rp r� rq r| r rf r r� r� �status)rs �output�lrH r r r � s � z wirepeer.pushkey.<locals>.decode)r� rv s olds newr� )r r� rx �old�newr r rH r r � s ��zwirepeer.pushkeyc C s | � d�S )Ns stream_out)� _callstreamrH r r r � stream_out� s zwirepeer.stream_outc K s* t �|�}| �dtd�� i }|�d�p,t� }t �|�D ]�\}}|d u rJq8tj�|�}|d u rnt � d| ��nj|dkr�t� |�}nV|dkr�d� |�}nB|dkr�d� t |��}n*|d kr�d t|� }n|d kr�td | ��|||< q8| jdi t �|���}td d� |D ���rt�| j|�S t�|d�S d S )N� getbundle� look up remote changess bundlecapss$ Unexpectedly None keytype for key %sr� s csvr& s scsvs booleans %is plains unknown getbundle option type %sc s s | ]}|� d �V qdS )s HG2N)� startswith)r! Zcapr r r r$ � r% z%wirepeer.getbundle.<locals>.<genexpr>� UN)r� )r Z byteskwargsrk r �get�setr+ r ZGETBUNDLE_ARGUMENTSr rS r� r* �sortedr� �KeyError�_callcompressablerX �anyr � getunbundlerr� �changegroupmod� cg1unpacker) r �source�kwargsr Z bundlecapsrx �valueZkeytyper r r r � getbundle� s6 � zwirepeer.getbundlec C s� |dgkr8| � d�r8t�dt�d�t|����� g�}n t�|�}t� |d�r�| j d||d�\}}|dkrzt � t d�|��z t|�}W n" ty� t � t d �|��Y n0 |�d �D ]}| j�t d �|� q�n | jd||d�}t�| j|�}|S ) a� 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. s forces unbundlehashs hashedr% s deltaheaders unbundle)r s push failed:r� Tr� )r� r r� r Zsha1r* r� �digestr Z safehasattr� _callpushr rf r rq r| r� r� r� �_calltwowaystreamr r� )r Zbundler �url�retr� r� �streamr r r �unbundle� s( � � zwirepeer.unbundlec C s^ t �|�}| jd|d�}zdd� |�� D �}|W S tyX | �t�td�|�� Y n0 d S )Ns branches)r� c S s g | ]}t t�|���qS r )�tupler r{ r� r r r r� r% z%wirepeer.branches.<locals>.<listcomp>rz ) r r� rl r� r| rr r rf r )r r� �nrs �brr r r �branches� s zwirepeer.branchesc C s� d}g }t �dt|�|�D ]x}d�dd� |||| � D ��}| jd|d�}z |�dd � |�� D �� W q ty� | �t � t d �|�� Y q0 q|S ) N� r ro c S s g | ]}t �|d ��qS )� -r� )r! �pr r r r� s �z$wirepeer.between.<locals>.<listcomp>s between)�pairsc s s | ]}|rt �|�pg V qd S r )r r{ )r! r� r r r r$ s �z#wirepeer.between.<locals>.<genexpr>rz ) r ZxrangerW r* rl �extendr� r| rr r rf r )r r� Zbatch�r�ir� rs r r r �between s �� � zwirepeer.betweenc C s$ t �|�}| jd|d�}t�|d�S )Ns changegroup)�rootsr� )r r� r� r� r� )r r� r� r� r r r r r s zwirepeer.changegroupc C s@ | � dtd�� t�|�}t�|�}| jd||d�}t�|d�S )Ns changegroupsubsetr� )�basesr r� )rk r r r� r� r� r� )r r� r r� r r r r �changegroupsubset! s �zwirepeer.changegroupsubsetc c s� | j }|jrJ|�dd�rJ|�d� |D ]"\}}d}|�||t|�f � q&tj}| jdt|�d�}|� d�}|g} |r�d|vr�|r�|� d�}| � |� qvd � | �} d| v r�| � dd �\} } || �V q�|� d�}| |g} qr|d � | ��V d S ) zqrun batch request <req> on the server Returns an iterator of the raw responses from the server. s devels debug.peer-requests$ devel-peer-request: batched-content s+ devel-peer-request: - %s (%d arguments) s batch)r. i r( r% r N) r� Z debugflagZ configboolr� rW r Zunescapebatchargr� r0 �readr, r* rp ) r r- r� r/ r �msgZ unescapeargZrsp�chunkZworkZmerged�oner r r r_ , s* zwirepeer._submitbatchc C s | j |fi t�|���S r )rl r rX )r r/ r r r r r I s zwirepeer._submitoneNc C s: i }|d ur||d< |d ur$||d< | j d||d�|��S )N�three�four� debugwireargs)r� �two)r� )rl )r r� r� r� r� Zfiver r r r � debugwireargsL s zwirepeer.debugwireargsc K s t � �dS )z�execute <cmd> on the server The command is expected to return a simple string. returns the server reply as a string.N��NotImplementedError�r r r r r r rl U s zwirepeer._callc K s t � �dS )a 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. Nr� r� r r r r� ] s zwirepeer._callstreamc K s t � �dS )aE 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. Nr� r� r r r r� h s zwirepeer._callcompressablec K s t � �dS )z�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. Nr� �r r �fpr r r r r� u s zwirepeer._callpushc K s t � �dS )zqexecute <cmd> on server The command will send a stream to the server and get a stream in reply. Nr� r� r r r r� � s zwirepeer._calltwowaystreamc C s t � �dS )zBclearly abort the wire protocol connection and raise the exceptionNr� )r � exceptionr r r rr � s zwirepeer._abort)NNN) r r9 r: r; rj rm r ry r r� r� r� r r� r� r� r� r� r r� r_ r r� rl r� r� r� r� rr r r r r ri D s> - ri )(Z __future__r r[ rA Zi18nr Znoder r r r � r r r� r r r r� r r Z interfacesr Z interfaceutilZutilsr r� r r0 r3 r4 r1 Z implementerZipeercommandexecutor�objectr< Z ipeercommandsZipeerlegacycommandsrF ri r r r r �<module> s& ( ! T�