File: //lib/python3/dist-packages/awscli/customizations/__pycache__/s3uploader.cpython-310.pyc
o
.&�aj � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZ d dl m
Z
d dl m Z e�
e�ZG dd� de�ZG dd� de�ZG d d
� d
e
�ZdS ) � N)�TransferManager)�BaseSubscriber)�collections_abcc @ s e Zd Zdd� ZdZdS )�NoSuchBucketErrorc K s( | j jdi |��}t�| |� || _d S )N� )�fmt�format� Exception�__init__�kwargs)�selfr �msgr r �B/usr/lib/python3/dist-packages/awscli/customizations/s3uploader.pyr
s
zNoSuchBucketError.__init__zaS3 Bucket does not exist. Execute the command to create a new bucket
aws s3 mb s3://{bucket_name}N)�__name__�
__module__� __qualname__r
r r r r r r s r c @ sp e Zd ZdZedd� �Zejdd� �Z ddd�Zd d
� Zdd d �Z d
d� Z
dd� Z dd� Z ddd�Z
dS )�
S3Uploaderz�
Class to upload objects to S3 bucket that use versioning. If bucket
does not already use versioning, this class will turn on versioning.
c C s | j S )zO
Metadata to attach to the object(s) uploaded by the uploader.
)�_artifact_metadata)r r r r �artifact_metadata1 s z S3Uploader.artifact_metadatac C s& |d urt |tj�std��|| _d S )Nz(Artifact metadata should be in dict type)�
isinstancer �Mapping� TypeErrorr )r �valr r r r 8 s
NFc C sB || _ || _|p d | _|| _|| _|| _|s t| j�| _d | _d S �N)� bucket_name�prefix�
kms_key_id� force_upload�s3�transfer_managerr r )r � s3_clientr r r r r r r r r
> s
zS3Uploader.__init__c
C s� | j rt| j �dkrd�| j |�}| js&| �|�r&t�d�|�� | �|�S z3ddi}| jr7d|d< | j|d< | j r?| j |d< t
||�}| j � || j
|||g�}|�� | �|�W S tjjyz } z|jd d
}|d krtt| j
d ��|�d
}~ww )z�
Uploads given file to S3
:param file_name: Path to the file that will be uploaded
:param remote_path: be uploaded
:return: VersionId of the latest upload
r z{0}/{1}z:File with same data already exists at {0}. Skipping upload�ServerSideEncryption�AES256zaws:kms� SSEKMSKeyId�Metadata�Error�Code� NoSuchBucket)r N)r �lenr r � file_exists�LOG�debug�make_urlr r �ProgressPercentager �uploadr �result�botocore�
exceptions� ClientError�responser )r � file_name� remote_path�additional_args�print_progress_callback�future�ex�
error_coder r r r. P s@ �
�
�� ��zS3Uploader.uploadc C s* | � |�}|}|r|d | }| �||�S )z�
Makes and returns name of the S3 object based on the file's MD5 sum
:param file_name: file to upload
:param extension: String of file extension to append to the object
:return: S3 URL of the uploaded object
�.)�
file_checksumr. )r r4 � extension�filemd5r5 r r r �upload_with_dedup� s
z S3Uploader.upload_with_dedupc C s2 z | j j| j|d� W dS tjjy Y dS w )z�
Check if the file we are trying to upload already exists in S3
:param remote_path:
:return: True, if file exists. False, otherwise
)�Bucket�KeyTF)r � head_objectr r0 r1 r2 )r r5 r r r r) � s ��zS3Uploader.file_existsc C s d� | j|�S )Nz s3://{0}/{1})r r )r �obj_pathr r r r, � s �zS3Uploader.make_urlc C s� t |d��;}t�� }d}|�� }|�d� |�|�}t|�dkr0|�|� |�|�}t|�dks |�|� |�� W d � S 1 sCw Y d S )N�rbi r ) �open�hashlib�md5�tell�seek�readr( �update� hexdigest)r r4 � file_handlerG �
block_size�curpos�bufr r r r< � s
�
$�zS3Uploader.file_checksumc C s. | j jj}d�|| j|�}|rd�||�}|S )z�
This link describes the format of Path Style URLs
http://docs.aws.amazon.com/AmazonS3/latest/dev/UsingBucket.html#access-bucket-intro
z {0}/{1}/{2}z{0}?versionId={1})r �meta� endpoint_urlr r )r �key�version�baser/ r r r �to_path_style_s3_url� s
z S3Uploader.to_path_style_s3_url)NNFNr )r r r �__doc__�propertyr �setterr
r. r? r) r, r< rV r r r r r + s"
�
1r c @ s e Zd Zdd� Zdd� ZdS )r- c C s2 || _ || _ttj�|��| _d| _t� � | _
d S )Nr ) � _filename� _remote_path�float�os�path�getsize�_size� _seen_so_far� threading�Lock�_lock)r �filenamer5 r r r r
� s
zProgressPercentage.__init__c K sr | j �, | j|7 _| j| j d }tj�d| j| j| j|f � tj�� W d � d S 1 s2w Y d S )N�d z#
Uploading to %s %s / %s (%.2f%%))rd ra r` �sys�stdout�writer[ �flush)r r8 �bytes_transferredr �
percentager r r � on_progress� s ��� "�z ProgressPercentage.on_progressN)r r r r
rm r r r r r- � s r- )rF �loggingrb r] rg r0 �botocore.exceptions�s3transfer.managerr �s3transfer.subscribersr �
awscli.compatr � getLoggerr r* r r �objectr r- r r r r �<module> s
$