124 lines
4.7 KiB
Python
124 lines
4.7 KiB
Python
from fastapi import APIRouter, BackgroundTasks, Depends, Form, Request, Response
|
|
from fastapi.responses import HTMLResponse, RedirectResponse
|
|
from fastapi.templating import Jinja2Templates
|
|
from sqlalchemy import desc, func, or_, select
|
|
from sqlalchemy.orm import Session
|
|
|
|
from app.config import Settings, get_settings
|
|
from app.db import SessionLocal, get_db
|
|
from app.models import CrawlError, CrawlRun, Employee
|
|
from app.security import SESSION_COOKIE, require_admin, sign_session, verify_admin
|
|
from app.services.crawler import run_crawl
|
|
from app.version import BACKEND_VERSION, FRONTEND_VERSION
|
|
|
|
router = APIRouter(prefix="/admin")
|
|
templates = Jinja2Templates(directory="app/templates")
|
|
|
|
|
|
@router.get("", response_class=HTMLResponse)
|
|
def dashboard(request: Request, db: Session = Depends(get_db), settings: Settings = Depends(get_settings)):
|
|
require_admin(request, settings)
|
|
counts = {
|
|
"active": db.scalar(select(func.count()).select_from(Employee).where(Employee.status == "active")) or 0,
|
|
"dismissed": db.scalar(select(func.count()).select_from(Employee).where(Employee.status == "dismissed")) or 0,
|
|
"runs": db.scalar(select(func.count()).select_from(CrawlRun)) or 0,
|
|
"errors": db.scalar(select(func.count()).select_from(CrawlError)) or 0,
|
|
}
|
|
runs = db.scalars(select(CrawlRun).order_by(desc(CrawlRun.started_at)).limit(10)).all()
|
|
return _render(request, "dashboard.html", {"counts": counts, "runs": runs})
|
|
|
|
|
|
@router.get("/login", response_class=HTMLResponse)
|
|
def login_form(request: Request):
|
|
return _render(request, "login.html", {"error": None})
|
|
|
|
|
|
@router.post("/login")
|
|
def login(
|
|
response: Response,
|
|
request: Request,
|
|
username: str = Form(...),
|
|
password: str = Form(...),
|
|
settings: Settings = Depends(get_settings),
|
|
):
|
|
if not verify_admin(username, password, settings):
|
|
return _render(request, "login.html", {"error": "Неверный логин или пароль"}, status_code=401)
|
|
redirect = RedirectResponse("/admin", status_code=303)
|
|
redirect.set_cookie(SESSION_COOKIE, sign_session(username, settings), httponly=True, samesite="lax")
|
|
return redirect
|
|
|
|
|
|
@router.post("/logout")
|
|
def logout():
|
|
redirect = RedirectResponse("/admin/login", status_code=303)
|
|
redirect.delete_cookie(SESSION_COOKIE)
|
|
return redirect
|
|
|
|
|
|
@router.get("/employees", response_class=HTMLResponse)
|
|
def employees(
|
|
request: Request,
|
|
status: str | None = None,
|
|
q: str | None = None,
|
|
db: Session = Depends(get_db),
|
|
settings: Settings = Depends(get_settings),
|
|
):
|
|
require_admin(request, settings)
|
|
stmt = select(Employee)
|
|
if status:
|
|
stmt = stmt.where(Employee.status == status)
|
|
if q:
|
|
pattern = f"%{q}%"
|
|
stmt = stmt.where(or_(Employee.full_name.ilike(pattern), Employee.canonical_url.ilike(pattern)))
|
|
items = db.scalars(stmt.order_by(Employee.full_name).limit(200)).all()
|
|
return _render(request, "employees.html", {"employees": items, "status": status or "", "q": q or ""})
|
|
|
|
|
|
@router.get("/employees/{employee_id}", response_class=HTMLResponse)
|
|
def employee_detail(
|
|
employee_id: int,
|
|
request: Request,
|
|
db: Session = Depends(get_db),
|
|
settings: Settings = Depends(get_settings),
|
|
):
|
|
require_admin(request, settings)
|
|
employee = db.get(Employee, employee_id)
|
|
if not employee:
|
|
return RedirectResponse("/admin/employees", status_code=303)
|
|
snapshots = sorted(employee.snapshots, key=lambda item: item.captured_at, reverse=True)[:20]
|
|
return _render(request, "employee_detail.html", {"employee": employee, "snapshots": snapshots})
|
|
|
|
|
|
@router.get("/runs", response_class=HTMLResponse)
|
|
def runs(request: Request, db: Session = Depends(get_db), settings: Settings = Depends(get_settings)):
|
|
require_admin(request, settings)
|
|
items = db.scalars(select(CrawlRun).order_by(desc(CrawlRun.started_at)).limit(50)).all()
|
|
errors = db.scalars(select(CrawlError).order_by(desc(CrawlError.created_at)).limit(50)).all()
|
|
return _render(request, "runs.html", {"runs": items, "errors": errors})
|
|
|
|
|
|
@router.post("/runs")
|
|
def trigger_run(
|
|
request: Request,
|
|
background_tasks: BackgroundTasks,
|
|
settings: Settings = Depends(get_settings),
|
|
):
|
|
require_admin(request, settings)
|
|
|
|
def _crawl() -> None:
|
|
with SessionLocal() as db:
|
|
run_crawl(db, settings)
|
|
|
|
background_tasks.add_task(_crawl)
|
|
return RedirectResponse("/admin/runs", status_code=303)
|
|
|
|
|
|
def _render(request: Request, template: str, context: dict, status_code: int = 200) -> HTMLResponse:
|
|
payload = {
|
|
"request": request,
|
|
"backend_version": BACKEND_VERSION,
|
|
"frontend_version": FRONTEND_VERSION,
|
|
**context,
|
|
}
|
|
return templates.TemplateResponse(template, payload, status_code=status_code)
|