| 
1 | 1 | from fastapi import APIRouter, Depends  | 
2 | 2 | from sqlmodel.ext.asyncio.session import AsyncSession  | 
3 | 3 | 
 
  | 
4 |  | -from app import crud  | 
5 |  | -from app.db.session import get_db  | 
6 |  | -from app.schemas.overlay import OverlaySchema, OverlayCreate  | 
 | 4 | +from app.crud import overlay as crud  | 
 | 5 | +from app.db.session import get_session  | 
 | 6 | +from app.schemas.overlay import OverlaySchema  | 
7 | 7 | 
 
  | 
8 | 8 | router = APIRouter()  | 
9 | 9 | 
 
  | 
10 | 10 | 
 
  | 
11 | 11 | @router.get("/", response_model=list[OverlaySchema])  | 
12 |  | -async def get_overlays(session: AsyncSession = Depends(get_db)):  | 
 | 12 | +async def get_overlays(session: AsyncSession = Depends(get_session)):  | 
13 | 13 |     return await crud.get_overlays(session)  | 
14 | 14 | 
 
  | 
15 | 15 | 
 
  | 
16 | 16 | @router.get("/{overlay_id}", response_model=OverlaySchema)  | 
17 |  | -async def get_overlay(overlay_id: str, session: AsyncSession = Depends(get_db)):  | 
 | 17 | +async def get_overlay(overlay_id: str, session: AsyncSession = Depends(get_session)):  | 
18 | 18 |     return await crud.get_overlay(overlay_id, session)  | 
19 | 19 | 
 
  | 
20 | 20 | 
 
  | 
21 | 21 | @router.post("/", response_model=OverlaySchema)  | 
22 |  | -async def add_overlay(overlay: OverlayCreate, session: AsyncSession = Depends(get_db)):  | 
23 |  | -    return await crud.add_overlay.add_overlay(overlay, session)  | 
 | 22 | +async def create_overlay(overlay: OverlaySchema, session: AsyncSession = Depends(get_session)):  | 
 | 23 | +    return await crud.create_overlay(overlay, session)  | 
0 commit comments