File: //usr/lib/python3/dist-packages/duplicity/backends/__pycache__/dpbxbackend.cpython-310.pyc
o
�~�_�O � @ s d dl mZ d dl mZ d dlmZ e�� d dlmZ d dlmZ d dl Z d dl
Z
d dl Z d dl Z d dl
Z
d dlZd dlZd dlZd dlZd dlmZmZ d dlmZ d d lmZ d d
lmZ d dlZd Z d Z e � d
�Z dd� Z ddd�Z!G dd� dej"j#�Z$ej"�%de$� dS )� )�print_function)�division)�standard_library)�input)�strN)�log�config)�progress��BackendException)�ConnectionErrori i z \([0-9]+\)\.[^\.]+$c C sZ t �d| f � t�� }tj|d� |�d� |�� D ] }t �d|�� � q|� � d S )NzException [%s]:)�filer z| )
r �Error�io�StringIO� traceback� print_exc�seek� readlines�rstrip�close)�e�f�s� r �@/usr/lib/python3/dist-packages/duplicity/backends/dpbxbackend.py�
log_exception? s
r Tc C s dd� }|S )z6a decorator for handling authentication and exceptionsc s � fdd�}� j |_ |S )Nc
sx z � | g|�R � W S t y } z t|� td|f ��d }~w ty; } zt|� t�d|f tjj� � d }~ww )Nzdpbx api error "%s"zdpbx code error "%s")�ApiErrorr r � Exceptionr r � ErrorCode�backend_code_error)�self�argsr �r r r �wrapperL s ���z*command.<locals>.decorate.<locals>.wrapper)�__doc__)r r$ r r# r �decorateK s zcommand.<locals>.decorater )�login_requiredr& r r r �commandI s r( c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
e � dd� �Z dd� Z
dd� Ze � dd� �Ze � dd� �Ze � dd� �Ze � d d � �Ze � d d � �Zd d!� Zd"S )#� DPBXBackendz-Connect to remote store using Dr*pB*x servicec
C s� t jj�| |� z*ddlma ddlmamam a ddl
m a m a m
a
mamamama ddlma W n tyF } ztdt|� ��d }~ww d | _d | _d | _| �� d S )Nr )�Dropbox)� AuthError�
BadInputErrorr )�UploadSessionCursor�
CommitInfo� WriteMode�GetMetadataError� DeleteError�UploadSessionLookupError�ListFolderError)�DropboxOAuth2FlowNoRedirectzvThis backend requires the dropbox package version 6.9.0
To install use "sudo pip install dropbox==6.9.0"
Exception: %s)� duplicity�backend�Backend�__init__�dropboxr* �dropbox.exceptionsr+ r, r �
dropbox.filesr- r. r/ r0 r1 r2 r3 �
dropbox.oauthr4 � ImportErrorr r � api_account�
api_client� auth_flow�login)r! �
parsed_urlr r r r r8 _ s $��� zDPBXBackend.__init__c C s6 z| j �� }t�d| � W dS t�d� Y dS )NzUser authenticated as ,%sTzUser not authenticatedF)r? �users_get_current_accountr �Debug)r! �accountr r r �user_authenticated} s
z DPBXBackend.user_authenticatedc C s t j�dd �S )N�DPBX_ACCESS_TOKEN)�os�environ�get�r! r r r �load_access_token� s z DPBXBackend.load_access_tokenc C s t d| ��)Nz<dpbx: Please set DPBX_ACCESS_TOKEN="%s" environment variabler
)r! � access_tokenr r r �save_access_token� s �z DPBXBackend.save_access_tokenc
C s t �d� dD ]
}|tjvrtd| ��qtjd }tjd }tj�� r)tj�� s1t � dt j
j � t ||�}t �
d� |�� }t� td� td | � td
� td � td� td ��� }zt �
d
| � |�|�}W n ty~ } ztd| ��d }~ww t �d� | �|j� d S )Nz#dpbx: trying to obtain access token)� DPBX_APP_KEY�DPBX_APP_SECRETz%dpbx: %s environment variable not setrO rP z5dpbx error: cannot interact, but need human attentionzdpbx,auth_flow.start()�H------------------------------------------------------------------------z
1. Go to: z22. Click "Allow" (you might have to log in first).z 3. Copy the authorization code.z#Enter the authorization code here: zdpbx,auth_flow.finish(%s)z'dpbx: Unable to obtain access token: %sz dpbx: Authentication successfull)r �InforH rI r �sys�stdout�isatty�stdin�
FatalErrorr �backend_command_errorr4 rD �start�printr �strip�finishr rN rM ) r! �env_var�app_key�
app_secretr@ �
authorize_url� auth_code�
authresultr r r r �obtain_access_token� s<
�
�
��
z DPBXBackend.obtain_access_tokenc
C s� | � � d u r
| �� t| � � �| _d | _zt�d� | j�� | _t�d| j � W n" tt fyK } zt�d| � t�
d� | �� t d��d }~ww t�
d| jj j
� d S )Nz'dpbx,users_get_current_account([token])zdpbx,%szdpbx,exception: %sz>dpbx: Authentication failed. Trying to obtain new access tokenz3dpbx: Please update DPBX_ACCESS_TOKEN and try againz&dpbx: Successfully authenticated as %s)rL rc r* r? r> r rD rC r, r+ rR r �name� display_name)r! r r r r rA � s$
��
�zDPBXBackend.loginc C sv t |t�r3|j}t |t�r |�� r |�� �� rtjj S d S t |t
�r5|� � r7|j� � }|�� r9tjj S d S d S d S d S �N)
�
isinstancer �errorr0 �is_path�get_path� is_not_foundr r �backend_not_foundr1 �is_path_lookup�get_path_lookup)r! � operationr �err�lookupr r r � _error_code� s
�
�zDPBXBackend._error_codec C s� t j�| jj�d��}dtj�||�� �� � }tj�
|j �}t �
d|� |tk r0| �||�}n| �||�}|j|krDtd|j|f ��|j|krRtd|j|f ��d S )N�/r z-dpbx: result path mismatch: %s (expected: %s)z-dpbx: result size mismatch: %s (expected: %s))�urllib�parse�unquoterB �path�lstriprH �join�decoder �getsizerd r �report_transfer�DPBX_UPLOAD_CHUNK_SIZE�put_file_small�put_file_chunked� path_displayr �size)r! � source_path�remote_filename�
remote_dir� remote_path� file_size� res_metadatar r r �_put� s
�
��zDPBXBackend._putc C s� | � � s| �� tj�|j�}|�d�}z,t�d||f � | j j
|� � |t j
dd dd�}t�d| � t�||� |W |�� S |�� w )N�rbz!dpbx,files_upload(%s, [%d bytes])FT��mode�
autorename�client_modified�mutezdpbx,files_upload(): %s)rF rA rH rw r{ rd �openr rD r? � files_upload�readr/ � overwriter r| r )r! r� r� r� r r� r r r r~ � s
� zDPBXBackend.put_file_smallc
C s� | � � s| �� tj�|j�}|�d�}�zb|�t�}t �
dt |�|f � | j �
|�}t �
d| � t|j|�� �}t|tjdd dd�}d } t�|�� |� d }
t} tj} d}
|
r]| �sWz}|
d ure|
|_|�� |jkrr|�|j� |�| �}|�� |k}
|
s�t |�dkr�W qXd }
t} tj} |
s�t |�dks�J �t �
dt |�|jf � | j �||j|j� nt �
d t |�|jf � | j �|||�} |�� |_t �
d
|j|f � t�|j|� W nx t �y } z/|j }t |t ��r|� � �r|�!� j"}t �
d |j|f � |
d u�r t#d ��|}
W Y d }~qX� d }~w t$�yR } z0t �
d
| � | d8 } | � � �s2| �� | dk�r8� t �%d� t&�'d� td } d }
W Y d }~qXd }~ww |
r]| r]|�� |k�rbt#d��t �
d| � t�|�� |� | W |�(� S |�(� w )Nr� z6dpbx,files_upload_session_start([%d bytes]), total: %dz%dpbx,files_upload_session_start(): %sFTr� r z8dpbx,files_upload_sesssion_append([%d bytes], offset=%d)z8dpbx,files_upload_sesssion_finish([%d bytes], offset=%d)zprogress: %d of %dzEdpbx,files_upload_session_append: incorrect offset: %d (expected: %s)z dpbx: unable to chunk uploadz$dpbx,files_upload_session_append: %s� z'dpbx: sleeping a bit before chunk retry� � zdpbx: something wrongz'dpbx,files_upload_sesssion_finish(): %s))rF rA rH rw r{ rd r� r� r} r rD �lenr? �files_upload_session_startr- �
session_id�tellr. r/ r� r r| r � num_retries�offsetr �files_upload_session_append�files_upload_session_finishr rh rg r2 �is_incorrect_offset�get_incorrect_offset�correct_offsetr r rR �time�sleepr )r! r� r� r� r �buf�
upload_sid�
upload_cursor� commit_infor� �requested_offset�current_chunk_size� retry_number�is_eofr rh �
new_offsetr r r r � s�
� �
�� ��
�
�
�
���Lz DPBXBackend.put_file_chunkedc C s | � � s| �� tj�| jj�d��}dtj� ||�
� �� � }t �
d| � | j�|�\}}t �
d|||f � |j}d }t�d|� z&|�d�}|�t�D ]} |�| � t�|�� |� qOW |rf|�� |�� n |rq|�� |�� w tj�|j�}
|
|kr�td|
|f ��|� � d S )Nrs zdpbx,files_download(%s)z dpbx,files_download(%s): %s, %sr �wbz(dpbx: wrong file size: %d (expected: %d)) rF rA rt ru rv rB rw rx rH ry rz r r rD r? �files_downloadr� r r| r� � iter_content�DPBX_DOWNLOAD_BUF_SIZE�writer� r r{ rd r �setdata) r! r� �
local_pathr� r� r� �http_fdr� �to_fd�c�
local_sizer r r �_getl s<
�
�
�
� zDPBXBackend._getc
C s | � � s| �� dtj�| jj�d���� }t �
d| � g }z)| j � |�}t �
d||f � |�
dd� |jD �� |js@n| j �|j�}q1W n/ tyx } z#t|jt�rm|j�� rm|j�� �� rmt �
d||f � n� W Y d }~nd }~ww | �|� |S )Nrs zdpbx.files_list_folder(%s)zdpbx.list(%s): %sTc S s g | ]}|j �qS r )rd )�.0�entryr r r �
<listcomp>� s z%DPBXBackend._list.<locals>.<listcomp>z)dpbx.list(%s): ignore missing folder (%s))rF rA rt ru rv rB rw rx r r rD r? �files_list_folder�extend�entries�has_more�files_list_folder_continue�cursorr rg rh r3 ri rj rk �check_renamed_files)r! r� �res�respr r r r �_list� s4 � � ���
zDPBXBackend._listc C s^ | � � s| �� tj�| jj�d��}dtj� ||�
� �� � }t �
d| � | j�|� d S )Nrs zdpbx.files_delete(%s))rF rA rt ru rv rB rw rx rH ry rz r r rD r? � files_delete)r! �filenamer� r� r r r �_delete� s zDPBXBackend._deletec C s t �d� dS )z0close backend session? no! just "flush" the dataz
dpbx.close():N)r rD rK r r r �_close� s zDPBXBackend._closec C sv | � � s| �� tj�| jj�d��}dtj� ||�
� �� � }t �
d| � | j�|�}t �
d||f � d|jiS )Nrs zdpbx.files_get_metadata(%s)z dpbx.files_get_metadata(%s): %sr� )rF rA rt ru rv rB rw rx rH ry rz r r rD r? �files_get_metadatar� )r! r� r� r� �infor r r �_query� s
zDPBXBackend._queryc C s� | � � s| �� dd� |D �}t|�dkrd S t�d� t�d� t�d� t�d� t�d� t�d � t�d� t�d
� t�d � t�d� t�d � |D ] }t�d
| � qPt�d� t�d� t�d� d S )Nc S s g | ] }t �|�d ur|�qS rf )�DPBX_AUTORENAMED_FILE_RE�search)r� �xr r r r� � s z3DPBXBackend.check_renamed_files.<locals>.<listcomp>r rQ zGWarning! It looks like there are automatically renamed files on backendzAThey were probably created when using older version of duplicity.� zIPlease check your backup consistency. Most likely you will need to choosezMlargest file from duplicity-* (number).gpg and remove brackets from its name.z?These files are not managed by duplicity at all and will not bez removed/rotated automatically.zAffected files:z %sz.In any case it's better to create full backup.)rF rA r� r �Warn)r! � file_list�bad_listr� r r r r� � s*
z DPBXBackend.check_renamed_filesN)�__name__�
__module__� __qualname__r% r8 rF rL rN rc rA rr r( r� r~ r r� r� r� r� r� r� r r r r r) \ s0
r
"
r) �dpbx)T)&�
__future__r r �futurer �install_aliases�builtinsr r r rH �rerS r� r �urllib.requestrt � urllib.parse� urllib.errorr5 r r r �duplicity.errorsr �requests.exceptionsr �duplicity.backendr} r� �compiler� r r( r6 r7 r) �register_backendr r r r �<module> s<