File: //usr/lib/python3/dist-packages/duplicity/backends/__pycache__/onedrivebackend.cpython-310.pyc
o
K�a< � @ s� d dl mZ d dl mZ d dlmZ d dlmZ d dlmZ d dlZd dl Z d dl
Z
d dl Z d dl Z
d dlmZ d dl
mZ d d l
mZ G d
d � d e
jj�ZG d d
� d
e�ZG dd� de�ZG dd� de�Ze
j�de� dS )� )�print_function)�division)�old_div)�input)�strN)�BackendException)�config)�logc @ sX e Zd ZdZdZdZdd� Zdd� Zdd � Zd
d � Z d d
� Z
dd� Z dd� Z dd� Z
dS )�OneDriveBackendz8Uses Microsoft OneDrive (formerly SkyDrive) for backups.z!https://graph.microsoft.com/v1.0/i c C sl t jj�| |� |j�d�| _tj� dd�}d|| jf | _
| jdkr't d��t j
dkr0t d��| �� d S ) N�/�
ONEDRIVE_ROOTz
me/drive/rootz%s:/%s/� zTYou did not specify a path. Please specify a path, e.g. onedrive://duplicity_backupsl
zQYour --volsize is bigger than 10 GiB, which is the maximum file size on OneDrive.)� duplicity�backend�Backend�__init__�path�lstrip� directory�os�environ�get�directory_onedrive_pathr r �volsize�initialize_oauth2_session)�self�
parsed_url�
onedrive_root� r �D/usr/lib/python3/dist-packages/duplicity/backends/onedrivebackend.pyr 7 s
�
� zOneDriveBackend.__init__c C s@ t j�d�}t j�d�}|r|rt||�| _d S t| j�| _d S )N�OAUTH2_CLIENT_ID�OAUTH2_REFRESH_TOKEN)r r r �ExternalOAuth2Session� http_client�DefaultOAuth2Session�API_URI)r � client_id�
refresh_tokenr r r r I s
z)OneDriveBackend.initialize_oauth2_sessionc C s� g }| j | j�d� d } | j�|�}|jdkrn#|�� |�� }d|vr,td| ��||d 7 }d|v r;|d }nnqdd � |D �S )
Nr z
:/childrenT� �valuez-Malformed JSON: expected "value" member in %sz@odata.nextLinkc S s g | ]}|d �qS )�namer )�.0�xr r r �
<listcomp>f s z)OneDriveBackend._list.<locals>.<listcomp>) r% r �rstripr# r � status_code�raise_for_status�jsonr )r �accum�next_url�response� responseJsonr r r �_listQ s(
��
�zOneDriveBackend._listc C s� |� d�}|�d��1}| jj| j| j | d dd�}|�� |jdd�D ] }|r.|�|� q%|� � W d � d S 1 s>w Y d S )N�UTF-8�wbz :/contentT)�streami )�
chunk_size)
�decode�openr# r r% r r0 � iter_content�write�flush)r �remote_filename�
local_path�fr4 �chunkr r r �_geth s
�
�
"�zOneDriveBackend._getc C s� |� d�}tj�|j�}t�� }| j�| jd �}|� � d|�
� v rD|�
� d �dd �}|rDt � d| � ||krDt
d|j||f ��t � dt�� | � |�� ��}t�� }| j| j | d }| j�|�}|� � t
�|j� d��} d | vr�t
d
||jf ��| d }
d } td | j�} |�| | j �}
t|
�d kr�n3dt|
� d| | t|
� d |f d�}t � d||d f � | jj|
||
d�}|� � | t|
�7 } q�t � dt�� | � W d � d S 1 s�w Y d S )Nr7 zme/drive?$select=quota�quota� remainingzBytes available: %dzWOut of space: trying to store "%s" (%d bytes), but only %d bytes available on OneDrive.zChecked quota in %fsz:/createUploadSession� uploadUrlzAFile "%s" cannot be uploaded: could not create upload session: %sr i � Tz%dzbytes %d-%d/%d� )zContent-Length�
Content-Rangez PUT %s %srI )�headers�datazPUT file in %fs)r; r r �getsizer* �timer# r r% r0 r1 r �Debugr r< r �post�loads�contentr � REQUIRED_FRAGMENT_SIZE_MULTIPLE�read�len�put)r � source_pathr@ � source_size�startr4 � available� source_file�url�
response_jsonrG �offset�desired_num_fragmentsrC rJ r r r �_puts sf
���
��
�� �"�zOneDriveBackend._putc C sD |� d�}| j�| j| j | �}|jdkr td| ��|�� d S )Nr7 r( z.File "%s" cannot be deleted: it does not exist)r; r# �deleter% r r/ r r0 �r r@ r4 r r r �_delete� s
�� zOneDriveBackend._deletec C s` |� d�}| j�| j| j | �}|jdkrddiS d|�� vr(td|�� ��d|�� d iS )Nr7 �� �size���z,Malformed JSON: expected "size" member in %s)r; r# r r% r r/ r1 r ra r r r �_query� s
��zOneDriveBackend._queryc C s | � � d S �N)r )r r r r �_retry_cleanup� s z OneDriveBackend._retry_cleanupN)�__name__�
__module__� __qualname__�__doc__r% rR r r r6 rD r_ rb rf rh r r r r r
0 s 9 r
c @ s< e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd d � Z d
S )�OneDriveOAuth2SessionzDA tiny wrapper for OAuth2Session that handles some OneDrive details.z(https://login.live.com/oauth20_token.srfc
C sT dt jd< z ddlm} || _W n ty$ } ztdt|� ��d }~ww d | _d S )N�TRUE�OAUTHLIB_RELAX_TOKEN_SCOPEr )�
OAuth2SessionzeOneDrive backend requires python-requests-oauthlib to be installed. Please install it and try again.
) r r �requests_oauthlibrp �
session_class� ImportErrorr r �session)r rp �er r r r � s
����
z OneDriveOAuth2Session.__init__c O � | j j|i |��S rg )rt r �r �args�kwargsr r r r � � zOneDriveOAuth2Session.getc O rv rg )rt rU rw r r r rU � rz zOneDriveOAuth2Session.putc O rv rg )rt rO rw r r r rO � rz zOneDriveOAuth2Session.postc O rv rg )rt r` rw r r r r` � rz z OneDriveOAuth2Session.deleteN)
ri rj rk rl �OAUTH_TOKEN_URIr r rU rO r` r r r r rm � s rm c sH e Zd ZdZej�d�ZdZdZ dZ
g d�Z � fdd�Z d d
� Z
� ZS ) r$ z?A possibly-interactive console session using a built-in API keyz%~/.duplicity_onedrive_oauthtoken.json�000000004C12E85Dz,https://login.live.com/oauth20_authorize.srfz*https://login.live.com/oauth20_desktop.srf)z
Files.ReadzFiles.ReadWritez User.Read�offline_accessc
s� t t| ��� d }z t| j��
}t�|�}W d � n1 s w Y W n ty> } z
t� d| � W Y d }~nd }~ww | j
| j | j | j
|d| j i| j| jd�| _|d ur_| j�| j� | j�|d �}|jdkrnd }|d u r�tj�� r|tj�� s�t�d| j � | jj| jdd�\}}t� td | � t� t d
�� � }| jj | j|d �}| j�|d �}|� � z!t| jd ��}t� ||� W d � W d S 1 s�w Y W d S t!y� } zt� d
| j|f � W Y d }~d S d }~ww d S )NzMCould not load OAuth2 token. Trying to create a new one. (original error: %s)r& )�scope� redirect_uri�token�auto_refresh_kwargs�auto_refresh_url�
token_updater�merc z�The OAuth2 token could not be loaded from %s and you are not running duplicity interactively, so duplicity cannot possibly access OneDrive.�touch)�displayz�In order to authorize duplicity to access your OneDrive, please open %s in a browser and copy the URL of the blank page the dialog leads to.zURL of the blank page: )�authorization_response�wz�Could not save the OAuth2 token to %s. This means you need to do the OAuth2 authorization process on every start of duplicity. Original error: %s)"�superr$ r r< �OAUTH_TOKEN_PATHr1 �load�IOErrorr �Errorrr � CLIENT_ID� OAUTH_SCOPE�OAUTH_REDIRECT_URIr{ r� rt r' r r/ �sys�stdout�isatty�stdin�
FatalError�authorization_url�OAUTH_AUTHORIZE_URI�printr �strip� fetch_tokenr0 �dump� Exception) r �api_urir� rB ru �user_info_responser� �state�
redirected_to�� __class__r r r � sx �������
�
�� �&������z DefaultOAuth2Session.__init__c
C s~ z!t | jd��}t�||� W d � W d S 1 sw Y W d S ty> } zt�d| j|f � W Y d }~d S d }~ww )Nr� z�Could not save the OAuth2 token to %s. This means you may need to do the OAuth2 authorization process again soon. Original error: %s)r< r� r1 r� r� r r� )r r� rB ru r r r r� A s &�����z"DefaultOAuth2Session.token_updater)ri rj rk rl r r �
expanduserr� r� r� r� r� r r� �
__classcell__r r r� r r$ � s � Dr$ c s e Zd ZdZ� fdd�Z� ZS )r" z?Caller is managing tokens and provides an active refresh token.c sB t t| ��� d|i}| j||d|i| jd�| _| j�| j� d S )Nr' r& )r� r� r� )r� r" r rr r{ rt r' )r r& r' r� r� r r r O s ���
z ExternalOAuth2Session.__init__)ri rj rk rl r r� r r r� r r" M s r" �onedrive)�
__future__r r �
past.utilsr �builtinsr r rM r1 r r� �duplicity.backendr �duplicity.errorsr r r r r r
�objectrm r$ r" �register_backendr r r r �<module> s&
*_