def lambda_handler()

in source/CreateRevisionFunction/app.py [0:0]
77 lines of code
4 McCabe index (conditional complexity)

def lambda_handler(event, context):
    try:
        global log_level
        log_level = str(os.environ.get('LOG_LEVEL')).upper()
        if log_level not in [
                                'DEBUG', 'INFO',
                                'WARNING', 'ERROR',
                                'CRITICAL'
                            ]:
            log_level = 'ERROR'
        logging.getLogger().setLevel(log_level)
        
        logging.debug('event={}'.format(event))

        dataexchange = boto3.client(service_name='dataexchange')
        s3 = boto3.client(
            service_name='s3'
        ) 
        bucket = event['Bucket'] 
        key = event['Key']

        response = s3.get_object(Bucket=bucket,Key=key)
        manifest = json.loads(response['Body'].read(amt=response['ContentLength']))
        datasetId = manifest['Manifest']['DatasetId']
        productId = manifest['Manifest']['ProductId']
        logging.debug('datasetId={}'.format(datasetId))
        revision = dataexchange.create_revision(DataSetId=datasetId,Comment="from aws-data-exchange-publisher-coordinator")
        revisionId = revision['Id']
        print('revisionId={}'.format(revisionId))
        revisiondetails = {
            "ImportAssetsFromS3": {
                "AssetSources": 
                    manifest['Manifest']['Assets'] 
                ,
                "DataSetId": datasetId,
                "RevisionId": revisionId
            }
        }
        logging.debug('revision details = {}'.format(revisiondetails))

        jobresponse = dataexchange.create_job(Type='IMPORT_ASSETS_FROM_S3',Details=revisiondetails)
        jobArnparts = jobresponse['Arn'].split('/')
        jobId = jobArnparts[1]
        logging.info('jobId={}'.format(jobId))
          
        startjobresponse = dataexchange.start_job(JobId=jobId)
        httpResponse = startjobresponse['ResponseMetadata']['HTTPStatusCode']
        logging.debug('HTTPResponse={}'.format(httpResponse))
        jobresponse = dataexchange.get_job(JobId=jobId)  
        logging.debug('get job = {}'.format(jobresponse))
        jobStatus = jobresponse['State']
        sendMetrics=os.environ.get('AnonymousUsage')
        if sendMetrics=="Yes":
            metricdata = {
                "Version" : os.environ.get('Version'),
                "AssetCount" : len(manifest['Manifest']['Assets'])
            }
            solutionData={
                "Solution": os.environ.get('SolutionId'),
                "UUID": os.environ.get('UUID'),
                "TimeStamp": datetime.utcnow().strftime('%Y-%m-%d %H:%M:%S.%f'),
                "Data": metricdata
            }
            http = urllib3.PoolManager()
            metricURL = "https://metrics.awssolutionsbuilder.com/generic"
            encoded_data = json.dumps(solutionData).encode('utf-8')
            headers={'Content-Type': 'application/json'}
            http.request('POST',metricURL,
                                body=encoded_data,
                                headers=headers)
    except Exception as e:
       logging.error(e)
       raise e
    return {
        "statusCode": httpResponse,
        "Message": "Job Started",
        "ProductId" : productId,
        "DataSetId": datasetId,
        "RevisionId": revisionId,
        "JobId": jobId,
        "JobStatus" : jobStatus
    }