app.py

import json
from http import HTTPStatus

from flask import Flask
from flask import request as flask_request
from marshmallow.exceptions import ValidationError
from sqlalchemy.exc import NoResultFound

from .storm.exceptions import MissingPostData, EmptyResult
from .storm.db.client import client
from .storm.services.functions import delete_schedule, get_schedule, post_schedule

app = Flask(__name__)


@app.route("/msg-scheduler", methods=["POST"])
def msg_scheduler_POST():
    request_data = flask_request.get_json()

    if not flask_request:
        raise MissingPostData

    processed_response = post_schedule(client, request_data)

    return app.response_class(
        response=json.dumps(processed_response),
        status=HTTPStatus.CREATED,
    )


@app.route("/msg-scheduler", methods=["GET"])
def msg_scheduler_GET():
    request_id = flask_request.args
    processed_response = get_schedule(client, request_id)
    return app.response_class(
        status=HTTPStatus.OK,
        response=json.dumps(processed_response),
        mimetype="application/json",
    )


@app.route("/msg-scheduler", methods=["DELETE"])
def msg_scheduler_DELETE():
    request_id = {"id": flask_request.args.get("id")}
    processed_response = delete_schedule(client, request_id)
    return app.response_class(
        status=HTTPStatus.OK,
        response=json.dumps(processed_response),
        mimetype="application/json",
    )


@app.errorhandler(EmptyResult)
def handle_no_result(e):
    return app.response_class(
        status=HTTPStatus.NOT_FOUND,
        response="The requested object was not found",
        mimetype="application/json",
    )


@app.errorhandler(MissingPostData)
def handle_missing_post_data(e):
    return app.response_class(
        status=HTTPStatus.BAD_REQUEST,
        response="Missing POST data",
        mimetype="application/json",
    )


@app.errorhandler(ValidationError)
def handle_validation_error(e):
    message = ""
    for key in e.messages:
        message += f"{key}:"
        for item in e.messages[key]:
            message += f" {item}"
    return app.response_class(
        status=HTTPStatus.BAD_REQUEST,
        response=json.dumps({"message": message}),
        mimetype="application/json",
    )


@app.errorhandler(500)
def internal_error():
    return app.response_class(
        status=HTTPStatus.INTERNAL_SERVER_ERROR,
        response="Internal server error",
        mimetype="application/json",
    )