File: //lib/python3/dist-packages/awscli/customizations/ecs/__pycache__/deploy.cpython-310.pyc
o
.&�a�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 )%� ECSDeploy�deploya 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)�name� help_text�requiredztask-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>�codedeploy-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�\}}t| j||| j�}| �||�| _| jjd|j |j
t j | jd�d�}| �
|� | j�� | _| �||�| _| �||� d S )N�
codedeploy��user_agent_extra)� region_name�verifyr )�_load_file_args�task_definition�codedeploy_appspec� ECSClient�_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_obj�ecs_client_wrapper�codedeploy_client� r2 �B/usr/lib/python3/dist-packages/awscli/customizations/ecs/deploy.py� _run_mainr s, �� �� �
�zECSDeploy._run_mainc C s� t ||�}|�| j� |�| jd | jd �}tj�| jj |d�� |�
|| j � | jd }tj�| j j | j|d�� tj�
� d S )N�app_name�deployment_group_name)�idr
)�task_defr
)� CodeDeployer�update_task_def_arnr) �create_deploymentr �sys�stdout�write�MSG_CREATED_DEPLOYMENT�format�wait_for_deploy_successr' � MSG_SUCCESS�flush)r+ �client�appspec�deployer�
deployment_id� service_namer2 r2 r3 r* � s"
� �
��z)ECSDeploy._create_and_wait_for_deploymentc
C sz t j�t j�|��}z t|��
}|�� W d � W S 1 s w Y W d S tttfy< } zt j
||d��d }~ww )N)� file_path�error) �os�path�
expandvars�
expanduserr �read�OSError�IOError�UnicodeDecodeErrorr �
FileLoadError)r+ rI � full_path�f�er2 r2 r3 �_get_file_contents� s
(����z ECSDeploy._get_file_contentsc C sR |� � }|d }|d }t�|||j�}t�|||j�}||d ||d ||d�S )NrH � cluster_name� service_arn� cluster_arn)r
rY r rZ r5 r6 )�get_service_detailsr � get_app_name�codedeploy_application�get_deploy_group_name�codedeploy_deployment_group)r+ �args�
ecs_client�service_detailsrH rX �application_namer6 r2 r2 r3 r � s ���z ECSDeploy._get_resource_namesc C s0 | � |�}t�|�}| � |�}t�|�}||fS �N)rW �json�loadsr �
parse_appspec)r+ � task_def_arg� appspec_arg�task_def_stringr. �appspec_stringr/ r2 r2 r3 r � s
zECSDeploy._load_file_argsc C s: |� |�}|d d }tj�| jj|d�� tj�� |S )N�taskDefinition�taskDefinitionArn)�arn)�register_task_definitionr< r= r> �MSG_TASK_DEF_REGISTEREDr@ rC )r+ �task_def_kwargsra �responser) r2 r2 r3 r( � s
�
z ECSDeploy._register_task_defc C s& t || j�}|�� |�� || _d S rd )�CodeDeployValidatorr �describe_cd_resources� validate_allr% )r+ rD � validatorr2 r2 r3 r$ � s
z)ECSDeploy._validate_code_deploy_resourcesN)�__name__�
__module__� __qualname__�NAME� DESCRIPTION� ARG_TABLErp r? rB r r4 r* rW r r r( r$ r2 r2 r2 r3 r sR ���
� � � ��<� r c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dS )r9 zKWaiting for {deployment_id} to succeed (will wait up to {wait} minutes)...
c C � || _ || _d S rd )�_client�
_appspec_dict)r+ � cd_client� appspec_dictr2 r2 r3 �__init__� �
zCodeDeployer.__init__c
C sP | � ||�}z| jjdi |��}W |d S ty' } ztjd|d��d }~ww )Nzcreate deployment��actionrJ � deploymentIdr2 )�_get_create_deploy_requestr~ r; r r �ServiceClientError)r+ r5 �deploy_grp_name� request_objrr rV r2 r2 r3 r; � s �����z CodeDeployer.create_deploymentc C s$ t �| j�}t�|�}t�|��� S rd )re �dumpsr r � ensure_bytes�hashlib�sha256� hexdigest)r+ � appspec_str�appspec_encodedr2 r2 r3 �_get_appspec_hash� s
z CodeDeployer._get_appspec_hashc C s$ ||dt �| j�| �� d�d�d�S )N�AppSpecContent)�contentr� )� revisionType�appSpecContent)�applicationName�deploymentGroupName�revision)re r� r r� )r+ r5 r� r2 r2 r3 r� � s
���z'CodeDeployer._get_create_deploy_requestc C s� | j }t�d|d�}g }|| D ]&}|D ] }|| }t�||d�}|| } t�|| d�}
|| |
< q|�|� q|||< || _ 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>
r r �
propertiesrl N)r r �find_required_key�append) r+ �new_arnr/ �
resources_key�updated_resources�resourcer �resource_content�properties_key�properties_content� task_def_keyr2 r2 r3 r: � s& � ��
z CodeDeployer.update_task_def_arnc C sn | j �d�}|d ur|tkrt}n
|d u s|dk rd}t}|d | }||d�}| �||� |j||d� d S )N�deployment_successful� �< )�Delay� MaxAttempts)r� � WaiterConfig)r~ �
get_waiter� MAX_WAIT_MIN�DEFAULT_DELAY_SEC�_show_deploy_wait_msg�wait)r+ r7 �wait_min�waiter� delay_sec� max_attemptsr r2 r2 r3 rA ( s � z$CodeDeployer.wait_for_deploy_successc C s& t j�| jj||d�� t j�� d S )N)rG r� )r< r= r> � MSG_WAITINGr@ rC )r+ r7 r� r2 r2 r3 r� ; s ��z"CodeDeployer._show_deploy_wait_msgN) rw rx ry r� r� r; r� r� r: rA r� r2 r2 r2 r3 r9 � s
+ r9 c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )rs c C r} rd )r~ �_resource_names)r+ r� r r2 r2 r3 r� C r� z CodeDeployValidator.__init__c
C s� z
| j j| jd d�| _W n ty } ztjd|d��d }~ww z| jd }| jd }| j j||d�| _W d S tyJ } ztjd|d��d }~ww )Nr5 )r� z describe Code Deploy applicationr� r6 )r� r� z%describe Code Deploy deployment group) r~ �get_applicationr� � app_detailsr r r� �get_deployment_group�deployment_group_details)r+ rV �dgp�appr2 r2 r3 rt G s, ����
����z)CodeDeployValidator.describe_cd_resourcesc C sV t | d�r
| jd u r d S | jd }|d }|d d }d}|| d }|| }|t S )Nr� �deploymentGroupInfo� blueGreenDeploymentConfiguration�deploymentReadyOption�waitTimeInMinutes�)terminateBlueInstancesOnDeploymentSuccess� terminationWaitTimeInMinutes)�hasattrr� �TIMEOUT_BUFFER_MIN)r+ �dgp_info�blue_green_info�deploy_ready_wait_min�
terminate_key�termination_wait_min�configured_waitr2 r2 r3 r& X s
�
�z,CodeDeployValidator.get_deployment_wait_timec C s | � � | �� d S rd )�validate_application�validate_deployment_group)r+ r2 r2 r3 ru l s z CodeDeployValidator.validate_allc C s. | j d }| jd d dkrtjd|d��d S )Nr5 � application�computePlatform�ECS� Application�r� r )r� r� r �InvalidPlatformError)r+ r5 r2 r2 r3 r� p s
��z(CodeDeployValidator.validate_applicationc C s� | j d }| j d }| j d }| j d }| j d }| jd }|d }|dkr-tjd |d
��| jd d }|D ]*} | d }
|
|krL|
|krLtj|d|d
��| d } | |kr`| |kr`tj|d|d
��q6d S )Nr6 r
rY r rZ r� r� r� zDeployment Groupr� � ecsServices� serviceName)�dg_namer� �
resource_name� clusterName)r� r� r r� �InvalidProperyError) r+ r� r
rY r rZ �grp_info�compute_platform�target_services�target� target_serv�target_clusterr2 r2 r3 r� v s4
� �����z-CodeDeployValidator.validate_deployment_groupN) rw rx ry r� rt r& ru r� r� r2 r2 r2 r3 rs B s rs c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )r c C s6 || _ tj|d�| _|jd|j|j|j| jd�| _d S )Nr �ecs)r � endpoint_urlr r ) �_argsr r# �_custom_configr r! r� r" r~ )r+ �sessionr, r- r r2 r2 r3 r� � s �zECSClient.__init__c
C s� | j j}|d u s nd}z
| jj|| j jgd�}W n ty+ } ztjd|d��d }~ww t|d �dkr=tj | j j|d��|d d }t
� |d �}|d
|d |d |d �S )
N� �default)r �serviceszdescribe ECS servicer� r� r )r
r �
clusterArn�
serviceArnr� )rY rH rZ rX ) r� r r~ �describe_servicesr
r r r� �len�InvalidServiceErrorr �get_cluster_name_from_arn)r+ r �service_responserV rb rX r2 r2 r3 r[ � s6
����� ���z ECSClient.get_service_detailsc
C s@ z | j jdi |��}W |S ty } ztjd|d��d }~ww )Nz register ECS task definitionr� r2 )r~ ro r r r� )r+ �kwargsrr rV r2 r2 r3 ro � s �����z"ECSClient.register_task_definitionN)rw rx ry r� r[ ro r2 r2 r2 r3 r � s
r )r� re rK r< �botocorer r �botocore.exceptionsr �
awscli.compatr �awscli.customizations.ecsr r � awscli.customizations.commandsr r� r� r� r r9 rs r r2 r2 r2 r3 �<module> s"
9lR