X-Git-Url: https://plomlompom.com/repos/bar%20baz.html?a=blobdiff_plain;ds=sidebyside;f=plomtask%2Fhttp.py;h=9956a7251501758d795032903afb0d52c0e195b6;hb=b16ae56b9c1a5bb799594fdd759a400d40488350;hp=baa730ed7a77272982cb03c8ec61ea76f22d0a12;hpb=30bc39c0af52c9fc474d3be0c040b7effe761187;p=plomtask diff --git a/plomtask/http.py b/plomtask/http.py index baa730e..9956a72 100644 --- a/plomtask/http.py +++ b/plomtask/http.py @@ -6,13 +6,15 @@ 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)) @@ -46,7 +48,13 @@ class TaskHandler(BaseHTTPRequestHandler): def do_GET_calendar(self): """Show sorted Days.""" - days = [Day('2024-01-03'), Day('2024-01-01'), Day('2024-01-02')] + 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() days.sort() return self.server.jinja.get_template('calendar.html').render( days=days)