U

<�^y�@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�
Zcli�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r
r
r�#register_add_generate_db_auth_token*s
rcKst|
�
}||d
<dS)N�generate-db-auth-token)
�GenerateDBAuthTokenCommand)�
command_table�session�kwargsZcommandr
r
rr.s

rc
Kstj
|d
dd�
|d=dS)N�options-to-include�options�
�new_name�options-to-remove�rZrename_argument�Zargument_tablerr
r
rr
3s

�r
c
Kstj
|d
dd�
|d=dS)Nrrrrrrr
r
rr9s

�rcKsR|d
=|
�d�
}|�
d�
}tddt|
�
|
|d�|d<tdd|
|t|
�
d�|d<dS)Nzmodify-option-group�rdsZModifyOptionGroupzadd-option-to-option-group)�parent_name�name�operation_callerr�operation_modelzremove-option-from-option-group)rr rr"r!)Zget_service_modelr"rr)rrrZ	rds_modelZmodify_operation_modelr
r
rr	?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�portZintegerz-The port number the database is listening on.)r Z
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)Z
DBHostnameZPortZ
DBUsername�

r
)
Z_sessionZ
create_clientZregionZendpoint_urlZ
verify_sslZgenerate_db_auth_tokenr#r&r'r)�selfZparsed_argsZparsed_globalsr�tokenr
r
r�	_run_mainas


�


�

z$GenerateDBAuthTokenCommand._run_mainN)�__name__�
__module__�__qualname__�NAMEZDESCRIPTIONZ	ARG_TABLEr+r
r
r
rrSs
�
�
�
��	rN)�__doc__Zawscli.clidriverrrZawscli.customizationsrZawscli.customizations.commandsrZawscli.customizations.utilsrrrrr
rr	rr
r
r
r�<module>
s