Dashboard
PHP:
8.2.30
OS:
Linux
User:
dh_nv8b2m
/
/
usr
/
lib
/
python3
/
dist-packages
/
mercurial
/
__pycache__
📤 Upload
📝 New File
📁 New Folder
Close
Editing: wireprotov1peer.cpython-310.pyc
o �]LbV � @ 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 �bytesurl�__name__� _submitone)�args�opts�encoded_args_or_res�decode�self�cmd� encoded_res��f� �;/usr/lib/python3/dist-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 ])\}}t� fdd�|D ��sJ �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�� escapeargr r � <genexpr>P s � 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 �escapebatcharg�all�joinr � iteritems�append)�req�cmds�op�argsdictr r r% r �encodebatchcmdsG s � r4 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, | � � r tjj�| |�S | j�� | �|�S r" )�doner �futures�Future�result� _peerexecutor� sendcommands)r �timeoutr r r r9 d s zunsentfuture.resultr" )r � __module__� __qualname__�__doc__r9 r r r r r5 [ s r5 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 �exctype�excvalee�exctbr r r �__exit__~ s zpeerexecutor.__exit__c s� �j rt�d���jrt�d��t�jt���d ���s#t�d� ��� ���fdd�}t�dd�r=|� }t|_ �|_ |S �j rGt�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 r7 r8 rG �addrD 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) rB r �ProgrammingErrorrC r rA r �sysstrr5 � __class__r: rD r; )r rU r rW r r rT r � callcommand� s< ���� ���zpeerexecutor.callcommandc C s� | j rd S | js d S d| _ | jD ]}t|t�r tjj|_d |_ q| j}d | _t |�dkrd|d \}}}}|� � s:d S z |di t� |���}W n t y\ t�|t�� dd � � Y d S w |�|� d S g }g }|D ]L\}}}}|� � suqjz|j|jfi t� |���\} } W n t y� t�|t�� dd � � Y d S w | s�|�| � qj|�|| f� |�||t| f� qj|s�d S | j�|�} tj�d�| _| j�| j|| �| _d S )NTr r r ) rB rD rG � isinstancer5 r r7 r8 rZ r: �len� set_running_or_notify_cancel� strkwargs� Exception�future_set_exception_info�sys�exc_info� set_resultr! �__self__r/ rA � _submitbatch�ThreadPoolExecutorrH �submit�_readbatchresponserI ) r r �callsrU r rV r9 �requests�statesr r � wireresultsr r r r; � sd � � �� � �zpeerexecutor.sendcommandsc C s� | � � | jr d S d| _| jsd S z-| j�� W | jjdd� d | _d | _| jD ]}|�� s9|�t � t d�d �� q(d | _d S | jjdd� d | _d | _| jD ]}|�� s`|�t � t d�d �� qOd | _w )NT)�wait�" unfulfilled batch command response) r; rC rI r9 rH �shutdownrG r6 � set_exceptionr � ResponseErrorr )r r r r r rN s@ ��� � ���zpeerexecutor.closec C s� |D ]B\}}}}zt |�}W n ty" |�t�td�d �� Y qw z||�}W n ty> t�|t � � dd � � Y qw |� |� qd S )Nro r ) �next� StopIterationrq r rr r r` r ra rb rc rd ) r rl rm rU r r! r � remoteresultr9 r r r ri * s" ��� � �z peerexecutor._readbatchresponseN) r r= r> rK rM rR r[ r; rN ri 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@ rL r r r �commandexecutorP s zwirepeer.commandexecutorc C s | � dtd�� | �d�S )Ns clonebundless clone bundles)� requirecapr �_callrL r r r � clonebundlesU s zwirepeer.clonebundlesc s. � � dtd�� � fdd�}dt�|�i|fS )Ns lookups look up remote revisionc s< | d d� � dd�\}}t|�rt|�S � �t�|�� d S )N���� r )�split�intr �_abortr � RepoError)�d�success�datarL r r r ] s z wirepeer.lookup.<locals>.decode� key)rx r r � fromlocal)r �keyr r rL r �lookupY s zwirepeer.lookupc � � fdd�}i |fS )Nc s@ z t �| d d� �W S ty � �t�td�| �� Y d S w )Nr{ � unexpected response:)r � decodelist� ValueErrorr r rr r �r� rL r r r h s �z wirepeer.heads.<locals>.decoder �r r r rL r �headsf s zwirepeer.headsc s � fdd�}dt �|�i|fS )Nc sB z dd� t �| �D �W S ty � �t�td�| �� Y d S w )Nc S s g | ]}t t|���qS r )�boolr~ �r# �br r r � <listcomp>t s z2wirepeer.known.<locals>.decode.<locals>.<listcomp>r� )r � iterbytestrr� r r rr r r� rL r r r r s �z wirepeer.known.<locals>.decode� nodes�r � encodelist)r �nodesr r rL r �knownp s zwirepeer.knownc r� )Nc sv z%i }| � � D ]}|�dd�\}}t�t�|��}t�|�}|||< q|W S ty: � � t � t d�| �� Y d S w )Nr| r r� ) � splitlinesr} r �tolocal�urlreq�unquoter r� � TypeErrorr r rr r )r� � branchmap� branchpart� branchname� branchheadsrL r r r | s �z"wirepeer.branchmap.<locals>.decoder r� r rL r r� z s zwirepeer.branchmapc sB �� d�s i 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�debugr] � pushkeymod� decodekeysr� �� namespacer r r r � s � z!wirepeer.listkeys.<locals>.decode� namespace��capabler� r� r r� )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 sf | � dd�\} }ztt| ��} W n ty t�td�| ��w |�d�D ] }� j� td�|� q%| S )N� r �" push failed (unexpected response):T� remote: ) r} r� r~ r� r rr r r� r� �status)r� �output�lrL r r r � s ��z wirepeer.pushkey.<locals>.decode)r� r� s olds newr� )r r� r� �old�newr r rL r r � s ��zwirepeer.pushkeyc C s | � d�S )Ns stream_out)� _callstreamrL r r r � stream_out� s zwirepeer.stream_outc K s" t �|�}| �dtd�� i }|�d�pt� }t �|�D ]S\}}|d u r%q tj�|�}|d u r6t � d| ��|dkr@t� |�}n+|dkrJd� |�}n!|dkrVd� t |��}n|d krad t|� }n |d krktd | ��|||< q | jdi t �|���}td d� |D ��r�t�| j|�S t�|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# �capr r r r' � s � z%wirepeer.getbundle.<locals>.<genexpr>� UN)r� )r � byteskwargsrx r �get�setr. r �GETBUNDLE_ARGUMENTSr rX r� r- �sortedr� �KeyError�_callcompressabler_ �anyr � getunbundlerr� �changegroupmod� cg1unpacker) r �source�kwargsr � bundlecapsr� �value�keytyper r r r � getbundle� s6 � zwirepeer.getbundlec C s� |dgkr | � d�r t�dt�d�t|����� g�}nt�|�}t� |d�rf| j d||d�\}}|dkr=t � t d�|��zt|�}W n tyR t � t d �|��w |�d �D ] }| j�t d �|� qX|S | 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 hashed� s deltaheaders unbundle)r� s push failed:r� Tr� )r� r r� r �sha1r- r� �digestr � safehasattr� _callpushr rr r r~ r� r� r� r� �_calltwowaystreamr r� )r �bundler� �url�retr� r� �streamr r r �unbundle� s, � ���zwirepeer.unbundlec C s\ t �|�}| jd|d�}z dd� |�� D �}|W S ty- | �t�td�|�� Y d S w )Ns branches)r� c S s g | ] }t t�|���qS r )�tupler r� r� r r r r� s z%wirepeer.branches.<locals>.<listcomp>r� ) r r� ry r� r� r r rr r )r r� �nr� �brr r r �branches� s �zwirepeer.branchesc C s� d}g }t �dt|�|�D ];}d�dd� |||| � D ��}| jd|d�}z|�dd � |�� D �� W q tyH | �t � t d �|�� Y q w |S ) N� r r| c S s g | ]}t �|d ��qS )� -r� )r# �pr r r r� s ��z$wirepeer.between.<locals>.<listcomp>s between)�pairsc s s" � | ] }|r t �|�p g V qd S r" )r r� )r# r� r r r r' s � � �z#wirepeer.between.<locals>.<genexpr>r� ) r �xranger] r- ry �extendr� r� r r rr r )r r� �batch�r�ir� r� 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� )rx r r r� r� r� r� )r r� r� r� r r r r �changegroupsubset! s � zwirepeer.changegroupsubsetc c s � | j }|jr&|�dd�r&|�d� |D ]\}}d}|�||t|�f � qtj}| jdt|�d�}|� d�}|g} |rwd|vrR|rR|� d�}| � |� d|vrR|sBd � | �} d| v rl| � dd �\} } || �V d| v s[|� d�}| |g} |s<|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)r1 i r* r� r N) r� � debugflag� configboolr� r] r �unescapebatchargr� r4 �readr/ r- r} ) r r0 r� r2 r �msg� unescapearg�rsp�chunk�work�merged�oner r r rf , s2 � � � � zwirepeer._submitbatchc C s | j |fi t�|���S r" )ry r r_ )r r2 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� )ry )r r� r� r� r� �fiver r r r � debugwireargsL s zwirepeer.debugwireargsc K � t � �)z�execute <cmd> on the server The command is expected to return a simple string. returns the server reply as a string.��NotImplementedError�r r r r r r ry U s zwirepeer._callc K r� )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. r� r� r r r r� ] � zwirepeer._callstreamc K r� )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. r� r� r r r r� h s zwirepeer._callcompressablec K r� )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. r� �r r �fpr r r r r� u r� zwirepeer._callpushc K r� )zqexecute <cmd> on server The command will send a stream to the server and get a stream in reply. r� r� r r r r� � s zwirepeer._calltwowaystreamc C r� )zBclearly abort the wire protocol connection and raise the exceptionr� )r � exceptionr r r r � s zwirepeer._abort)NNN) r r= r> r? rw rz r! r� r� r� r� r� r r� r� r� r� r� r r� rf r r� ry r� r� r� r� r r r r r rv D s@ - rv )(� __future__r rb rE �i18nr �noder r r r � r r r� r r r r� r r � interfacesr � interfaceutil�utilsr r� r! r4 r7 r8 r5 � implementer�ipeercommandexecutor�objectr@ � ipeercommands�ipeerlegacycommandsrJ rv r r r r �<module> s( ( ! T�
Save
Cancel