File: //lib/python3.6/site-packages/awscli/customizations/ecs/__pycache__/deploy.cpython-36.pyc
3
�T_�D � @ s� d dl Z d dlZd dlZd dlZd dlmZmZ d dlmZ d dl m
Z
d dl m Z m
Z
d dlmZ dZdZd ZG d
d � d e�ZG d d
� d
�ZG dd� d�ZG dd� d�ZdS )� N)�compat�config)� ClientError)� compat_open)�
exceptions� filehelpers)� BasicCommand�
� ih c @ s� e Zd ZdZdZdddd�dddd�d d
dd�d d d
d�ddd
d�ddd
d�gZdZdZdZdZ dd� Z
dd� Z dd� Z d d � Z
d d � Zd d!� Zd"d#� Zd$S )%� ECSDeployZdeploya Deploys a new task definition to the specified ECS service. Only services that use CodeDeploy for deployments are supported. This command will register a new task definition, update the CodeDeploy appspec with the new task definition revision, create a CodeDeploy deployment, and wait for the deployment to successfully complete. This command will exit with a return code of 255 if the deployment does not succeed within 30 minutes by default or up to 10 minutes more than your deployment group's configured wait time (max of 6 hours).�servicezJThe short name or full Amazon Resource Name (ARN) of the service to updateT)�nameZ help_textZrequiredztask-definitionz�The file path where your task definition file is located. The format of the file must be the same as the JSON output of: <codeblock>aws ecs register-task-definition --generate-cli-skeleton</codeblock>zcodedeploy-appspeca The file path where your AWS CodeDeploy appspec file is located. The appspec file may be in JSON or YAML format. The <code>TaskDefinition</code> property will be updated within the appspec with the newly registered task definition ARN, overwriting any placeholder values in the file.�clusterz�The short name or full Amazon Resource Name (ARN) of the cluster that your service is running within. If you do not specify a cluster, the "default" cluster is assumed.Fzcodedeploy-applicationz�The name of the AWS CodeDeploy application to use for the deployment. The specified application must use the 'ECS' compute platform. If you do not specify an application, the application name <code>AppECS-[CLUSTER_NAME]-[SERVICE_NAME]</code> is assumed.zcodedeploy-deployment-groupa* The name of the AWS CodeDeploy deployment group to use for the deployment. The specified deployment group must be associated with the specified ECS service and cluster. If you do not specify a deployment group, the deployment group name <code>DgpECS-[CLUSTER_NAME]-[SERVICE_NAME]</code> is assumed.z6Successfully registered new ECS task definition {arn}
z%Successfully created deployment {id}
z8Successfully deployed {task_def} to service '{service}'
zcustomization/ecs-deployc C s� | j |j|j�\}}t| j||| j�}| j||�| _| jjd|j |j
t j | jd�d�}| j
|� | jj� | _| j||�| _| j||� d S )NZ
codedeploy)�user_agent_extra)� region_name�verifyr )�_load_file_argsZtask_definitionZcodedeploy_appspec� ECSClientZ_session�USER_AGENT_EXTRA�_get_resource_names� resources�
create_client�region�
verify_sslr �Config� _validate_code_deploy_resources�
_cd_validator�get_deployment_wait_time� wait_time�_register_task_def� task_def_arn� _create_and_wait_for_deployment)�self� parsed_args�parsed_globals�register_task_def_kwargs� appspec_objZecs_client_wrapperZcodedeploy_client� r' � /usr/lib/python3.6/deploy.py� _run_mainr s
zECSDeploy._run_mainc C s� t ||�}|j| j� |j| jd | jd �}tjj| jj |d�� |j
|| j � | jd }tjj| j j | j|d�� tjj
� d S )N�app_name�deployment_group_name)�idr )Ztask_defr )� CodeDeployer�update_task_def_arnr �create_deploymentr �sys�stdout�write�MSG_CREATED_DEPLOYMENT�format�wait_for_deploy_successr � MSG_SUCCESS�flush)r" �clientZappspecZdeployer�
deployment_id� service_namer' r' r( r! � s
z)ECSDeploy._create_and_wait_for_deploymentc C sl t jjt jj|��}y t|��
}|j� S Q R X W n6 tttfk
rf } zt j
||d��W Y d d }~X nX d S )N)� file_path�error) �os�path�
expandvars�
expanduserr �read�OSError�IOError�UnicodeDecodeErrorr Z
FileLoadError)r" r; � full_path�f�er' r' r( �_get_file_contents� s
z ECSDeploy._get_file_contentsc C sR |j � }|d }|d }tj|||j�}tj|||j�}||d ||d ||d�S )Nr: � cluster_name� service_arn� cluster_arn)r rJ r rK r* r+ )�get_service_detailsr Z get_app_nameZcodedeploy_applicationZget_deploy_group_nameZcodedeploy_deployment_group)r" �args�
ecs_client�service_detailsr: rI Zapplication_namer+ r' r' r( r � s z ECSDeploy._get_resource_namesc C s0 | j |�}tj|�}| j |�}tj|�}||fS )N)rH �json�loadsr Z
parse_appspec)r" Z task_def_argZ appspec_argZtask_def_stringr% Zappspec_stringr& r' r' r( r � s
zECSDeploy._load_file_argsc C s: |j |�}|d d }tjj| jj|d�� tjj� |S )N�taskDefinitionZtaskDefinitionArn)Zarn)�register_task_definitionr0 r1 r2 �MSG_TASK_DEF_REGISTEREDr4 r7 )r" Ztask_def_kwargsrN �responser r' r' r( r � s
z ECSDeploy._register_task_defc C s& t || j�}|j� |j� || _d S )N)�CodeDeployValidatorr �describe_cd_resources� validate_allr )r" r8 Z validatorr' r' r( r � s z)ECSDeploy._validate_code_deploy_resourcesN)�__name__�
__module__� __qualname__�NAMEZ DESCRIPTIONZ ARG_TABLErT r3 r6 r r) r! rH r r r r r' r' r' r( r s>
r c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dS )r- zKWaiting for {deployment_id} to succeed (will wait up to {wait} minutes)...
c C s || _ || _d S )N)�_client�
_appspec_dict)r" � cd_clientZ appspec_dictr' r' r( �__init__� s zCodeDeployer.__init__c C sX | j ||�}y| jjf |�}W n0 tk
rN } ztjd|d��W Y d d }~X nX |d S )Nzcreate deployment)�actionr< � deploymentId)�_get_create_deploy_requestr] r/ r r �ServiceClientError)r" r* �deploy_grp_nameZ request_objrU rG r' r' r( r/ � s z CodeDeployer.create_deploymentc C s$ t j| j�}tj|�}tj|�j� S )N)rP �dumpsr^ r Z ensure_bytes�hashlib�sha256Z hexdigest)r" Z appspec_strZappspec_encodedr' r' r( �_get_appspec_hash� s
z CodeDeployer._get_appspec_hashc C s$ ||dt j| j�| j� d�d�d�S )NZAppSpecContent)Zcontentrh )Z revisionTypeZappSpecContent)�applicationName�deploymentGroupNameZrevision)rP rf r^ ri )r" r* re r' r' r( rc � s
z'CodeDeployer._get_create_deploy_requestc C s� | j }tjd|d�}g }x\|| D ]P}x@|D ]8}|| }tj||d�}|| } tj|| d�}
|| |
< q,W |j|� q"W |||< || _ dS )a�
Inserts the ARN of the previously created ECS task definition
into the provided appspec.
Expected format of ECS appspec (YAML) is:
version: 0.0
resources:
- <service-name>:
type: AWS::ECS::Service
properties:
taskDefinition: <value> # replace this
loadBalancerInfo:
containerName: <value>
containerPort: <value>
zcodedeploy-appspecr Z
propertiesrR N)r^ r Zfind_required_key�append) r" Znew_arnr& Z
resources_keyZupdated_resources�resourcer
Zresource_contentZproperties_keyZproperties_contentZ task_def_keyr' r' r( r. � s
z CodeDeployer.update_task_def_arnc C sn | j jd�}|d k r"|tkr"t}n|d ks2|dk r6d}t}|d | }||d�}| j||� |j||d� d S )NZdeployment_successful� �< )ZDelayZ MaxAttempts)rb Z WaiterConfig)r] Z
get_waiter� MAX_WAIT_MIN�DEFAULT_DELAY_SEC�_show_deploy_wait_msg�wait)r" r, �wait_minZwaiterZ delay_secZ max_attemptsr r' r' r( r5 ( s z$CodeDeployer.wait_for_deploy_successc C s& t jj| jj||d�� t jj� d S )N)r9 rs )r0 r1 r2 � MSG_WAITINGr4 r7 )r" r, rt r' r' r( rr ; s
z"CodeDeployer._show_deploy_wait_msgN) rY rZ r[ ru r` r/ ri rc r. r5 rr r' r' r' r( r- � s
+r- c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )rV c C s || _ || _d S )N)r] �_resource_names)r" r_ r r' r' r( r` C s z CodeDeployValidator.__init__c C s� y| j j| jd d�| _W n0 tk
rJ } ztjd|d��W Y d d }~X nX y*| jd }| jd }| j j||d�| _W n0 tk
r� } ztjd|d��W Y d d }~X nX d S )Nr* )rj z describe Code Deploy application)ra r< r+ )rj rk z%describe Code Deploy deployment group) r] Zget_applicationrv � app_detailsr r rd Zget_deployment_group�deployment_group_details)r" rG �dgpZappr' r' r( rW G s
z)CodeDeployValidator.describe_cd_resourcesc C s\ t | d� s| jd krd S | jd }|d }|d d }d}|| d }|| }|t S d S )Nrx �deploymentGroupInfoZ blueGreenDeploymentConfigurationZdeploymentReadyOptionZwaitTimeInMinutesZ)terminateBlueInstancesOnDeploymentSuccessZ terminationWaitTimeInMinutes)�hasattrrx �TIMEOUT_BUFFER_MIN)r" Zdgp_infoZblue_green_infoZdeploy_ready_wait_minZ
terminate_keyZtermination_wait_minZconfigured_waitr' r' r( r X s
z,CodeDeployValidator.get_deployment_wait_timec C s | j � | j� d S )N)�validate_application�validate_deployment_group)r" r' r' r( rX l s z CodeDeployValidator.validate_allc C s. | j d }| jd d dkr*tjd|d��d S )Nr* Z application�computePlatform�ECSZ Application)rm r
)rv rw r �InvalidPlatformError)r" r* r' r' r( r} p s
z(CodeDeployValidator.validate_applicationc C s� | j d }| j d }| j d }| j d }| j d }| jd }|d }|dkrZtjd |d
��| jd d }x\|D ]T} | d }
|
|kr�|
|kr�tj|d|d
��| d } | |krn| |krntj|d|d
��qnW d S )Nr+ r rJ r rK rz r r� zDeployment Group)rm r
Z ecsServices� serviceName)Zdg_namerm Z
resource_nameZ clusterName)rv rx r r� ZInvalidProperyError) r" ry r rJ r rK Zgrp_infoZcompute_platformZtarget_services�targetZ target_servZtarget_clusterr' r' r( r~ v s(
z-CodeDeployValidator.validate_deployment_groupN) rY rZ r[ r` rW r rX r} r~ r' r' r' r( rV B s rV c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )r c C s6 || _ tj|d�| _|jd|j|j|j| jd�| _d S )N)r Zecs)r � endpoint_urlr r ) �_argsr r Z_custom_configr r r� r r] )r" Zsessionr# r$ r r' r' r( r` � s zECSClient.__init__c C s� | j j}|d ksdrd}y| jj|| j jgd�}W n0 tk
rb } ztjd|d��W Y d d }~X nX t|d �dkr�tj | j j|d��|d d }t
j |d �}|d
|d |d |d �S )
N� �default)r �serviceszdescribe ECS service)ra r< r� r )r r Z
clusterArnZ
serviceArnr� )rJ r: rK rI ) r� r r] Zdescribe_servicesr r r rd �lenZInvalidServiceErrorr Zget_cluster_name_from_arn)r" r Zservice_responserG rO rI r' r' r( rL � s&
z ECSClient.get_service_detailsc C sH y| j jf |�}W n0 tk
rB } ztjd|d��W Y d d }~X nX |S )Nz register ECS task definition)ra r< )r] rS r r rd )r" �kwargsrU rG r' r' r( rS � s z"ECSClient.register_task_definitionN)rY rZ r[ r` rL rS r' r' r' r( r � s
r )rg rP r= r0 Zbotocorer r Zbotocore.exceptionsr Z
awscli.compatr Zawscli.customizations.ecsr r Z awscli.customizations.commandsr r| rq rp r r- rV r r' r' r' r( �<module> s 9lR