File: //lib/python3.6/site-packages/awscli/customizations/gamelift/__pycache__/uploadbuild.cpython-36.pyc
3
�T_� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ d dlm Z d dl
m Z G dd� de �Z dd� Z
d d
� ZG d d � d e�ZdS )
� N)�
S3Transfer)� BasicCommand)�human_readable_sizec @ sH e Zd ZdZdZdddd�dddd�d dd
d�d d d
d�gZdd� ZdS )�UploadBuildCommandz upload-buildz#Upload a new build to AWS GameLift.�nameTzThe name of the build)r ZrequiredZ help_textz
build-versionzThe version of the buildz
build-rootz8The path to the directory containing the build to uploadzoperating-systemFz&The operating system the build runs onc C sP | j jd|j|j|jd�}t|j�s:tjj d|j � dS |j
|j d�}|j rX|j |d< |j
f |�}|d d }|j|d �}|d
}|d d }|d d
} |d }
|d } |d } | j jd|
| | |j|jd�}
t|
�}zFtjd| �\}}t||j� |j||| t|d|j d d�d� W d tj|� tj|� X tjj d|j|f � dS )NZgamelift)� region_name� endpoint_url�verifyzHFail to upload %s. The build root directory is empty or does not exist.
� )�NameZVersionZOperatingSystemZBuild�BuildId)r ZUploadCredentialsZStorageLocationZBucketZKeyZ AccessKeyIdZSecretAccessKeyZ SessionTokenZs3)Zaws_access_key_idZaws_secret_access_keyZaws_session_tokenr r z%s.zipz
Uploading �:)�label)�callbackz6Successfully uploaded %s to AWS GameLift
Build ID: %s
r )Z_sessionZ
create_clientZregionr Z
verify_ssl�validate_directory�
build_root�sys�stderr�writer Z
build_versionZoperating_systemZ create_buildZrequest_upload_credentialsr �tempfileZmkstemp�
zip_directoryZ upload_file�ProgressPercentage�os�close�remove�stdout)�self�argsZparsed_globalsZgamelift_clientZcreate_build_kwargsZresponseZbuild_idZupload_credentialsZbucket�keyZ
access_keyZ
secret_keyZ
session_tokenZ s3_clientZs3_transfer_mgr�fdZtemporary_zipfile� r �!/usr/lib/python3.6/uploadbuild.py� _run_main) sX
z UploadBuildCommand._run_mainN)�__name__�
__module__� __qualname__�NAMEZ DESCRIPTIONZ ARG_TABLEr" r r r r! r s
r c C s� t jj|�}t| d��~}tj|dtjd�}tj|��V}xNt j |�D ]@\}}}x4|D ],}t jj
||�} t jj | |�}
|j | |
� qRW qBW W d Q R X W d Q R X d S )N�wb�wT)
r �path�abspath�open�zipfileZZipFileZ ZIP_DEFLATED�
contextlib�closing�walk�join�relpathr ) Z zipfile_name� source_root�fZzip_fileZzf�root�dirs�files�filename� full_pathZ
relative_pathr r r! r n s
r c C s. | sdS x t j| �D ]\}}}|rdS qW dS )NFT)r r/ )r2 r) r5 r6 r r r! r { s r c @ s e Zd Zddd�Zdd� ZdS )r Nc C sD || _ || _| jd kr | j | _ttjj|��| _d| _tj � | _
d S )Nr ) Z _filename�_label�floatr r) �getsize�_size� _seen_so_far� threadingZLock�_lock)r r7 r r r r! �__init__� s
zProgressPercentage.__init__c
C sn | j �^ | j|7 _| jdkr`| j| j d }tjjd| jt| j�t| j�|f � tjj� W d Q R X d S )Nr �d z
%s %s / %s (%.2f%%)) r? r= r< r r r r9 r �flush)r Z bytes_amountZ
percentager r r! �__call__� s
zProgressPercentage.__call__)N)r# r$ r% r@ rC r r r r! r � s
r )r> r- r r r r, Z
s3transferr Z awscli.customizations.commandsr Z awscli.customizations.s3.utilsr r r r �objectr r r r r! �<module>
s T