1 """Web server stuff."""
3 from http.server import BaseHTTPRequestHandler
4 from http.server import HTTPServer
5 from urllib.parse import urlparse, parse_qs
6 from os.path import split as path_split
7 from jinja2 import Environment as JinjaEnv, FileSystemLoader as JinjaFSLoader
8 from plomtask.days import Day, todays_date
9 from plomtask.exceptions import HandledException, BadFormatException, \
11 from plomtask.db import DatabaseConnection, DatabaseFile
12 from plomtask.processes import Process
14 TEMPLATES_DIR = 'templates'
17 class TaskServer(HTTPServer):
18 """Variant of HTTPServer that knows .jinja as Jinja Environment."""
20 def __init__(self, db_file: DatabaseFile,
21 *args: Any, **kwargs: Any) -> None:
22 super().__init__(*args, **kwargs)
24 self.jinja = JinjaEnv(loader=JinjaFSLoader(TEMPLATES_DIR))
27 class TaskHandler(BaseHTTPRequestHandler):
28 """Handles single HTTP request."""
31 def do_GET(self) -> None:
32 """Handle any GET request."""
34 conn, site, params = self._init_handling()
35 if 'calendar' == site:
36 start = params.get('start', [''])[0]
37 end = params.get('end', [''])[0]
38 html = self.do_GET_calendar(conn, start, end)
40 date = params.get('date', [todays_date()])[0]
41 html = self.do_GET_day(conn, date)
42 elif 'process' == site:
43 id_ = params.get('id', [None])[0]
45 id__ = int(id_) if id_ else None
46 except ValueError as e:
47 raise BadFormatException(f'Bad ?id= value: {id_}') from e
48 html = self.do_GET_process(conn, id__)
49 elif 'processes' == site:
50 html = self.do_GET_processes(conn)
52 self._redirect('/day')
55 raise NotFoundException(f'Unknown page: /{site}')
59 except HandledException as error:
60 self._send_msg(error, code=error.http_code)
62 def do_GET_calendar(self, conn: DatabaseConnection,
63 start: str, end: str) -> str:
65 days = Day.all(conn, date_range=(start, end), fill_gaps=True)
66 return self.server.jinja.get_template('calendar.html').render(
67 days=days, start=start, end=end)
69 def do_GET_day(self, conn: DatabaseConnection, date: str) -> str:
70 """Show single Day."""
71 day = Day.by_date(conn, date, create=True)
72 return self.server.jinja.get_template('day.html').render(day=day)
74 def do_GET_process(self, conn: DatabaseConnection, id_: int | None) -> str:
75 """Show process of id_."""
76 return self.server.jinja.get_template('process.html').render(
77 process=Process.by_id(conn, id_, create=True))
79 def do_GET_processes(self, conn: DatabaseConnection) -> str:
80 """Show all Processes."""
81 return self.server.jinja.get_template('processes.html').render(
82 processes=Process.all(conn))
84 def do_POST(self) -> None:
85 """Handle any POST request."""
87 conn, site, params = self._init_handling()
88 length = int(self.headers['content-length'])
89 postvars = parse_qs(self.rfile.read(length).decode(),
90 keep_blank_values=True)
92 date = params.get('date', [''])[0]
93 self.do_POST_day(conn, date, postvars)
94 elif 'process' == site:
95 id_ = params.get('id', [''])[0]
97 id__ = int(id_) if id_ else None
98 except ValueError as e:
99 raise BadFormatException(f'Bad ?id= value: {id_}') from e
100 self.do_POST_process(conn, id__, postvars)
104 except HandledException as error:
105 self._send_msg(error, code=error.http_code)
107 def do_POST_day(self, conn: DatabaseConnection,
108 date: str, postvars: dict[str, list[str]]) -> None:
109 """Update or insert Day of date and fields defined in postvars."""
110 day = Day.by_date(conn, date, create=True)
111 day.comment = postvars['comment'][0]
114 def do_POST_process(self, conn: DatabaseConnection, id_: int | None,
115 postvars: dict[str, list[str]]) -> None:
116 """Update or insert Process of id_ and fields defined in postvars."""
117 process = Process.by_id(conn, id_, create=True)
118 process.title.set(postvars['title'][0])
119 process.description.set(postvars['description'][0])
120 effort = postvars['effort'][0]
122 process.effort.set(float(effort))
123 except ValueError as e:
124 raise BadFormatException(f'Bad effort value: {effort}') from e
127 def _init_handling(self) -> \
128 tuple[DatabaseConnection, str, dict[str, list[str]]]:
129 conn = DatabaseConnection(self.server.db)
130 parsed_url = urlparse(self.path)
131 site = path_split(parsed_url.path)[1]
132 params = parse_qs(parsed_url.query)
133 return conn, site, params
135 def _redirect(self, target: str) -> None:
136 self.send_response(302)
137 self.send_header('Location', target)
140 def _send_html(self, html: str, code: int = 200) -> None:
141 """Send HTML as proper HTTP response."""
142 self.send_response(code)
144 self.wfile.write(bytes(html, 'utf-8'))
146 def _send_msg(self, msg: Exception, code: int = 400) -> None:
147 """Send message in HTML formatting as HTTP response."""
148 html = self.server.jinja.get_template('msg.html').render(msg=msg)
149 self._send_html(html, code)