Christian Krakau-Louis
44 lines
1.4 KiB
Python
44 lines
1.4 KiB
Python
"""
|
|
Base setup for views, containing shared functionality and imports.
|
|
"""
|
|
from fastapi import APIRouter, Request, Depends, HTTPException
|
|
from fastapi.templating import Jinja2Templates
|
|
from pathlib import Path
|
|
from sqlalchemy.orm import Session
|
|
import logging
|
|
|
|
from app.auth import require_login
|
|
from app.database import SessionLocal
|
|
from app.config import settings
|
|
|
|
# Set up Jinja2 templates
|
|
templates_dir = Path(__file__).parent.parent.parent / "frontend" / "templates"
|
|
templates = Jinja2Templates(directory=str(templates_dir))
|
|
|
|
# Customize Jinja2Templates to include app_version in all templates
|
|
original_template_response = templates.TemplateResponse
|
|
|
|
def template_response_with_version(*args, **kwargs):
|
|
"""Wrapper for TemplateResponse to include version in all templates"""
|
|
# If context dict is provided, add version to it
|
|
if len(args) >= 2 and isinstance(args[1], dict):
|
|
args[1].setdefault("version", settings.version)
|
|
elif "context" in kwargs and isinstance(kwargs["context"], dict):
|
|
kwargs["context"].setdefault("version", settings.version)
|
|
return original_template_response(*args, **kwargs)
|
|
|
|
templates.TemplateResponse = template_response_with_version
|
|
|
|
# Set up logging
|
|
logger = logging.getLogger(__name__)
|
|
|
|
def get_db():
|
|
"""
|
|
Dependency to get a database session.
|
|
"""
|
|
db = SessionLocal()
|
|
try:
|
|
yield db
|
|
finally:
|
|
db.close()
|