X-Git-Url: https://plomlompom.com/repos/feed.xml?a=blobdiff_plain;f=plomtask%2Fhttp.py;h=9956a7251501758d795032903afb0d52c0e195b6;hb=b16ae56b9c1a5bb799594fdd759a400d40488350;hp=3fde34b7b9f3ee99efa525a9532a81f4d4c95597;hpb=b557c789f4eec704db0e6276390395fac5d8db9e;p=plomtask diff --git a/plomtask/http.py b/plomtask/http.py index 3fde34b..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): - super().__init__(self, *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)