| Server IP : 162.144.4.212 / Your IP : 216.73.216.108 Web Server : Apache System : Linux gator2125.hostgator.com 5.14.0-162.23.1.9991722448259.nf.el9.x86_64 #1 SMP PREEMPT_DYNAMIC Wed Jul 31 18:11:45 UTC 2024 x86_64 User : cozeellc ( 2980) PHP Version : 8.3.31 Disable Function : NONE MySQL : OFF | cURL : ON | WGET : ON | Perl : ON | Python : OFF | Sudo : ON | Pkexec : ON Directory : /lib64/python3.9/site-packages/mercurial/__pycache__/ |
Upload File : |
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 sR t j� g }| D ]8\}}d�� fdd�t�|�D ��}|�d||f � qd�|�S )z;Return a ``cmds`` argument value for the ``batch`` command.� ,c 3 s& | ] \}}d � |�� |�f V qdS )s %s=%sNr )�.0�k�v�Z escapeargr r � <genexpr>R s �z"encodebatchcmds.<locals>.<genexpr>s %s %s� ;)r Zescapebatcharg�joinr
� iteritems�append)�req�cmds�opZargsdictr r r$ r �encodebatchcmdsG s �r- 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 �N)�doner
�futures�Future�result�
_peerexecutor� sendcommands)r �timeoutr r r r3 d s
zunsentfuture.result)N)r �
__module__� __qualname__�__doc__r3 r r r r r. [ s r. 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
r1 r2 rA �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
Zsysstrr. � __class__r4 r> r5 )r rN r rP r r rM 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> rA �
isinstancer. r
r1 r2 rR r4 �lenZ set_running_or_notify_cancel� strkwargs� Exception�future_set_exception_info�sys�exc_info�
set_resultr �__self__r) r; � _submitbatchZThreadPoolExecutorrB Zsubmit�_readbatchresponserC ) r r ZcallsrN r rO r3 Zrequests�statesr r � wireresultsr r r r5 � 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) r5 r= rC r3 rB ZshutdownrA r0 �
set_exceptionr �
ResponseErrorr )r r r r r rH
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 )Nrb r ) �next�
StopIterationrc r rd r rW r
rX rY rZ r[ ) r r_ r` rN r r r Z remoteresultr3 r r r r^ * s ��
z peerexecutor._readbatchresponseN)
r r7 r8 rE rG rK rS r5 rH 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: rF r r r �commandexecutorP s zwirepeer.commandexecutorc C s | � dtd�� | �d�S )Ns clonebundless
clone bundles)�
requirecapr �_callrF 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�datarF r r r ] s z wirepeer.lookup.<locals>.decode� key)ri r r
� fromlocal)r �keyr r rF r �lookupY s zwirepeer.lookupc s � fdd�}i |fS )Nc sB zt �| d d� �W S ty< � �t�td�| �� Y n0 d S )Nrl � unexpected response:)r �
decodelist�
ValueErrorrp r rd r �rq rF r r r h s z wirepeer.heads.<locals>.decoder �r r r rF 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 )�boolro �r! �br r r �
<listcomp>t � z2wirepeer.known.<locals>.decode.<locals>.<listcomp>rx )r
Z iterbytestrrz rp r rd r r{ rF r r r r s z wirepeer.known.<locals>.decode� nodes�r �
encodelist)r �nodesr r rF 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 )Nrm r rx )
�
splitlinesrn r
Ztolocal�urlreq�unquoter ry � TypeErrorrp r rd r )rq � branchmapZ
branchpartZ
branchnameZ branchheadsrF r r r | s
z"wirepeer.branchmap.<locals>.decoder r| r rF 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�debugrU �
pushkeymodZ
decodekeysr{ �� namespacer r r r � s �z!wirepeer.listkeys.<locals>.decode� namespace��capabler� r� r
ru )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: )
rn r~ ro rz r rd r r� r� �status)rq �output�lrF r r r � s �
z wirepeer.pushkey.<locals>.decode)r� rt s olds newr� )r r� rv �old�newr r rF r r � s
��zwirepeer.pushkeyc C s
| � d�S )Ns
stream_out)� _callstreamrF 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 byteskwargsri r �get�setr( r ZGETBUNDLE_ARGUMENTSr rQ r� r' �sortedr~ �KeyError�_callcompressablerV �anyr � getunbundlerr� �changegroupmod� cg1unpacker) r �source�kwargsr Z
bundlecapsrv �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 rd r ro rz 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 ry r r r r r� r� z%wirepeer.branches.<locals>.<listcomp>rx ) r r� rj r� rz rp r rd r )r r� �nrq �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 rm 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 ry )r! r� r r r r% s �z#wirepeer.between.<locals>.<genexpr>rx ) r
ZxrangerU r' rj �extendr� rz rp r rd r )r r� Zbatch�r�ir� rq 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� )ri 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� rU r Zunescapebatchargr� r- �readr) r' rn ) r r* r� r, r �msgZ unescapeargZrsp�chunkZworkZmerged�oner r r r] , s*
zwirepeer._submitbatchc C s | j |fi t�|���S r/ )rj r
rV )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� )rj )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 rj 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 rp � s zwirepeer._abort)NNN) r r7 r8 r9 rh rk r rw r} r� r� r� r r� r� r� r� r� r r� r] r r� rj r� r� r� r� rp r r r r rg D s>
-
rg )(Z
__future__r rY r? Zi18nr Znoder r
r r � r r r� r
r r r� r r Z
interfacesr Z
interfaceutilZutilsr r� r r- r1 r2 r. Z implementerZipeercommandexecutor�objectr: Z
ipeercommandsZipeerlegacycommandsrD rg r r r r �<module> s& (
!
T�