| Server IP : 80.87.202.40 / Your IP : 216.73.216.169 Web Server : Apache System : Linux rospirotorg.ru 5.14.0-539.el9.x86_64 #1 SMP PREEMPT_DYNAMIC Thu Dec 5 22:26:13 UTC 2024 x86_64 User : bitrix ( 600) PHP Version : 8.2.27 Disable Function : NONE MySQL : OFF | cURL : ON | WGET : ON | Perl : ON | Python : OFF | Sudo : ON | Pkexec : ON Directory : /usr/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 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�