77 lines
2.3 KiB
Python
77 lines
2.3 KiB
Python
from fastapi import APIRouter, Depends, Query
|
|
from sqlalchemy import func, select
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from ..core.authx import auth
|
|
from ..core.db import get_db
|
|
from ..models.photo_model import Category, Photo
|
|
from ..schemas.photo_schemas import CategoriesOut, PhotosOut, PhotosPaginatedOut
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.get("/photos", response_model=PhotosPaginatedOut, dependencies=[Depends(auth.access_token_required)])
|
|
async def get_photos(
|
|
page: int = Query(1, ge=1),
|
|
per_page: int = Query(24, ge=1),
|
|
session: AsyncSession = Depends(get_db), # noqa: B008
|
|
category: int | None = None,
|
|
):
|
|
offset = (page - 1) * per_page
|
|
|
|
query = select(Photo)
|
|
if category is not None:
|
|
query = query.where(Photo.category_id == category)
|
|
|
|
query = query.order_by(Photo.id).offset(offset).limit(per_page)
|
|
|
|
result = await session.execute(query)
|
|
photos = result.scalars().all()
|
|
|
|
return {
|
|
"photos": photos,
|
|
"page_inc": f"/photos?page={page + 1}&per_page={per_page}&category={category}",
|
|
"page_dec": f"/photos?page={max(1, page - 1)}&per_page={per_page}&category={category}",
|
|
}
|
|
|
|
|
|
@router.get("/categories", response_model=CategoriesOut, dependencies=[Depends(auth.access_token_required)])
|
|
async def categories(session: AsyncSession = Depends(get_db)): # noqa: B008
|
|
stmt = (
|
|
select(Category.id, Category.title, func.count(Photo.id).label("photos_count"))
|
|
.select_from(Category)
|
|
.outerjoin(Category.photos)
|
|
.group_by(Category.id, Category.title)
|
|
.order_by(Category.title)
|
|
)
|
|
rows = (await session.execute(stmt)).all()
|
|
|
|
return {
|
|
"categories": [
|
|
{"id": r.id, "title": r.title, "photos_count": r.photos_count} for r in rows
|
|
]
|
|
}
|
|
|
|
|
|
@router.get("/category", response_model=PhotosOut)
|
|
async def category(
|
|
page: int = Query(1, ge=1),
|
|
per_page: int = Query(24, ge=1),
|
|
session: AsyncSession = Depends(get_db), # noqa: B008
|
|
id: int = Query(...),
|
|
):
|
|
offset = (page - 1) * per_page
|
|
|
|
query = (
|
|
select(Photo)
|
|
.where(Photo.category_id == id)
|
|
.order_by(Photo.id)
|
|
.offset(offset)
|
|
.limit(per_page)
|
|
)
|
|
|
|
result = await session.execute(query)
|
|
photos = result.scalars().all()
|
|
|
|
return {"photos": photos}
|