lecturer.py

import logging
from typing import Any, Literal

from auth_lib.fastapi import UnionAuth
from fastapi import APIRouter, Depends
from fastapi_sqlalchemy import db
from sqlalchemy.orm import Query, joinedload

from calendar_backend.exceptions import ObjectNotFound
from calendar_backend.methods.image import get_photo_webpath
from calendar_backend.models.db import ApproveStatuses, Lecturer
from calendar_backend.models.db import Photo as DbPhoto
from calendar_backend.routes.models import GetListLecturer, LecturerGet, LecturerPatch, LecturerPost
from calendar_backend.settings import get_settings


settings = get_settings()
logger = logging.getLogger(__name__)
router = APIRouter(prefix="/lecturer", tags=["Lecturer"])


@router.get("/{id}", response_model=LecturerGet)
async def get_lecturer_by_id(id: int) -> LecturerGet:
    lecturer = Lecturer.get(id, session=db.session)
    result = LecturerGet.model_validate(Lecturer.get(id, session=db.session))
    if lecturer.avatar_id:
        result.avatar_link = get_photo_webpath(lecturer.avatar.link)
    return result


@router.get("/", response_model=GetListLecturer)
async def get_lecturers(
    query: str = "",
    limit: int = 10,
    offset: int = 0,
    order_by: Literal['first_name', 'last_name'] | None = None,
) -> dict[str, Any]:
    query: Query = Lecturer.get_all(session=db.session).filter(Lecturer.search(query))
    query = query.options(joinedload(Lecturer.avatar))  # Сразу загружаем аватарки
    if order_by:
        query = query.order_by(order_by)
    query = query.order_by('id')
    if limit:
        cnt, query = query.count(), query.offset(offset).limit(limit)
    else:
        cnt, query = query.count(), query.offset(offset)
    query = query.all()
    logger.debug(query)

    result = []
    for row in query:
        row_get = LecturerGet.model_validate(row)
        if row.avatar:
            row_get.avatar_link = get_photo_webpath(row.avatar.link)
        result.append(row_get)
    return {
        "items": result,
        "limit": limit,
        "offset": offset,
        "total": cnt,
    }


@router.post("/", response_model=LecturerGet)
async def create_lecturer(
    lecturer: LecturerPost, _=Depends(UnionAuth(scopes=["timetable.lecturer.create"]))
) -> LecturerGet:
    dblecturer = Lecturer.create(session=db.session, **lecturer.model_dump())
    db.session.commit()
    return LecturerGet.model_validate(dblecturer)


@router.patch("/{id}", response_model=LecturerGet)
async def patch_lecturer(
    id: int, lecturer_inp: LecturerPatch, _=Depends(UnionAuth(scopes=["timetable.lecturer.update"]))
) -> LecturerGet:
    if lecturer_inp.avatar_id:
        photo = DbPhoto.get(lecturer_inp.avatar_id, session=db.session)
        if photo.lecturer_id != id or photo.approve_status != ApproveStatuses.APPROVED:
            raise ObjectNotFound(DbPhoto, lecturer_inp.avatar_id)
        lecturer_upd = Lecturer.update(
            id,
            session=db.session,
            **lecturer_inp.model_dump(exclude_unset=True),
            avatar_link=get_photo_webpath(photo.link),
        )
    else:
        lecturer_upd = Lecturer.update(id, session=db.session, **lecturer_inp.model_dump(exclude_unset=True))
    db.session.commit()
    return LecturerGet.model_validate(lecturer_upd)


@router.delete("/{id}", response_model=None)
async def delete_lecturer(id: int, _=Depends(UnionAuth(scopes=["timetable.lecturer.delete"]))) -> None:
    Lecturer.delete(id, session=db.session)
    db.session.commit()