def handler()

in Booking/post-bookings-lambda.py [0:0]


def handler(event, context):
    print("From API G/W: " + str(event))
    body = json.loads(event['body'])
    first_name = body['first_name']
    last_name = body['last_name']
    from_airport = body['from_airport']
    to_airport = body['to_airport']
    departure_date = body['departure_date']
    return_date = body['return_date']
    age_group = body['age_group']
    booking_class = body['booking_class']
    booking_number = randomString(8);

    # insert into DynamoDB
    response = table.put_item(
        Item={
            "booking_number": booking_number,
            "age_group": age_group,
            "first_name": first_name,
            "last_name": last_name,
            "from_airport": from_airport,
            "to_airport": to_airport,
            "departure_date": departure_date,
            "return_date": return_date,
            "booking_class": booking_class

        }
    )
    print("PutItem succeeded. Response is: " + str(response))

    sns_message = {
        'booking_number': booking_number,
        'from_airport': from_airport,
        'to_airport': to_airport,
        'departure_date': departure_date
    }
    response = topic.publish(
        Message = json.dumps({'default': json.dumps(sns_message)}),
        MessageStructure = 'json'
    )

    return {
        'statusCode': 200,
        'body': json.dumps(sns_message),
        'headers': {
            'Content-Type': 'application/json',
            'Access-Control-Allow-Origin': '*'
        },
    }