actual_server.py

from server import HTTPServer
import mimetypes
from urllib.parse import urljoin
mimetypes.init()
import graph

class ExampleHTTPServer(HTTPServer):
    URL_REWRITE = {b'/': b'/index.html'}
    def __init__(self, *args, **kwargs):
        HTTPServer.__init__(self,*args, **kwargs)
        self.graph = graph.GraphDrawer()
        
    def generate_response(self, caddr, headers):
        if headers[b'URL'] == b'/generate_light_graph.php':
            print('Generating light graph')
            self.graph.draw_light_level_graph()
        elif headers[b'URL'] == b'/generate_moisture_graph.php':
            print('Generating moisture graph')
            self.graph.draw_moisture_level_graph()
        try:
            with open(urljoin(b'.', headers[b'URL'].lstrip(b'/')), 'rb') as file:
                data = file.read()
            file_mime_type = mimetypes.guess_type(
                headers[b'URL'].decode('utf-8'))[0].encode()
            code = b'200 OK'
            headers = {b'Server': b'PyServer/0.1',
                   b'Content-Type': file_mime_type,
                       b'Cache-Control': b'no-store'}
        except FileNotFoundError:
            with open(b'404.html', 'rb') as file:
                data = file.read()
            code = b'404 NOT FOUND'
            headers = {b'Server': b'PyServer/0.1',
                   b'Content-Type': b'text/html'}
            
        return code, headers, data

def main():
    server = ExampleHTTPServer('', 8000, timeout=10, listen=50000)
    server.serve()

if __name__ == '__main__':
#    import cProfile as profile
#    profile.run('main()', sort='tottime')
    main()