o

.&�ay�@s�dZd
dl
mZ
d
dl
mZ
d
dlmZ
d
dlmZ
d
dlm	Z	
dd�Z
d	d
�Zdd�Zd
d�Z
dd�Zdd�ZGdd�de�ZdS)aS

This customization splits the modify-option-group into two separate commands:

* ``add-option-group``
* ``remove-option-group``

In both commands the ``--options-to-remove`` and ``--options-to-add`` args will
be renamed to just ``--options``.

All the remaining args will be available in both commands (which proxy
modify-option-group).

�)
�ServiceOperation)
�CLIOperationCaller)
�utils)
�BasicCommand)
�	uni_printc

Cs(|�d
t
�
|�dt�
|�dt�
dS)N�building-command-table.rdsz6building-argument-table.rds.add-option-to-option-groupz;building-argument-table.rds.remove-option-from-option-group)�register�_building_command_table�_rename_add_option�_rename_remove_option�
�cli�r�;/usr/lib/python3/dist-packages/awscli/customizations/rds.py�register_rds_modify_split"s


�
�rc

Cs|�d
t
�
dS)Nr)r�_add_generate_db_auth_tokenrrrr�#register_add_generate_db_auth_token*s
rcKst|
�
}||d
<dS)N�generate-db-auth-token)
�GenerateDBAuthTokenCommand)�
command_table�session�kwargs�commandrrrr.s

rc
K�tj
|d
dd�
|d=dS)N�options-to-include�options�
�new_name�options-to-remove�r�rename_argument��argument_tablerrrrr
3�

�
r
c
Kr)Nrrrrrr!rrrr9r#rcKsR|d
=|
�d�
}|�
d�
}tddt|
�
|
|d�|d<tdd|
|t|
�
d�|d<dS)Nzmodify-option-group�rds�ModifyOptionGroupzadd-option-to-option-group)�parent_name�name�operation_callerr�operation_modelzremove-option-from-option-group)r&r'rr)r()�get_service_modelr)rr)rrr�	rds_model�modify_operation_modelrrrr	?s








�



�r	c@s@eZ
dZd
ZdZdddd�dddd	d
�dddd�gZd
d�ZdS)rrzEGenerates an auth token used to connect to a db with IAM credentials.�hostnameTz+The hostname of the database to connect to.)r'�required�	help_text�port�integerz-The port number the database is listening on.)r'�
cli_type_namer.r/�usernamezThe username to log in as.cCsB|j�
d
|j|j|j�}|j|
j|
j|
jd�}t	|�

t	d�

dS)Nr$)�
DBHostname�Port�
DBUsername�

r
)
�_session�
create_client�region�endpoint_url�
verify_ssl�generate_db_auth_tokenr-r0r3r)�self�parsed_args�parsed_globalsr$�tokenrrr�	_run_mainas



�


�

z$GenerateDBAuthTokenCommand._run_mainN)�__name__�
__module__�__qualname__�NAME�DESCRIPTION�	ARG_TABLErBrrrrrSs
�
�
�
��	rN)�__doc__�awscli.clidriverrr�awscli.customizationsr�awscli.customizations.commandsr�awscli.customizations.utilsrrrrr
rr	rrrrr�<module>
s