handler.py

import http.server
import requests
import logger
from datetime import datetime

class ProxyHandler(http.server.BaseHTTPRequestHandler):

    def __init__(self, request, client_address, server):
        self.logger = logger.Logger()
        super().__init__(request, client_address, server)

    def __getattr__(self, name):
        if name.startswith("do_"):
            return self.handle_request
        raise AttributeError(name)
        
    def handle_request(self):
        print("Handler", self.command, self.path)
        start_time = datetime.now()
        response = self.forward_request()
        end_time = datetime.now()
        duration_ms = (end_time - start_time).total_seconds() * 1000
        self.log_transaction(response, self.body, duration_ms)
        self.send_response_to_client(response)


    def forward_request(self):
        method = self.command
        headers = self.headers
        url = self.path

        content_length = headers.get("Content-Length")

        if content_length:
            self.body = self.rfile.read(int(content_length))
        else:
            self.body = None

        response = requests.request(method=method, url=url, headers=headers, data=self.body)

        return response

    def log_transaction(self, response, request_body, duration_ms):
        self.logger.save_transaction(self.command, self.path, dict(self.headers), request_body, response.status_code, dict(response.headers), response.content, duration_ms)

    def send_response_to_client(self, response):
        status_code = response.status_code
        headers = response.headers
        content = response.content

        headers = dict(response.headers)
        headers.pop("Transfer-Encoding", None)
        headers.pop("Content-Encoding", None)

        self.send_response(status_code)
        for key, value in headers.items():
            self.send_header(key, value)

        self.end_headers()

        self.wfile.write(content)