Ukladani letu, query na seznam letist

This commit is contained in:
Michal Kváček
2023-06-06 07:34:45 +02:00
parent 014a999055
commit 69d30c8654
12 changed files with 301 additions and 64 deletions
+16
View File
@@ -0,0 +1,16 @@
from typing import List
from sqlalchemy import select
from strawberry.dataloader import DataLoader
from database import async_session
from database.models import Airport
async def load(ids: List[int]):
async with async_session() as session:
models = (await session.scalars(select(Airport).filter(Airport.id.in_(ids)))).all()
models_by_id = {model.id: model for model in models}
return [models_by_id.get(id_) for id_ in ids]
airport_dataloader = DataLoader(load_fn=load)