http_server.py

from encodings import utf_8
from http.server import HTTPServer, SimpleHTTPRequestHandler
from abstract_server import Server
import logging

logger = logging.getLogger(__name__)

class HttpServer(Server):
    class Handler(SimpleHTTPRequestHandler):
        def __init__(self, *args, **kwargs):
            self.http_directory = "./http-files"
            super().__init__(*args, directory=self.http_directory, **kwargs)

        def do_POST(self) -> None:
            self.send_response(200)
            self.end_headers()
            json = self.rfile.read(int(self.headers["Content-Length"]))
            self.wfile.write(bytes("", "utf_8").join([b"{\"status\": \"ok\", \"message\": \"POST request received.\", \"data\": ", (json if len(json) > 0 else b"null"), b"}"]))

        def do_PUT(self) -> None:
            self.send_response(200)
            self.end_headers()
            json = self.rfile.read(int(self.headers["Content-Length"]))
            self.wfile.write(bytes("", "utf_8").join([b"{\"status\": \"ok\", \"message\": \"PUT request received.\", \"data\": ", (json if len(json) > 0 else b"null"), b"}"]))

        def do_DELETE(self) -> None:
            self.send_response(200)
            self.end_headers()
            self.wfile.write(b"{\"status\": \"ok\", \"message\": \"DELETE request received.\"}")

        def do_PATCH(self) -> None:
            self.send_response(200)
            self.end_headers()
            self.wfile.write(b"{\"status\": \"ok\", \"message\": \"PATCH request received.\"}")


    def __init__(self) -> None:
        formatter = "%(name)s - - [%(asctime)s] %(message)s"
        logging.basicConfig(level=logging.INFO, format=formatter)
        self.address = ("", 80)

    def set_address(self, host: str, port: int) -> None:
        self.address = (host, port)

    def start(self) -> None:
        self._server = HTTPServer(self.address, self.Handler)
        logger.info(f"HTTP-server is ready and serving on {self._server.server_address[0]}:{self._server.server_port}.")
        self._server.serve_forever()

    def stop(self) -> None:
        self._server.server_close()
        logger.info("HTTP-server stopped...")
        self._server = None


if __name__ == "__main__":
    http_server = HttpServer()
    try:
        http_server.start()
    except KeyboardInterrupt:
        http_server.stop()