X-Git-Url: https://plomlompom.com/repos/feed.xml?a=blobdiff_plain;f=task.py;h=963143463bfcbba7125415ed1aa025f19adbb9a4;hb=51cdb0020cf8f1c0a0c11cc80be98863fce2448b;hp=a1510b0a202571796a5da96262fc115fb3e85a84;hpb=5744cbdb67c253d8a8d54eb9710958a66a64fbb6;p=plomtask diff --git a/task.py b/task.py index a1510b0..9631434 100755 --- a/task.py +++ b/task.py @@ -1,31 +1,67 @@ #!/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): """To identify Exceptions based on expected (if faulty) user behavior.""" +class TaskServer(HTTPServer): + """Variant of HTTPServer that knows .jinja as Jinja Environment.""" + + def __init__(self, *args, **kwargs): + super().__init__(self, *args, **kwargs) + self.jinja = JinjaEnv(loader=JinjaFSLoader(TEMPLATES_DIR)) + + class TaskHandler(BaseHTTPRequestHandler): """Handles single HTTP request.""" + server: TaskServer - def do_GET(self): - """Handle any GET request.""" - self.send_response(200) + def send_html(self, html: str, code: int = 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: str, code: int = 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 = TaskServer(('localhost', HTTP_PORT), TaskHandler) print(f'running at port {HTTP_PORT}') try: server.serve_forever()