Prepis strankovani na extension, sjednoceni volani resolveru

src/graphql_schema/dataloaders/base.py

@@ -1,5 +1,5 @@
 from collections import defaultdict
-from typing import Type, List, Optional
+from typing import Type
 from logger import log
 from database import models, async_session
 from database.query_builder import QueryBuilder
@@ -25,7 +25,7 @@ class BaseDataloader:
 
 
 class SingleModelByIdDataloader(BaseDataloader):
-    async def load(self, ids: List[int]):
+    async def load(self, ids: list[int]):
         async with async_session() as session:
             query = (
                 self.query_builder.get_simple_query(extra_select=[self.relationship_column], include_deleted=True)
@@ -80,7 +80,7 @@ class MultiModelsDataloader(BaseDataloader):
 
         return query
 
-    async def load(self, ids: List[int]):
+    async def load(self, ids: list[int]):
         query = self.get_query(ids)
 
         async with async_session() as db: