ꦫꦣꦺꦤ꧀ꦄꦤ꧀ꦠꦱꦺꦤ

PATH: usr / local / jetapps / var / lib / aws / lib / python2.7 / site-packages / awscli / customizations

# Copyright 2018 Amazon.com, Inc. or its affiliates. All Rights Reserved.
#
# 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.
"""Add S3 specific event streaming output arg."""
from awscli.arguments import CustomArgument

STREAM_HELP_TEXT = 'Filename where the records will be saved'

class DocSectionNotFoundError(Exception):
    pass

def register_event_stream_arg(event_handlers):
    event_handlers.register(
        'building-argument-table.s3api.select-object-content',
        add_event_stream_output_arg)
    event_handlers.register_last(
        'doc-output.s3api.select-object-content',
        replace_event_stream_docs
    )

def add_event_stream_output_arg(argument_table, operation_model,
                                session, **kwargs):
    argument_table['outfile'] = S3SelectStreamOutputArgument(
        name='outfile', help_text=STREAM_HELP_TEXT,
        cli_type_name='string', positional_arg=True,
        stream_key=operation_model.output_shape.serialization['payload'],
        session=session)

def replace_event_stream_docs(help_command, **kwargs):
    doc = help_command.doc
    current = ''
    while current != '======\nOutput\n======':
        try:
            current = doc.pop_write()
        except IndexError:
            # This should never happen, but in the rare case that it does
            # we should be raising something with a helpful error message.
            raise DocSectionNotFoundError(
                'Could not find the "output" section for the command: %s'
                % help_command)
    doc.write('======\nOutput\n======\n')
    doc.write("This command generates no output.  The selected "
              "object content is written to the specified outfile.\n")

class S3SelectStreamOutputArgument(CustomArgument):

def __init__(self, stream_key, session, **kwargs):
        super(S3SelectStreamOutputArgument, self).__init__(**kwargs)
        # This is the key in the response body where we can find the
        # streamed contents.
        self._stream_key = stream_key
        self._output_file = None
        self._session = session

def add_to_params(self, parameters, value):
        self._output_file = value
        self._session.register('after-call.s3.SelectObjectContent',
                               self.save_file)

def save_file(self, parsed, **kwargs):
        # This method is hooked into after-call which fires
        # before the error checking happens in the client.
        # Therefore if the stream_key is not in the parsed
        # response we immediately return and let the default
        # error handling happen.
        if self._stream_key not in parsed:
            return
        event_stream = parsed[self._stream_key]
        with open(self._output_file, 'wb') as fp:
            for event in event_stream:
                if 'Records' in event:
                    fp.write(event['Records']['Payload'])
        # We don't want to include the streaming param in
        # the returned response, it's not JSON serializable.
        del parsed[self._stream_key]

[-] codecommit.py [edit]
[+] eks
[-] iamvirtmfa.py [edit]
[-] cloudsearchdomain.pyc [edit]
[-] commands.py [edit]
[-] opsworks.py [edit]
[-] awslambda.py [edit]
[-] scalarparse.py [edit]
[-] utils.pyc [edit]
[-] sagemaker.pyc [edit]
[-] codecommit.pyc [edit]
[-] ecr.py [edit]
[-] cliinputjson.py [edit]
[-] arguments.py [edit]
[-] sessendemail.pyc [edit]
[+] configservice
[-] assumerole.py [edit]
[-] route53.pyc [edit]
[+] datapipeline
[-] s3uploader.py [edit]
[+] servicecatalog
[-] commands.pyc [edit]
[+] ..
[-] s3endpoint.py [edit]
[-] argrename.py [edit]
[-] translate.py [edit]
[-] argrename.pyc [edit]
[-] cliinputjson.pyc [edit]
[-] kms.pyc [edit]
[-] iot.py [edit]
[-] ecr.pyc [edit]
[-] generatecliskeleton.pyc [edit]
[-] awslambda.pyc [edit]
[-] iot_data.py [edit]
[-] cloudsearch.py [edit]
[-] streamingoutputarg.pyc [edit]
[-] __init__.pyc [edit]
[-] rds.py [edit]
[-] preview.pyc [edit]
[-] paginate.pyc [edit]
[-] removals.py [edit]
[-] arguments.pyc [edit]
[-] rekognition.pyc [edit]
[-] addexamples.py [edit]
[-] globalargs.py [edit]
[-] s3events.py [edit]
[-] cloudsearch.pyc [edit]
[-] toplevelbool.pyc [edit]
[-] iot.pyc [edit]
[+] cloudtrail
[+] configure
[+] gamelift
[-] globalargs.pyc [edit]
[-] cloudfront.py [edit]
[-] streamingoutputarg.py [edit]
[-] cloudsearchdomain.py [edit]
[-] utils.py [edit]
[-] opsworkscm.pyc [edit]
[-] sagemaker.py [edit]
[-] sessionmanager.py [edit]
[+] ecs
[+] cloudformation
[-] s3errormsg.pyc [edit]
[+] history
[-] s3endpoint.pyc [edit]
[-] opsworkscm.py [edit]
[+] ec2
[-] s3errormsg.py [edit]
[+] dlm
[-] scalarparse.pyc [edit]
[-] putmetricdata.pyc [edit]
[-] rekognition.py [edit]
[-] addexamples.pyc [edit]
[-] sessendemail.py [edit]
[-] translate.pyc [edit]
[-] s3events.pyc [edit]
[-] rds.pyc [edit]
[-] flatten.py [edit]
[-] flatten.pyc [edit]
[-] sessionmanager.pyc [edit]
[-] __init__.py [edit]
[-] paginate.py [edit]
[-] removals.pyc [edit]
[-] assumerole.pyc [edit]
[+] s3
[-] route53.py [edit]
[-] putmetricdata.py [edit]
[-] waiters.py [edit]
[-] generatecliskeleton.py [edit]
[-] waiters.pyc [edit]
[-] mturk.py [edit]
[-] sms_voice.py [edit]
[-] mturk.pyc [edit]
[-] kms.py [edit]
[-] s3uploader.pyc [edit]
[-] preview.py [edit]
[-] iamvirtmfa.pyc [edit]
[-] sms_voice.pyc [edit]
[-] opsworks.pyc [edit]
[+] emr
[-] iot_data.pyc [edit]
[+] codedeploy
[-] toplevelbool.py [edit]
[-] cloudfront.pyc [edit]