X-Git-Url: https://plomlompom.com/repos/%7B%7Bprefix%7D%7D/copy_structured?a=blobdiff_plain;f=task.py;h=3af1cacadc4c30700ef18c748ff81ee9f9a01ed2;hb=7c7602eae676c9806292ad37c58d22dcce12a14f;hp=a1510b0a202571796a5da96262fc115fb3e85a84;hpb=5744cbdb67c253d8a8d54eb9710958a66a64fbb6;p=plomtask diff --git a/task.py b/task.py index a1510b0..3af1cac 100755 --- a/task.py +++ b/task.py @@ -1,11 +1,14 @@ #!/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' class HandledException(Exception): @@ -15,17 +18,42 @@ class HandledException(Exception): class TaskHandler(BaseHTTPRequestHandler): """Handles single HTTP request.""" - def do_GET(self): - """Handle any GET request.""" - self.send_response(200) + def send_html(self, html, code=200): + """Send HTML as proper HTTP response.""" + self.send_response(code) self.end_headers() - html = '

hi there!

' self.wfile.write(bytes(html, 'utf-8')) + def send_msg(self, msg, code=400): + """Send message in HTML formatting as HTTP response.""" + html = self.server.jinja.get_template('msg.html').render(msg=msg) + self.send_html(html, code) + + def do_GET(self): + """Handle any GET request.""" + try: + parsed_url = urlparse(self.path) + site = path_split(parsed_url.path)[1] + if 'calendar' == site: + html = self.do_GET_calendar() + else: + raise HandledException('Test!') + self.send_html(html) + except HandledException as error: + self.send_msg(error) + + 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.""" server = HTTPServer(('localhost', HTTP_PORT), TaskHandler) + server.jinja = JinjaEnv(loader=JinjaFSLoader(TEMPLATES_DIR)) print(f'running at port {HTTP_PORT}') try: server.serve_forever()