diff --git a/src/graphql_schema/entities/airport.py b/src/graphql_schema/entities/airport.py index 5822adb..7ae94a1 100644 --- a/src/graphql_schema/entities/airport.py +++ b/src/graphql_schema/entities/airport.py @@ -2,8 +2,7 @@ from typing import List import strawberry from database import models from decorators.endpoints import authenticated_user_only -from graphql_schema.entities.resolvers.airport import get_airport_resolver -from graphql_schema.entities.resolvers.base import get_list, get_one +from graphql_schema.entities.resolvers.base import get_list, get_one, get_base_resolver from graphql_schema.sqlalchemy_to_strawberry_type import strawberry_sqlalchemy_type @@ -17,11 +16,11 @@ class AirportQueries: @strawberry.field() @authenticated_user_only() async def airports(root, info) -> List[Airport]: - query = get_airport_resolver(info.context.user_id) + query = get_base_resolver(models.Airport, user_id=info.context.user_id) return await get_list(models.Airport, query) @strawberry.field() @authenticated_user_only() async def airport(root, info, id: int) -> Airport: - query = get_airport_resolver(info.context.user_id, id) + query = get_base_resolver(models.Airport, user_id=info.context.user_id, object_id=id) return await get_one(models.Airport, query) diff --git a/src/graphql_schema/entities/resolvers/airport.py b/src/graphql_schema/entities/resolvers/airport.py deleted file mode 100644 index 4415ea6..0000000 --- a/src/graphql_schema/entities/resolvers/airport.py +++ /dev/null @@ -1,17 +0,0 @@ -from typing import Optional -from sqlalchemy import or_ -from database import models -from graphql_schema.entities.resolvers.base import get_base_resolver - - -def get_airport_resolver(user_id: int, airport_id: Optional[int] = None): - if airport_id: - return get_base_resolver(model=models.Airport, object_id=airport_id, user_id=user_id) - - return ( - get_base_resolver(model=models.Airport) - .filter(or_( - models.Airport.created_by_id == user_id, - models.Airport.created_by_id.is_(None), - )) - ) \ No newline at end of file