1 files changed, 51 insertions, 0 deletions

diff --git a/app/app_templating.py b/app/app_templating.py
new file mode 100644
index 0000000..427170b
--- /dev/null
+++ b/app/app_templating.py
@@ -0,0 +1,51 @@
+# UGE / L2 / Intro to relational databases / Python project prototype
+# Author: Pacien TRAN-GIRARD
+# Licence: EUPL-1.2
+
+from typing import Optional, NamedTuple
+from decimal import Decimal
+
+from fastapi import Depends, Request
+from fastapi.templating import Jinja2Templates
+
+from app_sessions import UserSession, FlashMessageQueue
+from app_database import db_transaction
+
+
+# Load and parse Jinja HTML templates pages and fragments.
+bare_templates = Jinja2Templates(directory='./templates/')
+
+
+class TemplateRenderer:
+    """
+    Template renderer using dependency injection to populate the template
+    parameters used by all pages.
+    """
+
+    def __init__(
+        self,
+        request: Request,
+        messages: FlashMessageQueue=Depends(FlashMessageQueue),
+        session: UserSession=Depends(UserSession),
+    ):
+        self._request = request
+        self._messages = messages
+        self._session = session
+
+    def _get_user(self) -> Optional[NamedTuple]:
+        if not self._session.is_logged_in():
+            return None
+
+        with db_transaction() as tx:
+            return tx.fetch_account(user_id=self._session.get_user_id())
+
+    def _shared_template_args(self) -> dict:
+        return {
+            'request': self._request,
+            'messages': self._messages,
+            'user': self._get_user(),
+        }
+
+    def __call__(self, template: str, **kwargs):
+        template_args = self._shared_template_args() | kwargs
+        return bare_templates.TemplateResponse(template, template_args)