def lambda_handler()

in lambdas/api_get_reviews/index.py [0:0]

• 28 lines of code
• 3 McCabe index (conditional complexity)

def lambda_handler(event, context):
  status_code = 400
  try:
    # print("[DEBUG] Received event: " + json.dumps(event, indent=2))
    output = None
    if event['queryStringParameters']:
      station_id = int(event['queryStringParameters']['stationId'])
      query_result = table.query(KeyConditionExpression=Key('station_id').eq(station_id))
    else:
      user_id = event['requestContext']['identity']['cognitoIdentityId']
      query_result = table.query(IndexName=STATION_REVIEW_GSI, KeyConditionExpression=Key('user_id').eq(user_id))

    output = query_result['Items']
    status_code = 200
  except Exception as e:
    print('ERROR: ', e)
    output = '{}'.format(e)

  # Construct API response
  response = {
    'statusCode': status_code,
    'headers': {
      'Access-Control-Allow-Origin': '*',
      'Access-Control-Allow-Credentials': True,
      'Access-Control-Allow-Headers': 'Content-Type,X-Amz-Date,Authorization,X-Api-Key,X-Amz-Security-Token,X-Amz-User-Agent',
      'Access-Control-Allow-Methods': 'GET,POST,PUT,DELETE,OPTIONS,HEAD,PATCH',
      'Content-Type': 'application/json'
    },
    'body': json.dumps(output, cls=DecimalEncoder)
  }

  print('[INFO] Query response: {}'.format(json.dumps(response)))

  return response