File: //usr/lib/python3/dist-packages/duplicity/backends/__pycache__/pcabackend.cpython-310.pyc
o
���`o1 � @ sn d dl mZ d dlZd dlZd dlmZ d dlmZ d dlmZ d dl Z G dd� dej
j �Z ej
�
de � dS ) � )�strN)�log)�util)�BackendExceptionc @ sj e Zd ZdZdd� Zdd� Zdd� Zdd � Zdd d �Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zd
S )�
PCABackendz
Backend for OVH PCA
c
C s( t jj�| |� zddlm} ddlm} W n ty* } ztdt |� ��d }~ww || _
|| _ i }dt j
v rLdt j
v rLt j
d |d< t j
d |d< n0d t j
vrUtd
��d t j
vr^td ��d
t j
vrgtd��t j
d |d< t j
d |d< t j
d
|d< i }dt j
v �r t j
d |d< t j
d dk�r dt j
v r�|�dt j
d i� dt j
v r�|�dt j
d i� dt j
v r�|�dt j
d i� dt j
v r�|�d t j
d i� d t j
v r�|�d t j
d i� d t j
v r�|�d t j
d i� d!t j
v r�|�d"t j
d! i� d#t j
v �r |�d$t j
d# i� d%t j
v �r |�d&t j
d% i� nd'|d< d t j
v �r.t j
d |d < d%t j
v �r>|�d&t j
d% i� ||d(< d|d)< || _d*d+� |j�d,�D �}|�d�| _|�rfd-d,�|� | _nd.| _d/}d0} d }
z|d6i | j��| _| j�| j�}
W n, |�y� Y n# t�y� } zt�d1|jj t |�f tj j � W Y d }~nd }~ww |
d u �r�t� d2| j � zt | |gg�} | jj!| j| d3� W d S t�y� } zt�d4|jj t |�f tj j � W Y d }~d S d }~ww |�r|
| �"� |k�rt�d5| j|
| �"� |f � d S d S d S )7Nr )�
Connection)�ClientExceptionzBPCA backend requires the python-swiftclient library.
Exception: %s�PCA_PREAUTHURL�PCA_PREAUTHTOKEN�
preauthurl� preauthtoken� PCA_USERNAMEz*PCA_USERNAME environment variable not set.� PCA_PASSWORDz*PCA_PASSWORD environment variable not set.� PCA_AUTHURLz)PCA_AUTHURL environment variable not set.�user�key�authurl�PCA_AUTHVERSION� auth_version�3�PCA_USER_DOMAIN_NAME�user_domain_name�PCA_USER_DOMAIN_ID�user_domain_id�PCA_PROJECT_DOMAIN_NAME�project_domain_name�PCA_PROJECT_DOMAIN_ID�project_domain_id�PCA_TENANTNAME� tenant_name�PCA_ENDPOINT_TYPE�
endpoint_type�
PCA_USERID�user_id� PCA_TENANTID� tenant_id�PCA_REGIONNAME� region_name�2�
os_options�retriesc S s g | ]}|d kr|�qS )� � )�.0�xr, r, �?/usr/lib/python3/dist-packages/duplicity/backends/pcabackend.py�
<listcomp>p s z'PCABackend.__init__.<locals>.<listcomp>�/z%s/r+ �PCAzX-Storage-Policy�Connection failed: %s %szCreating container %s)�headersz Container creation failed: %s %sz>Container '%s' exists but its storage policy is '%s' not '%s'.r, )#� duplicity�backend�Backend�__init__� swiftclientr r � ImportErrorr r �resp_exc�conn_cls�os�environ�update� conn_kwargs�path�split�pop� container�join�prefix�conn�head_container� Exceptionr �
FatalError� __class__�__name__� ErrorCode�connection_failed�Info�dict�
put_container�lower) �self�
parsed_urlr r �er@ r) � url_parts�policy�
policy_header�container_metadatar4 r, r, r/ r8 $ s� ���
� ����
����
��zPCABackend.__init__c C s& t || j�r|jdkrtjjS d S d S )Ni� )�
isinstancer; � http_statusr rM �backend_not_found)rS � operationrU r, r, r/ � _error_code� s
�zPCABackend._error_codec C s0 | j �| j| jt�|� tt�|j�d�� d S )N�rb)rG �
put_objectrD rF r �fsdecode�open�name)rS � source_path�remote_filenamer, r, r/ �_put� s �zPCABackend._putc C sd | � t�|��}|r0tt�|j�d��}|D ]}|�|� qW d � d S 1 s)w Y d S d S )N�wb)�unsealr ra rb rc �write)rS re �
local_path�body�f�chunkr, r, r/ �_get� s �"��zPCABackend._getNc C s� | j j| jd| jdd�}|d }|r:|d d }|d �d�}| j j| j||d| jdd�d }|r8|d �|� |s|d urEt||d �S |d S ) NFzpolicy_extra=true)� full_listingrA � query_string� ���rc �
version_id)�marker�version_markerro rA rp )rG �
get_containerrD rF �get�extend�filter)rS �ffilter�rv�listingrt ru r, r, r/ � __list_objs� s& � ���zPCABackend.__list_objsc s � fdd�� � � D �S )Nc s( g | ]}t �|d t� j�d� ��qS )rc N)r �fsencode�lenrF )r- �o�rS r, r/ r0 � s ( z$PCABackend._list.<locals>.<listcomp>)�_PCABackend__list_objsr� r, r� r/ �_list� s zPCABackend._listc C s | j �| j| jt�|� � d S �N)rG �
delete_objectrD rF r ra )rS �filenamer, r, r/ �_delete� s zPCABackend._deletec C s, | j �| j| jt�|� �}dt|d �iS )N�sizezcontent-length)rG � head_objectrD rF r ra �int)rS r� �sobjectr, r, r/ �_query� s zPCABackend._queryc
C s� z| j j| j| j| dd�\}}t�d| � |W S | jyo } zI|jdkrMt|j d �}t
|d�\}}t
|d�\}}d|||f } t�d|| f � nt� d |j j
t|�f tjj� W Y d }~d S W Y d }~d S d }~ww )
Ni )�resp_chunk_sizez"File %s was successfully unsealed.i� z Retry-After�< �
%dh%02dm%02dsz/File %s is being unsealed, operation ETA is %s.r3 )rG �
get_objectrD rF r rO r; r[ r� �http_response_headers�divmodrJ rK rL r rM rN )
rS re �_rk rU �duration�m�s�h�etar, r, r/ rh � s.
�
��
���zPCABackend.unsealc s� d}t ttj|��� | j� fdd�d�}d}|D ]1}t�|d �}|d }t�d||f � |d krCt�d
| � | �|d � d }q|d krId }q|rX| � � �}t
� |� |sLt�d
� dS )z�
This is called before downloading volumes from this backend
by main engine. For PCA, volumes passed as argument need to be unsealed.
This method is blocking, showing a status at regular interval
r� c s | d � v S �Nrc r, �r. ��u_remote_filenamesr, r/ �<lambda>� s z7PCABackend.pre_process_download_batch.<locals>.<lambda>�rz Frc �policy_retrieval_statezVolume %s. State : %s. �sealed� Launching unseal of volume %s.T� unsealingz%All volumes to download are unsealed.N) �list�mapr ra r� r rO �Noticerh �
unseal_status�time�sleep)rS �remote_filenames�retry_interval�objs�one_object_not_unsealedr� r� r� r, r� r/ �pre_process_download_batch� s( �
�z%PCABackend.pre_process_download_batchc
s� d}| j � fdd�d�}d}|D ]D}|d }t�|d �}|dkr9t�d | � t�d
| � | �|d � d }q|d krTt|d
�}t�d||f � ||krR|}d }qt|d�\} }
t| d�\} } d| | |
f } t�d| � |S )z:
Shows unsealing status for input volumes
Fc s t �| d �� v S r� )r ra r� r� r, r/ r� � s z*PCABackend.unseal_status.<locals>.<lambda>r� r r� rc r� z,Error: volume is still in sealed state : %s.r� Tr� �policy_retrieval_delayz%s available in %d seconds.r� r� z0Need to wait %s before all volumes are unsealed.) r� r ra r r� rh r� rO r� )
rS r� r� r� � max_durationr� r� r� r� r� r� r� �max_duration_etar, r� r/ r� � s. �zPCABackend.unseal_statusr� )rL �
__module__� __qualname__�__doc__r8 r^ rf rn r� r� r� r� rh r� r� r, r, r, r/ r s o
!r �pca)�builtinsr r= �duplicity.backendr5 r r �duplicity.errorsr r� r6 r7 r �register_backendr, r, r, r/ �<module> s u