File: //usr/lib/python3/dist-packages/S3/__pycache__/CloudFront.cpython-310.pyc
o
�Qa�� � @ s� d dl mZ d dlZd dlZd dlZd dlmZ d dlmZmZm Z m
Z
z d dl m m
Z W n ey> d dlm
Z Y nw ddlmZ ddlmZ ddlT dd lmZmZmZmZmZmZ dd
lm Z m Z m Z m Z m Z dd l!m"Z" dd l#m#Z#m$Z$ dd
l%m%Z% ddl&m&Z& ej'dkZ(dZ)dde)i Z*dd� Z+dd� Z,G dd� de-�Z.G dd� de-�Z/G dd � d e-�Z0G d d � d e-�Z1G d d � d e-�Z2G d!d"� d"e-�Z3G d#d$� d$e-�Z4G d%d&� d&e-�Z5G d'd(� d(e-�Z6dS ))� )�absolute_importN)�datetime)�debug�info�warning�error� )�S3)�Config)�*)�getTreeFromXml�appendXmlTextNode�getDictFromTree�dateS3toPython� encode_to_s3�decode_from_s3)�getBucketFromHostname�getHostnameFromBucket� deunicodise�urlencode_string� convertHeaderTupleListToDict)�sign_string_v2)�S3Uri�S3UriS3)�ConnMan)�
SortedDict)� r z
2010-11-01z/%(api_ver)s/distribution�api_verc C s t j�| d � d S )N�
)�sys�stdout�write)�message� r# �//usr/lib/python3/dist-packages/S3/CloudFront.py�output' s r% c C s" d| � d�} td| |f � d S )Nz%s:� z%s %s)�ljustr% )�labelr" r# r# r$ �
pretty_output* s r) c @ �$ e Zd Zdd� Zdd� Zdd� ZdS )�DistributionSummaryc C s& |j dkr td|j ��| �|� d S )Nr+ z1Expected <DistributionSummary /> xml, got: <%s />)�tag�
ValueError�parse��self�treer# r# r$ �__init__@ s
z DistributionSummary.__init__c C s\ t |�| _| jd �� dk| jd< d| jv r*t| jd �tkr,| jd g| jd< d S d S d S )N�Enabled�true�CNAME)r r �lower�type�listr/ r# r# r$ r. E s
�zDistributionSummary.parsec C � t d| jd �S �Nzcf://%s�Id�r r �r0 r# r# r$ �uriK � zDistributionSummary.uriN��__name__�
__module__� __qualname__r2 r. r> r# r# r# r$ r+ / s r+ c @ s e Zd Zdd� Zdd� ZdS )�DistributionListc C �. t |�}|jdkrtd|j ��| �|� d S )NrD z.Expected <DistributionList /> xml, got: <%s />�r r, r- r. �r0 �xmlr1 r# r# r$ r2 Z �
zDistributionList.__init__c C sL t |�| _| jd �� dk| jd< g | _|�d�D ]
}| j�t|�� qd S )N� IsTruncatedr4 z.//DistributionSummary)r r r6 �
dist_summs�findall�appendr+ )r0 r1 � dist_summr# r# r$ r. ` s
�zDistributionList.parseN)rA rB rC r2 r. r# r# r# r$ rD N s rD c @ r* )� Distributionc C rE )NrO z*Expected <Distribution /> xml, got: <%s />rF rG r# r# r$ r2 v rI zDistribution.__init__c C s8 t |�| _t| jd �| jd< t|�d�d�| jd< d S )N�LastModifiedTimez.//DistributionConfig)r1 �DistributionConfig)r r r rQ �findr/ r# r# r$ r. | s
zDistribution.parsec C r9 r: r< r= r# r# r$ r> � r? zDistribution.uriNr@ r# r# r# r$ rO i s
rO c @ sF e Zd ZdZddei Zddd�Zdd� Zd d
� Zd d � Z d
d� Z
dS )rQ zq<DistributionConfig><S3Origin><DNSName/></S3Origin><CallerReference/><Enabled>true</Enabled></DistributionConfig>z0http://cloudfront.amazonaws.com/doc/%(api_ver)s/r Nc C sD |d u rt j}|d u rt|�}|jdkrtd|j ��| �|� d S )NrQ z0Expected <DistributionConfig /> xml, got: <%s />)rQ � EMPTY_CONFIGr r, r- r. rG r# r# r$ r2 � s
zDistributionConfig.__init__c C s t |�| _| jd �� dk| jd< d| jvrg | jd< t| jd �tkr-| jd g| jd< dd� | jd D �| jd< d| jvrDd| jd< d| jvrNd| jd< |�d �}|rzt |d
�}t|d �\|d < }|sotd |d � td
| �| jd< d S d | jd< d S )Nr3 r4 r5 c S s g | ]}|� � �qS r# )r6 )�.0�cnamer# r# r$ �
<listcomp>� s z,DistributionConfig.parse.<locals>.<listcomp>�Comment� �DefaultRootObjectz
.//Loggingr �Bucketz%Logging to unparsable bucket name: %szs3://%(Bucket)s/%(Prefix)s�Logging) r r r6 r7 r8 rL r r r )r0 r1 �
logging_nodes� logging_dict�successr# r# r$ r. � s&
zDistributionConfig.parsec C s t �d�}tj|jd< tdd|�}td| jd d |� td| jd |� | jd D ]
}td|�� |� q*| jd rCtd| jd |� td t| jd ��� |� t| jd
�rbtd
t| jd
�|� | jd r�t �d �}td t | jd �
� �|� td
| jd � � |� |� |� |S )NrQ �xmlns�S3OriginrX �DNSName�CallerReferencer5 rW r3 rY r[ rZ �Prefix)
�ET�ElementrQ r_ �attribr
r r6 �strr �bucket�objectrM )r0 r1 �s3orgrU �
logging_elr# r# r$ �get_printable_tree� s$
z%DistributionConfig.get_printable_treec C � t t�| �� ��S �N�r rd �tostringrl r= r# r# r$ � __unicode__� r? z DistributionConfig.__unicode__c C �$ t r tj| �� dd�S t�| �� �S �N�unicode)�encoding��PY3rd rp rl r= r# r# r$ �__str__� � zDistributionConfig.__str__)NN) rA rB rC rS �cloudfront_api_versionr_ r2 r. rl rq rx r# r# r# r$ rQ � s
rQ c @ r* )� Invalidationc C rE )Nr{ z*Expected <Invalidation /> xml, got: <%s />rF rG r# r# r$ r2 � rI zInvalidation.__init__c C � t |�| _d S rn �r r r/ r# r# r$ r. � � zInvalidation.parsec C �
t | j�S rn �rg r r= r# r# r$ rx � �
zInvalidation.__str__N�rA rB rC r2 r. rx r# r# r# r$ r{ � s r{ c @ r* )�InvalidationListc C rE )Nr� z.Expected <InvalidationList /> xml, got: <%s />rF rG r# r# r$ r2 rI zInvalidationList.__init__c C r| rn r} r/ r# r# r$ r.
r~ zInvalidationList.parsec C r rn r� r= r# r# r$ rx r� zInvalidationList.__str__Nr� r# r# r# r$ r� � s r� c @ sD e Zd Zddg fdd�Zdd� Zdd� Zdd � Zd
d � Zd d
� ZdS )�InvalidationBatchNc C sL |r|| _ n|s
d}d|t�t�� d�t�dd�f | _ g | _| �|� d S )N�0z%s.%s.%sz %Y%m%d%H%M%Si� i' )� referencer �strftime�now�random�randint�paths� add_objects)r0 r� � distributionr� r# r# r$ r2 s
�zInvalidationBatch.__init__c C s | j �|� d S rn )r� �extend)r0 r� r# r# r$ r� * s z InvalidationBatch.add_objectsc C s | j S rn )r� r= r# r# r$ �
get_reference- s z InvalidationBatch.get_referencec C sX t �d�}| jD ]}t|�dk s|d dkrd| }tdt|�|� qtd| j|� |S )Nr� r r �/�Pathrb )rd re r� �lenr
r r� )r0 r1 �pathr# r# r$ rl 0 s
z$InvalidationBatch.get_printable_treec C rm rn ro r= r# r# r$ rq 9 r? z InvalidationBatch.__unicode__c C rr rs rv r= r# r# r$ rx < ry zInvalidationBatch.__str__) rA rB rC r2 r� r� rl rq rx r# r# r# r$ r� s r� c @ s e Zd Zddd�ddd�ddd�ddd�ddd�ddd�dd d�dd d�dd
d�d � Zd Zd
Zdd� Zdd� Zg d
d
d
fdd�Zg g d
d
d
d
fdd�Z dd� Z
dd� Z dd� Z d0d d �Z
d d � Zd d!� Zd"d#� Zd
d
d
d
efd$d%�Zd1d&d'�Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd
S )2�
CloudFront�POSTrX )�method�resource�DELETEz /%(dist_id)s�GETz/%(dist_id)s/config�PUTz/%(dist_id)s/invalidationz(/%(dist_id)s/invalidation/%(request_id)s) �
CreateDist�
DeleteDist�GetList� GetDistInfo�
GetDistConfig�
SetDistConfig�
Invalidate� GetInvalList� GetInvalInfo� Nc C s
|| _ d S rn )�config)r0 r� r# r# r$ r2 U r� zCloudFront.__init__c C s4 | � d�}t|d �|d< |d jd rtd��|S )Nr� �data� dist_listrJ z3List is truncated. Ask s3cmd author to add support.)� send_requestrD r �NotImplementedError)r0 �responser# r# r$ r� \ s
zCloudFront.GetListc
C s� t � }d|jd< |�� |jd d< t|�|jd< ||jd< |d kr)|�� |jd< n||jd< |D ]}|jd �|�d krD|jd �|� q0|rNt|�|jd
< t|�}td | � | j d |d
�} t
| d �| d< | S )NTr3 r` ra rb rY rW r5 r r[ z&CreateDistribution(): request_body: %sr� )�bodyr� r� ) rQ r � host_namerg �
public_url�countrM r r r� rO )
r0 r> �
cnames_add�comment�logging�default_root_object� dist_configrU � request_bodyr� r# r# r$ �CreateDistributiond s&
� z CloudFront.CreateDistributionc C s |j dkr td| ��td| � | �|�}|d } |d kr#|| jd< |d kr,|| jd< |d kr5|| jd< |D ]}
| jd �|
�d krK| jd �|
� q7|D ] }
| jd �|
�d krl| jd �|
� | jd �|
�d ksZqN|d kr�|d
kr{d
| jd < nt|�| jd < | �|| |d d
�}|S )N�cf� Expected CFUri instead of: %s� Checking current status of %sr� r3 rW rY r5 r Fr[ �headers�etag) r7 r- r r� r� rM �remover r� ) r0 �cfurir� �
cnames_remover� �enabledr� r� r� �dcrU r# r# r$ �ModifyDistributiony s4
��� z CloudFront.ModifyDistributionc C s� |j dkr td| ��td| � | �|�}|d jd rdtd� d|d jd< | �||d |d d �}td
� td � | �|�}|d
}|jd dkrZ|jd dkrZtd� n
td� t�d� q>t d d�}|d d |d< | j
d|� � |d�}|S )Nr� r� r� r� r3 z)Distribution is ENABLED. Disabling first.Fr� r� z,Waiting for Distribution to become disabled.z+This may take several minutes, please wait.Tr� �Status�Deployedz Distribution is now disabledzStill waiting...�
�� ignore_case�if-matchr� )�dist_idr� ) r7 r- r r� r� r r� �time�sleepr r� r� )r0 r� r� �dr� r# r# r$ �DeleteDistribution� s6
�
�
�z CloudFront.DeleteDistributionc C �< |j dkr td| ��| jd|�� d�}t|d �|d< |S )Nr� r� r� �r� r� r� )r7 r- r� r� rO �r0 r� r� r# r# r$ r� � �
zCloudFront.GetDistInfoc C r� )Nr� r� r� r� r� r� )r7 r- r� r� rQ r� r# r# r$ r� � r� zCloudFront.GetDistConfigc C sn |d krt d� | �|�d d }t d| � t|�}t d| � tdd�}||d< | jd |�� ||d
�}|S ) Nz1SetDistConfig(): Etag not set. Fetching it first.r� r� zSetDistConfig(): Etag = %sz!SetDistConfig(): request_body: %sTr� r� r� )r� r� r� )r r� rg r r� r� )r0 r� r� r� r� r� r� r# r# r$ r� � s
�zCloudFront.SetDistConfigc C s� |d ur9|r|r9g }d| }|D ]&}|� |�s||kr1|r"|�|� |r0|�|d t|� � � q|�|� q|}| �|�} t|�dkr�z:t�� }
tt|
�d��} | �td� |�d �� W d � n1 sgw Y t
dt|� � t
d|
� W t d�� t y� Y t d��w g } | D ]@}
t
|
�� |d�}td | � | jd
|
�� t|�d �}|
�� |d < |d
dkr�t|d �j}|d |d< td| � | �|� q�| S )Nr� i� �wr z2Request to invalidate %d paths (max 999 supported)z"All the paths are now saved in: %sz Too many paths to invalidate)r� r� z%InvalidateObjects(): request_body: %sr� )r� r� r� �status�� r� r; �
request_idz!InvalidateObjects(): response: %s)�endswithrM r� �get_dist_name_for_bucket�Utils� mktmpfile�openr r! �joinr � Exception�ParameterErrorr� r� r r� rg r{ r )r0 r> r� �default_index_file� invalidate_default_index_on_cf�#invalidate_default_index_root_on_cf� new_paths�default_index_suffixr� �cfuris� tmp_filename�fp� responsesr� �
invalbatchr� �
inval_infor# r# r$ �InvalidateObjects� sP
�
� �� � z CloudFront.InvalidateObjectsc C r� )Nr� r� r� r� r� �
inval_list)r7 r- r� r� r� r� r# r# r$ r� � r� zCloudFront.GetInvalListc C sV |j dkr td| ��|�� d u rtd��| jd|�� |�� d�}t|d �|d< |S )Nr� r� z Expected CFUri with Request IDr� �r� r� r� � inval_status)r7 r- r� r� r� r{ r� r# r# r$ r� s
zCloudFront.GetInvalInfoc
C sZ |d u r t dd�}| j| }|rd|d< | �||||�}| �� } td|d |d f � | j�|d |d ||d � | j�� }
i } |
j| d < |
j | d
< t
|
� � �| d< |
� � | d < t
�| � td | � | d d
kr�t| �} |r�td| � tt| �� td| �|� � t�| �|�� | j||||d d�S | �| d dk s�| d dkr�t| ��| S )NTr� z
text/plainz content-typezsend_request(): %s %sr� r� r� r� �reasonr� zCloudFront: response: %ri� zRetrying failed request: %szWaiting %d sec...r )�retries�� i+ )r �
operations�create_request�get_connectionr �c�request� getresponser� r� r �
getheaders�readr �put�CloudFrontErrorr rt �
_fail_waitr� r� r� )
r0 �op_namer� r� r� r� r� � operationr� �conn�
http_responser� �er# r# r$ r� s:
zCloudFront.send_requestc C s� t |d ||d� }|stdd�}d|v r#d|vr |d |d< |d= d|vr1t�dt�� �|d< t| jj�dkrD| j�� | jj|d < | � |�}d
| jj
d | |d < i }||d< ||d
< |d |d< |S )Nr� r� Tr� �date�
x-amz-datez%a, %d %b %Y %H:%M:%S +0000r zx-amz-security-tokenzAWS �:�
Authorizationr� r� ) �cloudfront_resourcer r� r� �gmtimer� r� � access_token� role_refresh� sign_request�
access_key)r0 r� r� r� r� r� � signaturer� r# r# r$ r� 2 s* �
zCloudFront.create_requestc C s, |d }t tt|���}td||f � |S )Nr� z"CloudFront.sign_request('%s') = %s)r r r r )r0 r� �string_to_signr r# r# r$ r O s zCloudFront.sign_requestc C s t j| jjdd�}|S )NT)�ssl)r �getr� �cloudfront_host)r0 r� r# r# r$ r� U s zCloudFront.get_connectionc C s | j | d d S )Nr r )� _max_retries)r0 r� r# r# r$ r� Y s zCloudFront._fail_waitc
C s6 |j dkr|gS |j dkrtd| ��td| � tjd u r|| �� }i t_|d jD ]J}d}d|jv r?t|jd d �d }n d
|jv rZt|jd
d �d }|d t |�
� �� }nq*tj� |d �d u rjt � tj|< tj| �
|�� � q*td tj � ztj|�
� W S ty� } z
t|� td | ��d }~ww )
Nr� �s3z,CloudFront or S3 URI required instead of: %sz _get_dist_name_for_bucket(%r)r� rX r` ra r � CustomOriginz
dist_list: %sz>Unable to translate S3 URI to CloudFront distribution name: %s)r7 r� r r� r� r� rK r r r� rh r �set�addr> r� )r0 r> r� r� �
distListIndexr� r# r# r$ r� ] s6
��z#CloudFront.get_dist_name_for_bucketrn )NNN)rA rB rC r� r
r� r2 r� r� r� r� r� r� r� r� r� r� r� r� r r� r� r� r# r# r# r$ r� D sB �
�
.
$ r� c @ sn e Zd ZdZG dd� de�Ze� Zedd� �Zedd� �Z edd � �Z
ed
d � �Z ed d
� �Z edd� �Z
dS )�Cmdz3
Class that implements CloudFront commands
c @ s4 e Zd Zg Zg ZdZdZdZdZdd� Z dd� Z
dS )z Cmd.OptionsNc C s dd� t | �D �S )Nc S s g | ] }|� d �r|�qS )�cf_)�
startswith)rT �optr# r# r$ rV � s z+Cmd.Options.option_list.<locals>.<listcomp>)�dirr= r# r# r$ � option_list� r? zCmd.Options.option_listc C s t tj||� d S rn )�setattrr �options)r0 �option�valuer# r# r$ �
update_option� r? zCmd.Options.update_option) rA rB rC �
cf_cnames_add�cf_cnames_remove�
cf_comment� cf_enable�
cf_logging�cf_default_root_objectr r r# r# r# r$ �Options� s r$ c C s4 t t� �}g }| D ]}|�t|��}|�|� q |S rn )r� r
r� r r� )�argsr� r� �arg�urisr# r# r$ � _parse_args� s
zCmd._parse_argsc C s t t� �}| sn|�� }|d jD ][}d|jv r"t�|jd d �}nd|jv r1d|jd d }nd}td|� td|�� � td |jd � d
|jv rWtd d � |jd
�� td
|jd
� td|jd � t
d� qd S t � | �}|D ]�}|�
|�}|d }|jd }d|jv r�t�|jd d �}nd|jv r�d|jd d }nd}td|� td|�� � td |jd � d
|jv r�td d � |jd
�� td
|jd
� td|jd � td|jd � td|jd � td|jd p�d� td|d d � qud S )Nr� r` ra r z
http://%s/z <unknown>�Origin�DistId�
DomainNamer5 �CNAMEs�, r� r3 rX r� rQ rW �DfltRootObjectrY r[ �Disabled�Etagr� r� )r� r
r� rK r r �httpurl_to_s3urir) r> r� r% r r( r� )r% r� r� r� �originr� r� r� r# r# r$ r � sP
�
�zCmd.infoc C sh t t� �}g }| D ](}t|�}|jdkrtd| ��|�� r$td| ��|�� s,td��|�|� q |s8td��|D ]w}td| � |j |t
j j t
j j
t
j jt
j jd�}|d }|jd }td
� td t�|jd d
�� td|�� � td|jd � tdd�|jd �� td|jd � td|jd � td|jd � td|jd � td|d d � q:d S )Nr z@Distribution can only be created from a s3:// URI instead of: %sz4Use s3:// URI with a bucket name only instead of: %sz3CloudFront can only handle lowercase-named buckets.zNo valid bucket names foundz Creating distribution from: %s)r� r� r� r� r� rQ zDistribution created:r) r` ra r* r+ r, r- r5 rW r� r3 rY r0 r� r� )r� r
r r7 r� ri �is_dns_compatiblerM r r� r r r r r" r# r% r) r r1 r> r� )r% r� �bucketsr& r> r� r� r� r# r# r$ �create� sB
�
�z
Cmd.createc C sX t t� �}t�| �}|D ] }|�|�}|d dkr#td||d f � td| � q d S )Nr� � z(Distribution %s could not be deleted: %sr� zDistribution %s deleted)r� r
r r( r� r r% )r% r� r� r� r� r# r# r$ �delete� s
�z
Cmd.deletec C sd t t� �}t| �dkrtd��z t�| �d }W n ty# td��w |j|tjj tjj
tjj tjj tjj
tjjd�}|d dkrLtd||d f � td
| � |�|�}|d }|jd }td
t�|jd d �� td|�� � td|jd � td|jd � tdd�|jd �� td|jd � td|jd � td|jd � td|d d � d S ) Nr z7Too many parameters. Modify one Distribution at a time.r z No valid Distribution URI found.)r� r� r� r� r� r� r� r6 z)Distribution %s could not be modified: %sr� zDistribution modified: %sr� rQ r) r` ra r* r+ r� r, r- r5 rW r3 rY r0 r� r� )r� r
r� r� r r( �
IndexErrorr� r r r r r! r" r# r r% r� r r) r r1 r> r� )r% r� r� r� r� r� r# r# r$ �modify� s>
��
z
Cmd.modifyc
C s& t t� �}t�| �}g }|D ]7}|�� r |�t|�� q|�|�}z|d jd D ]}|�d� d|�
� |d g�� q)W q t yE Y qw |D ]H}t |�}|�
|�}|d j} | d d }
t|
t�rgt|
�nd } td
t|�� td | d � td | d
� td| � td| d d � td� qHd S )Nr� �InvalidationSummaryr� zcf:/r; r� r� r� r �URIr� �Created�
CreateTimez Nr of paths� Referencerb rX )r� r
r r( r� rM rg r� r r� r� r� r r� �
isinstancer8 r� r) r% ) r% r� r� �requestsr� r� �i�reqr� �str� � nr_of_pathsr# r# r$ � invalinfo s6
� �
�z
Cmd.invalinfoN)rA rB rC �__doc__ri r$ r � staticmethodr( r r5 r7 r9 rE r# r# r# r$ r � s
+
!
r )7�
__future__r r r� r� r r� r r r r �xml.etree.ElementTree�etree� ElementTreerd � ImportError�elementtree.ElementTreer r
�
Exceptions� BaseUtilsr r
r r r r r� r r r r r �Cryptor r r r r � version_inforw rz r r% r) ri r+ rD rO rQ r{ r� r� r� r r# r# r# r$ �<module> sH �
T 1 A