File: //lib/python3/dist-packages/awscli/customizations/datapipeline/__pycache__/__init__.cpython-310.pyc
o
.&�a4B � @ s2 d dl Z d dlmZmZ d dlmZ d dlmZ d dlmZ d dl m
Z
d dl m Z d dl
mZ d Zd
Zd Zd Zd
ZG dd� de�ZG dd� de�Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Z d,d d �Z G d d!� d!e �Z G d"d#� d#e�Z G d$d%� d%e�Z!G d&d'� d'e�Z"G d(d)� d)e�Z#G d*d+� d+e�Z$dS )-� N)�datetime� timedelta)�
get_formatter)�CustomArgument)� BasicCommand)�
translator)�CreateDefaultRoles)�ListRunsFormatterz�The JSON pipeline definition. If the pipeline definition
is in a file you can use the file://<filename> syntax to
specify a filename.
a The JSON parameter objects. If the parameter objects are
in a file you can use the file://<filename> syntax to
specify a filename. You can optionally provide these in
pipeline definition as well. Parameter objects provided
on command line would replace the one in definition.
a The JSON parameter values. If the parameter values are
in a file you can use the file://<filename> syntax to
specify a filename. You can optionally provide these in
pipeline definition as well. Parameter values provided
on command line would replace the one in definition.
a The JSON parameter values. You can specify these as
key-value pairs in the key=value format. Multiple parameters
are separated by a space. For list type parameter values
you can use the same key name and specify each value as
a key value pair. e.g. arrayValue=value1 arrayValue=value2
�d c @ s e Zd ZdS )�DocSectionNotFoundErrorN)�__name__�
__module__� __qualname__� r r �M/usr/lib/python3/dist-packages/awscli/customizations/datapipeline/__init__.pyr 8 s r c s e Zd Z� fdd�Z� ZS )�ParameterDefinitionErrorc s" d| }t t| ��|� || _d S )NzError in parameter: %s
)�superr �__init__�msg)�selfr �full_msg�� __class__r r r = s
z!ParameterDefinitionError.__init__)r r
r r �
__classcell__r r r r r < s r c C s@ | � dt� | � dt� | � dt� | � dt� | �dt� d S )Nz<building-argument-table.datapipeline.put-pipeline-definitionz6building-argument-table.datapipeline.activate-pipelinez-after-call.datapipeline.GetPipelineDefinitionz#building-command-table.datapipelinez/doc-output.datapipeline.get-pipeline-definition)�register�add_pipeline_definition� activate_pipeline_definition�translate_definition�register_commands�
register_last�document_translation)�clir r r �register_customizationsC s( �����r" c K s t |�| d< t|�| d< d S )N� list-runszcreate-default-roles)�ListRunsCommandr )�
command_table�session�kwargsr r r r U s r c K sZ | j }d}|dkr!z|�� }W n
ty td| ��w |dks |�d� |�d� d S )N� z======
Output
======z7Could not find the "output" section for the command: %sz�
The output of this command is the pipeline definition, which is documented in the `Pipeline Definition File Syntax <http://docs.aws.amazon.com/datapipeline/latest/DeveloperGuide/dp-writing-pipeline-definition.html>`__)�doc� pop_write�
IndexErrorr �write)� help_commandr' r) �currentr r r r Z s" ����
�r c K sT t ddtd�| d< tddtd�| d< tddtd�| d< tdddtd �| d< | d
= d S ) Nzpipeline-definitionT��required� help_textzparameter-objectsF�parameter-values-uri�parameter-values�+�r0 �nargsr1 zpipeline-objects)�PipelineDefinitionArgument�DEFINITION_HELP_TEXT�ParameterObjectsArgument�PARAMETER_OBJECTS_HELP_TEXT�ParameterValuesArgument�PARAMETER_VALUES_HELP_TEXT� ParameterValuesInlineArgument�!INLINE_PARAMETER_VALUES_HELP_TEXT��argument_tabler' r r r r r s(
�
�
�
�
r c K s* t ddtd�| d< tdddtd�| d< d S )Nr2 Fr/ r3 r4 r5 )r; r< r= r> r? r r r r � s
��r c K s t �| � d S �N)r �api_to_definition)�parsedr' r r r r � s r c C sj g }| D ]$}|d |d d�}|d D ]}|� d|� d��||d < q|�|� q|d ur3|j|d� |S ) N�id�name)z@idrE �fields� stringValue�refValue�key)rI )�get�append�sort)�api_describe_objects�
sort_key_func� converted�obj�
new_fields�fieldr r r �convert_described_objects� s � � rS c @ s2 e Zd ZdZd dd�Zdd� Zdd� Zd d
� ZdS ) �QueryArgBuilderzG
Convert CLI arguments to Query arguments used by QueryObject.
Nc C s |d u rt �� }|| _d S rA )r �utcnow� current_time)r rV r r r r � s
zQueryArgBuilder.__init__c C s� g }|j d u r.|jd u r.| j}|tdd� }|�d�}|�d�}|�dd||gd�d�� n| �||� |jd ur?| �||� d|i}|S ) N� )�daysz%Y-%m-%dT%H:%M:%S�@actualStartTime�BETWEEN��type�values�� fieldName�operator� selectors) �start_interval�schedule_intervalrV r �strftimerK �_build_schedule_times�status�
_build_status)r � parsed_argsra � end_datetime�start_datetime�start_time_str� end_time_str�queryr r r � build_query� s$
�
�
zQueryArgBuilder.build_queryc C sx |j d ur |j d }|j d }|�dd||gd�d�� |jd ur:|jd }|jd }|�dd||gd�d�� d S d S )Nr � rY rZ r[ r^ �@scheduledStartTime)rb rK rc )r ra rh rk rl r r r re � s&
��
� ��z%QueryArgBuilder._build_schedule_timesc C s& |� dddd� |jD �d�d�� d S )Nz@status�EQc S � g | ]}|� � �qS r )�upper)�.0rf r r r �
<listcomp>� s z1QueryArgBuilder._build_status.<locals>.<listcomp>r[ r^ )rK rf )r ra rh r r r rg � s � �z QueryArgBuilder._build_statusrA )r r
r �__doc__r rn re rg r r r r rT � s
rT c @ � e Zd Zdd� ZdS )r7 c C sx |d u rd S t �|�}t�|�}t�|�}t�|�}||d< d|vr*|d ur*||d< d|vr8|d ur:||d< d S d S d S )N�pipelineObjects�parameterObjects�parameterValues)�json�loadsr �definition_to_api_objects� definition_to_api_parameters� definition_to_parameter_values)r �
parameters�valuerC � api_objects�parameter_objects�parameter_valuesr r r �
add_to_params� s
�z(PipelineDefinitionArgument.add_to_paramsN�r r
r r� r r r r r7 � � r7 c @ rw )r9 c C s, |d u rd S t �|�}t�|�}||d< d S )Nry )r{ r| r r~ )r r� r� rC r� r r r r�
s
z&ParameterObjectsArgument.add_to_paramsNr� r r r r r9 r� r9 c @ rw )r; c C sD |d u rd S |� dd �d urtd��t�|�}t�|�}||d< d S )Nrz �8Only parameter-values or parameter-values-uri is allowed)rJ � Exceptionr{ r| r r )r r� r� rC r� r r r r� s �
z%ParameterValuesArgument.add_to_paramsNr� r r r r r; r� r; c @ rw )r= c C s� |d u rd S |� dd �d urtd��i }|D ]?}z0|�dd�}|d }|d }||v rCt|| t�r:|| �|� n
|| |g||< n|||< W q tyU td| ��w d|i}t� |�}||d< d S )Nrz r� �=ro r z#Invalid inline parameter format: %sr] )
rJ r� �split�
isinstance�listrK r+ r r r ) r r� r� �parameter_object�argument�argument_componentsrI rC r� r r r r� # s4 � � ��
z+ParameterValuesInlineArgument.add_to_paramsNr� r r r r r= " r� r= c @ s� e Zd ZdZdZdddddd�d d
dd �d d
dddd�dddddd�gZg d�Zdd� Zdd� Zdd� Z dd� Z
dd� Z d d � Z d d � Z
d d!� Zd"S )#r$ r# z�Lists the times the specified pipeline has run. You can optionally filter the complete list of results to include only the runs you are interested in.z pipeline-idz The identifier of the pipeline.�storeT�string)rE r1 �actionr0 �
cli_type_namerf z�Filters the list to include only runs in the specified statuses. The valid statuses are as follows: waiting, pending, cancelled, running, finished, failed, waiting_for_runner, and waiting_on_dependencies.)rE r1 r� zstart-intervalzQFilters the list to include only runs that started within the specified interval.Fzschedule-intervalz`Filters the list to include only runs that are scheduled to start within the specified interval.) �waiting�pending� cancelled�running�finished�failed�waiting_for_runner�waiting_on_dependencies�
shutting_downc K s$ | � |� | �|� | �||� d S rA )� _set_client�_parse_type_args�
_list_runs)r rh �parsed_globalsr' r r r � _run_maind s
zListRunsCommand._run_mainc C s | j jd|j|j|jd�| _d S )N� datapipeline)� region_name� endpoint_url�verify)�_session�
create_client�regionr� �
verify_ssl�client)r r� r r r r� i s �zListRunsCommand._set_clientc C sz |j d urdd� |j �d�D �|_ |jd ur"dd� |j�d�D �|_|jd ur;dd� |j�d�D �|_| �|j� d S d S )Nc S rr r ��strip�rt �argr r r ru w � �z4ListRunsCommand._parse_type_args.<locals>.<listcomp>�,c S rr r r� r� r r r ru { r� c S rr r r� r� r r r ru r� )rb r� rc rf �_validate_status_choices)r rh r r r r� r s
�
�
��z ListRunsCommand._parse_type_argsc C s0 |D ]}|| j vrtd|d�| j �f ��qd S )Nz&Invalid status: %s, must be one of: %sz, )� VALID_STATUS�
ValueError�join)r �statusesrf r r r r� � s
���z(ListRunsCommand._validate_status_choicesc C sR t � �|�}| �|j|�}| �|j|�}t|dd� d�}| �|�}|| j|� d S )Nc S s | � d�| � d�fS )Nrp rE )rJ )�xr r r �<lambda>� s �z,ListRunsCommand._list_runs.<locals>.<lambda>)rN )rT rn �_query_objects� pipeline_id�_describe_objectsrS �_get_formatter�NAME)r rh r� rm �
object_ids�objectsrO � formatterr r r r� � s �
zListRunsCommand._list_runsc C sL g }t dt|�t�D ]}|||t � }| jj||d�}|�|d � q
|S )Nr )�
pipelineId� objectIdsrx )�range�len�MAX_ITEMS_PER_DESCRIBEr� �describe_objects�extend)r r� r� r� �i�current_object_ids�resultr r r r� � s �z!ListRunsCommand._describe_objectsc C s( | j �d�j|d|d�}|�� }|d S )N�
query_objects�INSTANCE)r� �sphererm �ids)r� �
get_paginator�paginate�build_full_result)r r� rm � paginatorrC r r r r� � s �z ListRunsCommand._query_objectsc C s |j }|d u r t|�S t||�S rA )�outputr r )r r� r� r r r r� � s
z ListRunsCommand._get_formatterN)r r
r r� � DESCRIPTION� ARG_TABLEr� r� r� r� r� r� r� r� r� r r r r r$ D s: ������
r$ rA )%r{ r r �awscli.formatterr �awscli.argumentsr � awscli.customizations.commandsr �"awscli.customizations.datapipeliner �5awscli.customizations.datapipeline.createdefaultrolesr �4awscli.customizations.datapipeline.listrunsformatterr r8 r: r< r> r� r� r r r" r r r r r rS �objectrT r7 r9 r; r= r$ r r r r �<module> s8
B "