SimplePythonHttpRouter/server.py at main · Exinie/SimplePythonHttpRouter · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
from http.server import BaseHTTPRequestHandler, HTTPServer
from urllib.parse import parse_qs
import os
import mimetypes

from config import IP, PORT, STATIC_DIR
from routes import RouteHandler

route_handler = RouteHandler()


class RequestHandler(BaseHTTPRequestHandler):

    def send_error_response(self, status, title, message, target_file):
        print(status, title, message, target_file)
        self.send_response(status)
        self.send_header('Content-type', 'text/html; charset=utf-8')
        self.end_headers()
        self.wfile.write(f"<html><body><h1>{status} - {title}</h1><p>{message}</p></body></html>".encode('utf-8'))
        return

    def do_GET(self):

        if ("GET", self.path) in route_handler.ROUTES:
            target__file = route_handler.ROUTES[("GET", self.path)]
            target_file_location = os.path.join(STATIC_DIR, target__file)

            # Check if the HTML file for the intended route exists
            if os.path.exists(target_file_location) and os.path.isfile(target_file_location):
                try:
                    # Read file
                    with open(target_file_location, "r", encoding="utf-8") as file:
                        file_content = file.read()

                        for key, value in route_handler.PLACEHOLDERS.items():  # Replace placeholders with associated strings
                            placeholder = f"{{ {key} }}"
                            file_content = file_content.replace(placeholder, str(value))

                        self.send_response(200)  # Send 200 OK response
                        mimetype, _ = mimetypes.guess_type(target_file_location)

                        if target__file.lower().endswith((".html", ".htm")):
                            self.send_header("Content-type", "text/html; charset=utf-8")
                        else:
                            self.send_header("Content-type", mimetype or "application/octet-stream")

                        self.end_headers()  # Finish header
                        self.wfile.write(file_content.encode("utf-8"))  # Write edited file as response
                    return
                except Exception as e:
                    # If error while serving file (500)
                    print(f"GET error: {e}")
                    self.send_error_response(500, "Internal server error", "", self.path)
                    return
            else:
                # If file for the intended route is not exist
                self.send_error_response(404, "Not found", "", self.path)
                return
        else:
            # If route not exist in ROUTES dictionary
            self.send_error_response(404, "Not found", "", self.path)
            return

    def do_POST(self):
        if ("POST", self.path) in route_handler.ROUTES:
            try:
                content_length = int(self.headers.get("Content-Length", 0))
                post_body = self.rfile.read(content_length).decode('utf-8')

                # Parsing URL, example output: username=Exinie&password=exinie123!
                post_data = parse_qs(post_body)

                handler = route_handler.ROUTES[("POST", self.path)]
                handler(post_data)

                # Response
                self.send_response(200)
                self.send_header("Location", "/index.html")
                self.end_headers()
                return
            except Exception as e:
                # If error while reading data (500)
                print(f"POST error: {e}")
                self.send_error_response(500, "Internal server error", "")
                return
        else:
            self.send_error_response(404, "Not found", "",)
            return


def run_server():

    try:
        server = HTTPServer((IP, PORT), RequestHandler)
        server.serve_forever()
    except KeyboardInterrupt:
        print(" Closing server.")
        server.server_close
    except Exception as e:
        print(f"Error starting server: {e}")


if __name__ == "__main__":
    run_server()