File: //lib/python3/dist-packages/awscli/customizations/__pycache__/iamvirtmfa.cpython-38.pyc
U
<�^� � @ sd d Z ddlZddlmZ ddlmZ ddlmZ dZdZde ZG d d
� d
e�Z G d d � d e
�Z dS )
a\
This customization makes it easier to deal with the bootstrapping
data returned by the ``iam create-virtual-mfa-device`` command.
You can choose to bootstrap via a QRCode or via a Base32String.
You specify your choice via the ``--bootstrap-method`` option
which should be either "QRCodePNG" or "Base32StringSeed". You
then specify the path to where you would like your bootstrapping
data saved using the ``--outfile`` option. The command will
pull the appropriate data field out of the response and write it
to the specified file. It will also remove the two bootstrap data
fields from the response.
� N)�StatefulArgument)�resolve_given_outfile_path)�is_parsed_result_successful)Z QRCodePNGZBase32StringSeedzMThe output path and file name where the bootstrap information will be stored.zAMethod to use to seed the virtual MFA. Valid values are: %s | %sc s e Zd Z� fdd�Z� ZS )� FileArgumentc s t |�}tt| ��||� d S )N)r �superr �
add_to_params)�selfZ
parameters�value�� __class__� �B/usr/lib/python3/dist-packages/awscli/customizations/iamvirtmfa.pyr ) s zFileArgument.add_to_params)�__name__�
__module__� __qualname__r �
__classcell__r r r
r
r ' s r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�IAMVMFAWrapperc C sL || _ tdtdd�| _tdttdd�| _| j �d| j � | j �d| j
� d S )N�outfileT)� help_text�required�bootstrap-method)r �choicesr z5building-argument-table.iam.create-virtual-mfa-devicez%after-call.iam.CreateVirtualMFADevice) Z_event_handlerr � OUTPUT_HELP�_outfiler �BOOTSTRAP_HELP�CHOICES�_method�register� _add_options�
_save_file)r Z
event_handlerr r r
�__init__2 s( � �� �zIAMVMFAWrapper.__init__c K s | j |d< | j|d< d S )Nr r )r r )r Zargument_table�kwargsr r r
r ? s
zIAMVMFAWrapper._add_optionsc K s~ t |�s d S | jj}| jj}||d krz|d | }t|d��}|�t�|�� W 5 Q R X tD ]}||d kr^|d |= q^d S )NZVirtualMFADevice�wb) r r r r �open�write�base64Z b64decoder )r Zparsedr! �methodr Zbody�fpZchoicer r r
r C s zIAMVMFAWrapper._save_fileN)r r r r r r r r r r
r 0 s
r ) �__doc__r% Z awscli.customizations.argumentsr r r r r r r �objectr r r r r
�<module>
s �