Use different do_GET handlers for different paths.

[plomtask] / task.py

diff --git a/task.py b/task.py
index 2a6c699f99cb956192a33727a69cb7f808cd8ff2..0b8825476353fff066f4b5480c7fcbf2d5ef00b5 100755 (executable)
--- a/task.py
+++ b/task.py
@@ -1,10 +1,11 @@
 #!/usr/bin/env python3
-"""
-plom's task manager
-"""
+"""plom's task manager"""
 from http.server import BaseHTTPRequestHandler
 from http.server import HTTPServer
+from urllib.parse import urlparse
+from os.path import split as path_split
 from jinja2 import Environment as JinjaEnv, FileSystemLoader as JinjaFSLoader
+from days import Day
 
 HTTP_PORT = 8082
 TEMPLATES_DIR = 'templates'
@@ -19,11 +20,28 @@ class TaskHandler(BaseHTTPRequestHandler):
 
     def do_GET(self):
         """Handle any GET request."""
+        parsed_url = urlparse(self.path)
+        site = path_split(parsed_url.path)[1]
+        if 'calendar' == site:
+            html = self.do_GET_calendar()
+        else:
+            html = self.do_GET_msg('I have nothing to say at this point.')
         self.send_response(200)
         self.end_headers()
-        html = self.server.jinja.get_template('index.html').render()
         self.wfile.write(bytes(html, 'utf-8'))
 
+    def do_GET_msg(self, msg):
+        """Show message."""
+        return self.server.jinja.get_template('msg.html').render(
+                msg=msg)
+
+    def do_GET_calendar(self):
+        """Show sorted Days."""
+        days = [Day('2024-01-03'), Day('2024-01-01'), Day('2024-01-02')]
+        days.sort()
+        return self.server.jinja.get_template('calendar.html').render(
+                days=days)
+
 
 def main():
     """Main loop."""