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()