o
    !d                     @   sP   d dl mZ d dl mZ d dl mZ d dl mZ d dlmZ G dd deZdS )	    )argumentschema)emrutils)helptext)	steputils)Commandc                   @   sJ   e Zd ZdZdZddejddddejej	dd	d
ej
dgZdd ZdS )AddStepsz	add-stepsz!Add a list of steps to a cluster.z
cluster-idT)namerequired	help_textsteps+)r   r	   nargsZschemar
   zexecution-role-arnFc                 C   sj   |j }t|j| j| j|j|j}tj	|| j|d}|j|d}|j
d ur*|j
|d< t| jd|| dS )N)Zparsed_step_listregionrelease_label)Z	JobFlowIdZStepsZExecutionRoleArnZAddJobFlowStepsr   )r   r   Zget_release_labelZ
cluster_idZ_sessionr   Zendpoint_urlZ
verify_sslr   Zbuild_step_config_listZexecution_role_arnZcall_and_display_response)selfZparsed_argsZparsed_globalsZparsed_stepsr   Z	step_list
parameters r   Fusr/lib/python3.10/site-packages/awscli/customizations/emr/addsteps.py_run_main_command(   s$   


zAddSteps._run_main_commandN)__name__
__module____qualname__NAMEZDESCRIPTIONr   Z
CLUSTER_IDr   ZSTEPS_SCHEMAZSTEPSZEXECUTION_ROLE_ARNZ	ARG_TABLEr   r   r   r   r   r      s$    r   N)Zawscli.customizations.emrr   r   r   r   Z!awscli.customizations.emr.commandr   r   r   r   r   r   <module>   s   