X-Git-Url: https://plomlompom.com/repos/foo.html?a=blobdiff_plain;f=plomtask%2Fhttp.py;h=b6019ef0761b1dc3a86d3a12f395cd5b96f40c5c;hb=e51477ebea9fb701d88de85088b2fda1247c73a2;hp=baa730ed7a77272982cb03c8ec61ea76f22d0a12;hpb=30bc39c0af52c9fc474d3be0c040b7effe761187;p=plomtask diff --git a/plomtask/http.py b/plomtask/http.py index baa730e..b6019ef 100644 --- a/plomtask/http.py +++ b/plomtask/http.py @@ -1,18 +1,20 @@ -"""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 from plomtask.misc import HandledException +from plomtask.db import DatabaseConnection class TaskServer(HTTPServer): """Variant of HTTPServer that knows .jinja as Jinja Environment.""" - def __init__(self, templates_dir, *args, **kwargs): + def __init__(self, templates_dir, db_file, *args, **kwargs): super().__init__(*args, **kwargs) + self.db = db_file self.jinja = JinjaEnv(loader=JinjaFSLoader(templates_dir)) @@ -34,19 +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.""" - days = [Day('2024-01-03'), Day('2024-01-01'), Day('2024-01-02')] - days.sort() + 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)