36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
from typing import Annotated
|
|
|
|
import asyncpg
|
|
from dishka.integrations.fastapi import DishkaRoute, FromDishka
|
|
from fastapi import APIRouter, Query
|
|
|
|
from utils.policy import repository
|
|
from utils.policy.models import FolderSpec
|
|
|
|
router = APIRouter(prefix="/api/folders", tags=["folders"], route_class=DishkaRoute)
|
|
|
|
|
|
def _serialize(spec: FolderSpec) -> dict:
|
|
return {
|
|
"folder_id": spec.folder_id,
|
|
"order_index": spec.order_index,
|
|
"title": spec.title,
|
|
"include_ids": sorted(spec.include_ids),
|
|
"exclude_ids": sorted(spec.exclude_ids),
|
|
"pinned_ids": sorted(spec.pinned_ids),
|
|
"contacts": spec.contacts,
|
|
"non_contacts": spec.non_contacts,
|
|
"groups": spec.groups,
|
|
"broadcasts": spec.broadcasts,
|
|
"bots": spec.bots,
|
|
"is_chatlist": spec.is_chatlist,
|
|
}
|
|
|
|
|
|
@router.get("")
|
|
async def list_folders(
|
|
pool: FromDishka[asyncpg.Pool], account_id: Annotated[int, Query()]
|
|
) -> list[dict]:
|
|
folders = await repository.list_folders(pool, account_id)
|
|
return [_serialize(spec) for spec in folders]
|