app.py

from chalice import Chalice, Response
from chalice import BadRequestError, ConflictError, NotFoundError

from chalicelib.services import create_new_user, verify_user_code, delete_user

app = Chalice(app_name='lambdatotp')


@app.route('/', methods=['POST'])
def create_user():

    body = app.current_request.json_body

    try:
        user_id = body['user']
    except KeyError:
        raise BadRequestError('No user found in the request body')

    try:
        response = create_new_user(user_id)
    except ValueError:
        raise ConflictError('User already exists')

    return Response(body=response, status_code=201)


@app.route('/verify', methods=['GET'])
def verify():

    user = app.current_request.query_params.get('user')
    code = app.current_request.query_params.get('code')

    if not code or not user:
        raise BadRequestError('No code or user found in the query string')

    try:
        return bool(verify_user_code(user, code))
    except ValueError:
        raise NotFoundError('User not found')


@app.route('/', methods=['DELETE'])
def delete(user):
    try:
        delete_user(user)
    except ValueError:
        raise NotFoundError('User not found')
    else:
        return Response(body='', status_code=204)