summaryrefslogtreecommitdiff
path: root/day9/task5/main.py
blob: 8cdf7a1c4603e637715717dd90761a4a6810d46f (plain)
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
from http.server import HTTPServer, BaseHTTPRequestHandler
import logging


class MyHTTPRequestHandler(BaseHTTPRequestHandler):
    def update_post(self, query):
        return f'<h1>{query}</h1>'

    def index_get(self):
        return 'INDEX!!!!'

    def _set_response(self):
        self.send_response(200)
        self.send_header('Content-type', 'text/html')
        self.end_headers()

    def do_GET(self):
        if self.path == '/':
            response = self.index_get()
        else:
            response = '<center><h1>ERROR 404 NOT FOUND</h1></center>'

        self._set_response()
        self.wfile.write(response.encode('utf-8'))

    def do_POST(self):
        content_length = int(self.headers['Content-Length'])
        post_data = self.rfile.read(content_length).decode('utf-8')

        response = self._router['POST'].get(
            self.path, lambda *args: '<center><h1>ERROR 404 NOT FOUND</h1></center>'
        )(post_data)

        self._set_response()
        self.wfile.write(response.encode('utf-8'))


server_address = ("", 8000)
httpd = HTTPServer(server_address, MyHTTPRequestHandler)
httpd.serve_forever()