Dotazovani na detail kopilota jen po prihlaseni nebo jen s username

This commit is contained in:
Michal Kváček
2023-11-02 16:29:02 +01:00
parent f2e2c1a1ab
commit e8aa678555
+6
View File
@@ -1,5 +1,8 @@
from typing import List, Optional from typing import List, Optional
import strawberry import strawberry
from fastapi import HTTPException
from starlette.status import HTTP_401_UNAUTHORIZED
from database import models from database import models
from decorators.endpoints import authenticated_user_only from decorators.endpoints import authenticated_user_only
from graphql_schema.entities.resolvers.base import BaseMutationResolver from graphql_schema.entities.resolvers.base import BaseMutationResolver
@@ -17,6 +20,9 @@ class CopilotQueries:
@strawberry.field() @strawberry.field()
async def copilot(root, info, id: int, pilot_username: Optional[str] = None) -> Copilot: async def copilot(root, info, id: int, pilot_username: Optional[str] = None) -> Copilot:
if not info.context.user_id and not pilot_username:
raise HTTPException(HTTP_401_UNAUTHORIZED)
return await CopilotQueryResolver().get_one( return await CopilotQueryResolver().get_one(
id, id,
user_id=info.context.user_id, user_id=info.context.user_id,