File: //usr/lib/python3/dist-packages/duplicity/backends/__pycache__/pydrivebackend.cpython-310.pyc
o
��`V4 � @ s� d dl mZ d dl mZ d dlZd dlmZ d dlmZ d dlmZ d dl ZG dd� dej
j �Z ej
�
d e � ej
�
d
e � ej
�
d e � ej
j�g d �� dS )
� )�next)�strN)�log)�util)�BackendExceptionc @ sX e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dS )�PyDriveBackendz)Connect to remote store using PyDrive APIc
s t jj�| |� z dd l}ddlm} W n ty( } ztdt |� ��d }~ww i | _
d | _ d|j v rE|j d d | _ d| j ddd�| _
zddl
m} dd lm} dd
lm}m} W n= ty� } z1zddlm} dd lm} dd
lm}m} W n ty� } ztd t |� ��d }~ww W Y d }~nd }~ww z dd lm} d| _W n dd
lm}
ddl m } d| _Y dt j v r�t j d } | jr�| |j d |j! | dd�}
n| j"�#| �}|
|j d |j! |dd�}
|
�$|�%� � |dd�}|
|_&nZdt j v �r
|t j d dd�}|�'� nFdt j v �rLdt j v �rL|dd�}|�(t j d � |�)t j d � |j&d u �r6|�'� n
|j*�r?|�+� n|�,� |�-t j d � ntd��||�| _.| j �r]| j }n0| j.�/dd i��0� }|�ru|d d d d }n| j.�1d d i�}|�2� |d d d }|�3� |j4�5d!�}|D ]c� � �s��q�dd"| d# i}|�6| j
� | j.�/|��0� }t7� fd$d%�|D �d �}|d u �r� d&d |igd'�}|d d �6| j
� |�6| j
� | j.�1|�}| j �r�|j2d(did)� n|�2� |d }�q�|| _8i | _9d S )*Nr )�buildz}PyDrive backend requires PyDrive and Google API client installation.
Please read the manpage for setup details.
Exception: %s�driveID� teamDriveT)�corpora� teamDriveId�includeTeamDriveItems�supportsTeamDrives)�
GoogleAuth)� GoogleDrive��ApiRequestError�FileNotUploadedErrorzhPyDrive backend requires PyDrive installation. Please read the manpage for setup details.
Exception: %s)� SignedJwtAssertionCredentials)�ServiceAccountCredentials)�cryptF�GOOGLE_DRIVE_ACCOUNT_KEY�@z%https://www.googleapis.com/auth/drive)�scopes�< )� http_timeout�GOOGLE_DRIVE_SETTINGS)�
settings_filer �GOOGLE_SECRETS_FILE�GOOGLE_CREDENTIALS_FILEzoGOOGLE_DRIVE_ACCOUNT_KEY or GOOGLE_DRIVE_SETTINGS environment variable not set. Please read the manpage to fix.�qz#'Root' in parents and trashed=false�parents�id�title� i_am_in_root�/�'� ' in parents and trashed=falsec 3 s, � | ]}|d � kr|d dkr|V qdS )r# �mimeType�"application/vnd.google-apps.folderN� ��.0�item�� folder_namer* �C/usr/lib/python3/dist-packages/duplicity/backends/pydrivebackend.py� <genexpr>� s � ��z*PyDriveBackend.__init__.<locals>.<genexpr>r) )r# r( r! r ��param):� duplicity�backend�Backend�__init__�httplib2�apiclient.discoveryr � ImportErrorr r �
api_params�shared_drive_id�
query_args�
pydrive2.authr �pydrive2.driver �pydrive2.filesr r � pydrive.auth�
pydrive.drive�
pydrive.files�oauth2client.clientr � oldClient� oauth2client.service_accountr � oauth2clientr �os�environ�username�hostname�Signer� from_string� authorize�Http� credentials�CommandLineAuth�LoadClientConfigFile�LoadCredentialsFile�access_token_expired�Refresh� Authorize�SaveCredentialsFile�drive�ListFile�GetList�
CreateFile�Upload�Delete�path�split�updater �folder�id_cache)�self�
parsed_urlr8 r �er r r r r r r � account_keyrP �signer�gauth�parent_folder_id� file_list� file_in_root� folder_names�list_file_argsra �create_file_argsr* r. r0 r7 ! s� ���
� ��� ���
��� �
�
� �
�
zPyDriveBackend.__init__c
C s� zddl m} W n ty ddlm} Y nw t�|�}|| jv r�| j| }| j�d|i�}z*|d |krX|d d sX|d D ]}|d | j krWt
� d||f � | W S q@W n |yu } z|j d j
jd krk� W Y d }~nd }~ww t
� d
||f � | j|= d || j f }d }||d
�} | �| j� | j�| ��� }
t|
�dkr�t
�td�|f � n |
r�|
d d }|
d d | j|< t
� d||f � |
d S t
� d|f � d S )Nr )r r" r# �labels�trashedr! z7PyDrive backend: found file '%s' with id %s in ID cachei� zCPyDrive backend: invalidating '%s' (previously ID %s) from ID cachez0title='%s' and '%s' in parents and trashed=falsez>items(title,id,fileSize,downloadUrl,exportLinks),nextPageToken�r �fields� z,PyDrive backend: multiple files called '%s'.zFPyDrive backend: found file '%s' with id %s on server, adding to cachez:PyDrive backend: file '%s' not found in cache or on server)r@ r r: rC r �fsdecoderb rX r[ ra r �Info�args�resp�statusr` r; rY rZ �len�
FatalError�_) rc �filenamer �file_id�
drive_file�parent�errorr rr rm �flistr* r* r0 � file_by_name� s^ �
�
�� ����
��zPyDriveBackend.file_by_namec C s | � |�}|d u r dS |d S )N� r" )r� )rc r| r~ r* r* r0 �
id_by_name� s
zPyDriveBackend.id_by_namec C s� t �|�}| �|�}|d u r1|d| jd�gd�}|d d �| j� | j�|�}t� d|f � n t� d||d f � |�
t �|j �� | j rQ|j
d d
id � n|�
� |d | j|< d S ) Nzdrive#fileLink)�kindr" )r# r! r! r z'PyDrive backend: creating new file '%s'z:PyDrive backend: replacing existing file '%s' with id '%s'r" r Tr2 )r rt r� ra r` r; rX r[ r ru �SetContentFile�namer< r\ rb )rc � source_path�remote_filenamer~ rn r* r* r0 �_put� s&
�� �zPyDriveBackend._putc C s | � |�}|�t�|j�� d S )N)r� �GetContentFiler rt r� )rc r� �
local_pathr~ r* r* r0 �_get� s
zPyDriveBackend._getc C s~ d| j d dd�}|�| j� | j�|��� }tdd� |D ��}t| j� � �D ]}||vr:| �
|�d ur:|� |� q(t|�S )Nr& r' z items(title,id),nextPageTokenrq c s s � | ]}|d V qdS )r# Nr* r+ r* r* r0 r1 � s � z'PyDriveBackend._list.<locals>.<genexpr>) ra r` r; rX rY rZ �set�listrb �keysr� �add)rc rm � drive_files� filenamesr| r* r* r0 �_list� s �
�zPyDriveBackend._listc C st | � |�}|dkrt�dt�|�f � d S | jr+| jjj� � j
|ddid�� � d S | jjj� � j
|d�� � d S )Nr� z2File '%s' does not exist while trying to delete itr T)�fileIdr3 )r� ) r� r �Warnr rt r< rX �auth�service�files�delete�execute)rc r| r} r* r* r0 �_delete� s
$ zPyDriveBackend._deletec C s2 | � |�}|d u rd}d|iS t|d �}d|iS )N����fileSize�size)r� �int)rc r| r~ r� r* r* r0 �_query s
�zPyDriveBackend._queryc C sd z
ddl m}m} W n ty ddlm}m} Y nw t||�r%tjjS t||�r.tjj S tjj
S )Nr r ) r@ r r r: rC �
isinstancer � ErrorCode�backend_not_found�backend_permission_denied�
backend_error)rc � operationr� r r r* r* r0 � _error_code s �
zPyDriveBackend._error_codeN)
�__name__�
__module__� __qualname__�__doc__r7 r� r� r� r� r� r� r� r� r* r* r* r0 r s v0 r �pydrive�
pydrive+gdocs�gdocs)r� r� r� )�builtinsr r rH r4 r r �duplicity.errorsr �duplicity.backendr5 r6 r �register_backend� uses_netloc�extendr* r* r* r0 �<module> s w