WIP: Рендеринг таблицы на стороне клиента.

1 files changed, 18 insertions, 5 deletions

diff --git a/day9/task5/server.py b/day9/task5/server.py
index d9fdc78..47e29f9 100644
--- a/day9/task5/server.py
+++ b/day9/task5/server.py
@@ -1,21 +1,31 @@
 from http.server import HTTPServer, BaseHTTPRequestHandler
 from urllib.parse import parse_qs
+from json import dumps
 
 from router import run
 from utils import HTTP_STATUS_CODES
 
 
 class MyHTTPRequestHandler(BaseHTTPRequestHandler):
-    def _set_response(self, code):
+    def _set_response(self, code, content_type):
         self.send_response(code)
-        self.send_header('Content-type', 'text/html')
+        self.send_header('Content-type', content_type)
         self.end_headers()
 
     def do_GET(self):
+        try:
+            content_length = int(self.headers['Content-Length'])
+            get_data = parse_qs(self.rfile.read(content_length).decode('utf-8'))
+        except TypeError:
+            get_data = {}
+
+        for key in get_data:
+            get_data[key] = get_data[key][0]
+
         self.finalize_request(run({
             'url': self.path,
             'method': 'GET',
-            'query': {}
+            'query': get_data
         }))
 
     def do_POST(self):
@@ -33,10 +43,13 @@ class MyHTTPRequestHandler(BaseHTTPRequestHandler):
 
     def finalize_request(self, response):
         if isinstance(response, int):
-            self._set_response(response)
+            self._set_response(response, 'text/html')
             response = f'<center><h1>ERROR {response} {HTTP_STATUS_CODES[response].upper()}</h1></center>'
+        elif isinstance(response, (dict, list, tuple)):
+            self._set_response(200, 'application/json')
+            response = dumps(response)
         else:
-            self._set_response(200)
+            self._set_response(200, 'text/html')
 
         self.wfile.write(response.encode('utf-8'))