aws-cli/awscli/customizations/iamvirtmfa.py at develop · rubythonode/aws-cli · GitHub

History

78 lines (67 loc) · 3.2 KB

Raw

# Licensed under the Apache License, Version 2.0 (the "License"). You

# may not use this file except in compliance with the License. A copy of

# the License is located at

# http://aws.amazon.com/apache2.0/

# or in the "license" file accompanying this file. This file is

# distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF

# ANY KIND, either express or implied. See the License for the specific

# language governing permissions and limitations under the License.

"""

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.

"""

import base64

from awscli.customizations.arguments import StatefulArgument

from awscli.customizations.arguments import resolve_given_outfile_path

from awscli.customizations.arguments import is_parsed_result_successful

CHOICES = ('QRCodePNG', 'Base32StringSeed')

OUTPUT_HELP = ('The output path and file name where the bootstrap '

'information will be stored.')

BOOTSTRAP_HELP = ('Method to use to seed the virtual MFA. '

'Valid values are: %s | %s' % CHOICES)

class FileArgument(StatefulArgument):

def add_to_params(self, parameters, value):

# Validate the file here so we can raise an error prior

# calling the service.

value = resolve_given_outfile_path(value)

super(FileArgument, self).add_to_params(parameters, value)

class IAMVMFAWrapper(object):

def __init__(self, event_handler):

self._event_handler = event_handler

self._outfile = FileArgument(

'outfile', help_text=OUTPUT_HELP, required=True)

self._method = StatefulArgument(

'bootstrap-method', help_text=BOOTSTRAP_HELP,

choices=CHOICES, required=True)

self._event_handler.register(

'building-argument-table.iam.create-virtual-mfa-device',

self._add_options)

self._event_handler.register(

'after-call.iam.CreateVirtualMFADevice', self._save_file)

def _add_options(self, argument_table, **kwargs):

argument_table['outfile'] = self._outfile

argument_table['bootstrap-method'] = self._method

def _save_file(self, parsed, **kwargs):

if not is_parsed_result_successful(parsed):

return

method = self._method.value

outfile = self._outfile.value

if method in parsed['VirtualMFADevice']:

body = parsed['VirtualMFADevice'][method]

with open(outfile, 'wb') as fp:

fp.write(base64.b64decode(body))

for choice in CHOICES:

if choice in parsed['VirtualMFADevice']:

del parsed['VirtualMFADevice'][choice]

aws-cli/awscli/customizations/iamvirtmfa.py at develop · rubythonode/aws-cli · GitHub

Expand file tree

Breadcrumbs

iamvirtmfa.py

Latest commit

History

Breadcrumbs

iamvirtmfa.py

File metadata and controls