home · contact · privacy
Add single-day view / retrieval.
[plomtask] / plomtask / http.py
index 1af49d16b197bc607310f0f7c710b407b93e905b..b6019ef0761b1dc3a86d3a12f395cd5b96f40c5c 100644 (file)
@@ -1,7 +1,7 @@
-"""plom's task manager"""
+"""Web server stuff."""
 from http.server import BaseHTTPRequestHandler
 from http.server import HTTPServer
-from urllib.parse import urlparse
+from urllib.parse import urlparse, parse_qs
 from os.path import split as path_split
 from jinja2 import Environment as JinjaEnv, FileSystemLoader as JinjaFSLoader
 from plomtask.days import Day
@@ -36,24 +36,29 @@ class TaskHandler(BaseHTTPRequestHandler):
     def do_GET(self):
         """Handle any GET request."""
         try:
+            conn = DatabaseConnection(self.server.db)
             parsed_url = urlparse(self.path)
             site = path_split(parsed_url.path)[1]
+            params = parse_qs(parsed_url.query)
             if 'calendar' == site:
-                html = self.do_GET_calendar()
+                html = self.do_GET_calendar(conn)
+            elif 'day' == site:
+                date = params.get('date', ['2024-01-01'])[0]
+                html = self.do_GET_day(conn, date)
             else:
                 raise HandledException('Test!')
+            conn.commit()
+            conn.close()
             self.send_html(html)
         except HandledException as error:
             self.send_msg(error)
 
-    def do_GET_calendar(self):
-        """Show sorted Days."""
-        conn = DatabaseConnection(self.server.db)
-        Day('2024-01-03').save(conn)
-        Day('2024-01-01').save(conn)
-        Day('2024-01-02').save(conn)
-        days = Day.all(conn)
-        conn.commit()
-        conn.close()
+    def do_GET_calendar(self, conn: DatabaseConnection):
+        """Show Days."""
         return self.server.jinja.get_template('calendar.html').render(
-                days=days)
+                days=Day.all(conn))
+
+    def do_GET_day(self, conn: DatabaseConnection, date: str):
+        """Show single Day."""
+        day = Day.by_date(conn, date)
+        return self.server.jinja.get_template('day.html').render(day=day)