import logging
from flask import Flask, Response, json, request
from sms_broker_parsers import get_mimetype_parser
from sms_broker_parsers.errors import NoMatchingSchema, NoParserForMimeType
from .utils import allowed_sender, log_request, print_message
APP = Flask(__name__)
logging.basicConfig(level=logging.DEBUG)
@APP.route("/", methods=["GET"])
def main() -> Response:
return Response(status=204)
@APP.route("/sms-printer", methods=["POST"])
def sms_printer() -> Response:
# If this throws, flask takes care of it
sms_input = request.get_json(force=True)
try:
parser = get_mimetype_parser(request.mimetype)
except NoParserForMimeType:
logging.warning("No parsers available for the given mimetype!")
logging.debug("mimetype: {}".format(request.mimetype))
logging.debug("header: {}".format(request.headers["Content-Type"]))
return Response(status=415)
try:
sms_list = parser(sms_input)
log_request(sms_input)
except NoMatchingSchema:
logging.warning("Unrecognized message format!")
logging.debug(json.dumps(sms_input))
return Response(status=415)
for sms in sms_list:
logging.info("Recieved sms from %s", sms.sender)
if not allowed_sender(sms.sender):
logging.warning("Unauthorized sender %s", sms.sender)
return Response(status=401)
logging.info("Printing sms from %s", sms.sender)
logging.debug("Text: %s", sms.text)
if not print_message(sms.text):
logging.error("Error printing sms")
return Response(status=500)
return Response(status=200)