X-Git-Url: https://plomlompom.com/repos/%7B%7Bprefix%7D%7D/copy_structured?a=blobdiff_plain;f=plomtask%2Fhttp.py;h=9956a7251501758d795032903afb0d52c0e195b6;hb=b16ae56b9c1a5bb799594fdd759a400d40488350;hp=4ce72ec69e823f07f4ce718cd82aea66024f9c0d;hpb=ee501dc4d2b67747e2bfb626bfa65f44ad237f61;p=plomtask diff --git a/plomtask/http.py b/plomtask/http.py index 4ce72ec..9956a72 100644 --- a/plomtask/http.py +++ b/plomtask/http.py @@ -6,16 +6,16 @@ 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 - -TEMPLATES_DIR = 'templates' +from plomtask.db import DatabaseConnection class TaskServer(HTTPServer): """Variant of HTTPServer that knows .jinja as Jinja Environment.""" - def __init__(self, *args, **kwargs): - x = super().__init__(*args, **kwargs) - self.jinja = JinjaEnv(loader=JinjaFSLoader(TEMPLATES_DIR)) + def __init__(self, templates_dir, db_file, *args, **kwargs): + super().__init__(*args, **kwargs) + self.db = db_file + self.jinja = JinjaEnv(loader=JinjaFSLoader(templates_dir)) class TaskHandler(BaseHTTPRequestHandler): @@ -48,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)