File: //usr/lib/python3/dist-packages/hgext/largefiles/__pycache__/remotestore.cpython-310.pyc
o
�]Lb{ � @ sv d Z ddlmZ ddlmZ ddlmZmZmZ ddl m
Z
m Z ddl m
Z
mZmZ ejZejZG dd � d e
j
�Zd
S ) z4remote largefile store; the base class for wirestore� )�absolute_import)�_)�error�pycompat�util)�
stringutil�urlutil� )� basestore�lfutil�
localstorec sh e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
� ZS )� remotestorez)a largefile store accessed over a networkc s@ t t| ��|||� d | _|d ur t�| j| j| j�| _d S d S )N)�superr
�__init__�_lstorer �ui�repo)�selfr r �url�� __class__� �>/usr/lib/python3/dist-packages/hgext/largefiles/remotestore.pyr $ s
�zremotestore.__init__c C sN | � ||�rt�td�|t�| j�f ��| j�td�|t�| j�f � d S )Ns0 remotestore: could not put %s to remote store %ss' remotestore: put %s to remote store %s
) �sendfiler �Abortr r � hidepasswordr r �debug)r �source�hashr r r �put* s ����zremotestore.putc C s dd� t �| �|��D �S )Nc S s i | ] \}}||d k�qS )r r )�.0�h�sr r r �
<dictcomp>6 s ��z&remotestore.exists.<locals>.<dictcomp>)r � iteritems�_stat�r �hashesr r r �exists5 s
��zremotestore.existsc
C s� | j �d||f � z!t�| j |��}| �||�W d � W S 1 s$w Y W d S tyE } zt�td�|t �
|�f ��d }~ww )Ns remotestore: sendfile(%s, %s)
s' remotestore: could not open file %s: %s) r r r � httpsendfile�_put�IOErrorr r r r � forcebytestr)r �filenamer �fd�er r r r = s
(� ����zremotestore.sendfilec C s� z| � |�}W nJ tjy } z t�||| jt�|���d }~w tjy: } zt �
dt � | j�|j
f ��d }~w tyQ } z t�||| jt�|���d }~ww t�||�S )Ns %s: %s)�_get�urlerr� httperrorr
�
StoreErrorr r r, �urlerrorr r r r �reasonr+ r � copyandhash)r �tmpfiler- r �chunksr/ r r r �_getfileH s& ������� zremotestore._getfilec s" | j �|�� � fdd�|D �}|S )Nc s g | ]}� | r|�qS r r )r r ��existslocallymapr r �
<listcomp>a s z7remotestore._hashesavailablelocally.<locals>.<listcomp>)r r( )r r' � localhashesr r: r �_hashesavailablelocally_ s z#remotestore._hashesavailablelocallyc s� d}dd� |D �}| � |�� | �� fdd�|D ��}|D ]M\}}}|� v r7|||f} | j�|| g�}
|
r6d}q || } | ri| dkrP| j�td�||f � d}q | dkrc| j�td �||f � d}q td
| ��q |S ) NFc S s g | ]\}}}|�qS r r )r �csetr- � expectedhashr r r r< f s
�z,remotestore._verifyfiles.<locals>.<listcomp>c s g | ]}|� vr|�qS r r )r r@ �r= r r r< k s
�Tr s" changeset %s: %s: contents differ
� s changeset %s: %s missing
s6 verify failed: unexpected response from statlfile (%r))r> r% r � _verifyfilesr �warnr � RuntimeError) r �contents� filestocheck�failed�expectedhashes�statsr? r- r@ � filetocheck� verifyresult�statr rA r rC d sP �
��
��������zremotestore._verifyfilesc C � t d��)z1Put file with the given hash in the remote store.� abstract method��NotImplementedError)r r r. r r r r* � � zremotestore._putc C rN )z/Get a iterator for content with the given hash.rO rP )r r r r r r0 � rR zremotestore._getc C rN )z�Get information about availability of files specified by
hashes in the remote store. Return dictionary mapping hashes
to return code where 0 means that file is available, other
values if not.rO rP r& r r r r% � s zremotestore._stat)�__name__�
__module__� __qualname__�__doc__r r r( r r9 r> rC r* r0 r% �
__classcell__r r r r r
! s +r
N)rV �
__future__r �mercurial.i18nr � mercurialr r r �mercurial.utilsr r � r
r r r1 �urlreqr
r r r r �<module> s