File: //usr/lib/python3/dist-packages/hgext/infinitepush/__pycache__/store.cpython-310.pyc
o
�]Lb� � @ s� d dl mZ d dlZd dlZd dlZd dlmZ d dlmZ d dl m
Z
d dl m Z m
Z
G dd� de�ZG d d
� d
e�ZG d d � d e�ZG d
d� de�Zddd�ZG dd� de�ZdS )� )�absolute_importN)�hex)�open)�pycompat)�hashutil�procutilc @ � e Zd ZdS )�BundleWriteExceptionN��__name__�
__module__� __qualname__� r r �:/usr/lib/python3/dist-packages/hgext/infinitepush/store.pyr � r c @ r )�BundleReadExceptionNr
r r r r r r r c @ s2 e Zd ZdZejZejdd� �Zejdd� �Z dS )�abstractbundlestorea6 Defines the interface for bundle stores.
A bundle store is an entity that stores raw bundle data. It is a simple
key-value store. However, the keys are chosen by the store. The keys can
be any Python object understood by the corresponding bundle index (see
``abstractbundleindex`` below).
c C � dS )z�Write bundle data to the store.
This function receives the raw data to be written as a str.
Throws BundleWriteException
The key of the written data MUST be returned.
Nr )�self�datar r r �write( � zabstractbundlestore.writec C r )z�Obtain bundle data for a key.
Returns None if the bundle isn't known.
Throws BundleReadException
The returned object should be a file object supporting read()
and close().
Nr )r �keyr r r �read1 r zabstractbundlestore.readN)
r r r
�__doc__�abc�ABCMeta�
__metaclass__�abstractmethodr r r r r r r s
r c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
�filebundlestorezkbundle store in filesystem
meant for storing bundles somewhere on disk and on network filesystems
c C sT || _ || _|�dd�| _| js| jj�dd�| _tj�| j�s(t� | j� d S d S )Ns
scratchbranchs storepaths scratchbranchess filebundlestore)
�ui�repo�
configpath� storepath�vfs�join�os�path�exists�makedirs)r r r! r r r �__init__B s ��zfilebundlestore.__init__c C s" t j�| j|dd� |dd� �S )z�First two bytes of the hash are the name of the upper
level directory, next two bytes are the name of the
next level directoryr � � )r& r' r% r# )r � hashvaluer r r �_dirpathM s "zfilebundlestore._dirpathc C s t j�| �|�|�S )N)r& r' r% r. )r �filenamer r r � _filepathS s zfilebundlestore._filepathc C sr t t�|��� �}| �|�}tj�|�st�|� t | �
|�d��}|� |� W d � |S 1 s2w Y |S )Ns wb) r r �sha1�digestr. r& r' r( r) r r0 r )r r r/ �dirpath�fr r r r V s
��zfilebundlestore.writec C sV z t | �|�d��
}|�� W d � W S 1 sw Y W d S ty* Y d S w )Ns rb)r r0 r �IOError)r r r4 r r r r b s (� �zfilebundlestore.readN) r r r
r r* r. r0 r r r r r r r < s r c C sL g }| D ] }|r|dkr|� |� q|r |dkr |� |� q|� |� q|S )z�Formats `args` with Infinitepush replacements.
Hack to get `str.format()`-ed strings working in a BC way with
bytes.
s
{filename}s {handle})�append)�argsr/ �handle�formatted_args�argr r r �format_placeholders_argsj s r; c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�externalbundlestorec C s || _ || _|| _|| _dS )a6
`put_binary` - path to binary file which uploads bundle to external
storage and prints key to stdout
`put_args` - format string with additional args to `put_binary`
{filename} replacement field can be used.
`get_binary` - path to binary file which accepts filename and key
(in that order), downloads bundle from store and saves it to file
`get_args` - format string with additional args to `get_binary`.
{filename} and {handle} replacement field can be used.
N)�put_args�get_args�
put_binary�
get_binary)r r? r= r@ r> r r r r* | s
z externalbundlestore.__init__c C s< t jt�tj|�t jt jdd�}|�� \}}|j}|||fS )NT)�stdout�stderr� close_fds) �
subprocess�Popenr �rapplyr � tonativestr�PIPE� communicate�
returncode)r r7 �prA rB rJ r r r � _call_binary� s �
z externalbundlestore._call_binaryc C s� t �� �K}|�|� |�� |�d� t| j|jd�}| �| j g| �\}}}|dkr1t
d| ��|� � }t |�dkrF|d W d � S t
d| j |f ��1 sRw Y d S )Nr )r/ s& Failed to upload to external store: %s� s Bad output from %s: %s)
r �
namedtempfiler �flush�seekr; r= �namerL r? r �
splitlines�len)r r �tempr9 rJ rA rB � stdout_linesr r r r � s,
�
�� � ��zexternalbundlestore.writec C sr t �� �+}t| j|j|d�}| �| jg| �\}}}|dkr$td| ��|�� W d � S 1 s2w Y d S )N)r/ r8 r s* Failed to download from external store: %s) r rN r; r> rQ rL r@ r r )r r8 rT r9 rJ rA rB r r r r � s
�
��$�zexternalbundlestore.readN)r r r
r* rL r r r r r r r< { s
r< )NN)�
__future__r r r& rD �mercurial.noder �mercurial.pycompatr � mercurialr �mercurial.utilsr r � Exceptionr r �objectr r r; r< r r r r �<module> s
.