3

�T_!�@s�dZd
dl
Z
d
dlZd
dlmZ
dd�Zdd�Zdd	�Zd
d�Zdd
�Z	dd�Z
dd�Zdd�Zdd�Z
dd�Zddd�
Zddd�
Zdd�ZdS) zC
Utility functions to make it easier to work with customizations.

�N)
�ClientErrorcCs ||
}|||<||_||
=dS)
N)
�name)�argument_table�
existing_name�new_name�current�r�/usr/lib/python3.6/utils.py�rename_arguments



r
cCs$||
}tj|�
}||_
|||<|S)
N)�copyr)rZcurrent_nameZ	copy_namer�copy_argrrr	�_copy_arguments






r
cCs6||
}t||
|�}d
|_
|jr2d|_d|_d
|_dS)a{
Create a hidden alias for an existing argument.

    This will copy an existing argument object in an arg table,
    and add a new entry to the arg table with a different name.
    The new argument will also be undocumented.

    This is needed if you want to check an existing argument,
    but you still need the other one to work for backwards
    compatibility reasons.

    TFN)r
�
_UNDOCUMENTEDZrequiredZ_DOCUMENT_AS_REQUIRED)rr�
alias_namerrrrr	�make_hidden_alias&s




rcCs ||
}|||<||_||
=dS)
N)
r)�
command_tablerrrrrr	�rename_command?s



rcCs||
}t||
|�
d
|_
dS)aW
Moves an argument to a new name, keeping the old as a hidden alias.

    :type command_table: dict
    :param command_table: The full command table for the CLI or a service.

    :type existing_name: str
    :param existing_name: The current name of the command.

    :type new_name: str
    :param new_name: The new name for the command.
    TN)r
r)rrrrrrr	�
alias_commandFs

rcCst||
|�}d
|_
dS)a�Create a hidden alias for an exiting command.

    This will copy an existing command object in a command table and add a new
    entry to the command table with a different name. The new command will
    be undocumented.

    This is needed if you want to change an existing command, but you still
    need the old name to work for backwards compatibility reasons.

    :type command_table: dict
    :param command_table: The full command table for the CLI or a service.

    :type existing_name: str
    :param existing_name: The current name of the command.

    :type alias_name: str
    :param alias_name: The new name for the command.
    TN)r
r)rrr�newrrr	�make_hidden_command_aliasWs
rcs�f
d
d�}
|
S)Nc
st|f
���S)
N)
�validate_mutually_exclusive)�parsed_args�kwargs)
�groupsrr	�_handleros
z5validate_mutually_exclusive_handler.<locals>._handlerr)rrr)
rr	�#validate_mutually_exclusive_handlerns
rc
s�t|�
�
t
d
d�|
D�
�
�t��
fdd��
D�
�
s6dSd}xZ�
f
dd��
D�
D]D}t||
�}|dkrfqN|dkrt|}qN||ksNtd|dj|�
f�
�
qNWdS)	z7Validate mututally exclusive groups in the parsed args.c
ss|]}
|
D]
}|V
q
qdS)
Nr)�.0�group�argrrr	�	<genexpr>wsz.validate_mutually_exclusive.<locals>.<genexpr>c
3s"|]}
�
|
dk	r|
�kV
qdS)
Nr)r�
k)�all_args�	args_dictrr	rxsNc
sg|]}
�|
dk	r|
�qS)
Nr)rr )
r"rr	�
<listcomp>}sz/validate_mutually_exclusive.<locals>.<listcomp>zVThe key "%s" cannot be specified when one of the following keys are also specified: %sz, )�vars�set�any�_get_group_for_key�
ValueError�join)rrZ
current_group�keyZ	key_groupr)r!r"r	rts










rcCsx|
D]}||kr|SqWdS)
Nr)r*rrrrr	r'�s



r'cCsZd
}y|j|
d�

Wn@t
k
rT
}
z$t|jdd�
}|dkrDd}WYdd}~XnX|S)NT)
ZBucket�ErrorZCodei�
F)Zhead_bucketr�intZresponse)Z	s3_clientZbucket_nameZ
bucket_exists�
eZ
error_coderrr	�s3_bucket_exists�s





r.cCsVi}d
|kr|j|d<d|kr(|j
|d<d|kr:|j|d<|rH|j|�

|j|
f
|�
S)a
Creates a service client, taking parsed_globals into account

    Any values specified in overrides will override the returned dict. Note
    that this override occurs after 'region' from parsed_globals has been
    translated into 'region_name' in the resulting dict.
    �regionZregion_name�endpoint_url�
verify_sslZverify)r/r0r1�updateZ
create_client)ZsessionZservice_nameZparsed_globalsZ	overridesZclient_argsrrr	�!create_client_from_parsed_globals�s












r3cCsr|
d
krtj
}
y|
j|�

WnHtk
rd


t|
dd�}|d
krDd}|j|d�j|�
}|
j|�

YnX|
j�
d
S)z�
    This function is used to properly write unicode to a file, usually
    stdout or stdderr.  It ensures that the proper encoding is used if the
    statement is not a string type.
    N�encoding�ascii�replace)�sys�stdout�write�UnicodeEncodeError�getattr�encode�decode�flush)Z	statementZout_fileZnew_encodingZ
new_statementrrr	�	uni_print�s







r?c
Cs,|j�}
|
j
d
�
rdS|
j
d�
r$dSdSdS)z7Method to return region value as expected by policy arnzcn-zaws-cnzus-govz
aws-us-govZawsN)�lower�
startswith)r/Z
region_stringrrr	�get_policy_arn_suffix�s





rB)
N)
N)�__doc__rr7Zbotocore.exceptionsrr
r
rrrrrrr'r.r3r?rBrrrr	�<module>s 



+