openwebrx-clone/owrx/controllers/__init__.py

from datetime import datetime, timezone


class BodySizeError(Exception):
    pass


class Controller(object):
    def __init__(self, handler, request, options):
        self.handler = handler
        self.request = request
        self.options = options
        self.responseCookies = None

    def send_response(
        self, content, code=200, content_type="text/html", last_modified: datetime = None, max_age=None, headers=None
    ):
        self.handler.send_response(code)
        if headers is None:
            headers = {}
        if content_type is not None:
            headers["Content-Type"] = content_type
        if last_modified is not None:
            headers["Last-Modified"] = last_modified.astimezone(tz=timezone.utc).strftime("%a, %d %b %Y %H:%M:%S GMT")
        if max_age is not None:
            headers["Cache-Control"] = "max-age={0}".format(max_age)
        for key, value in headers.items():
            self.handler.send_header(key, value)
        if self.responseCookies is not None:
            self.handler.send_header("Set-Cookie", self.responseCookies.output(header=""))
        self.handler.end_headers()
        if type(content) == str:
            content = content.encode()
        while len(content):
            w = self.handler.wfile.write(content)
            content = content[w:]

    def send_redirect(self, location, code=303):
        self.handler.send_response(code)
        if self.responseCookies is not None:
            self.handler.send_header("Set-Cookie", self.responseCookies.output(header=""))
        self.handler.send_header("Location", location)
        self.handler.end_headers()

    def set_response_cookies(self, cookies):
        self.responseCookies = cookies

    def get_body(self, max_size=None):
        if "Content-Length" not in self.handler.headers:
            return None
        length = int(self.handler.headers["Content-Length"])
        if max_size is not None and length > max_size:
            raise BodySizeError("HTTP body exceeds maximum allowed size")
        return self.handler.rfile.read(length)

    def handle_request(self):
        action = "indexAction"
        if "action" in self.options:
            action = self.options["action"]
        getattr(self, action)()