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
import strawberry
from fastapi import HTTPException
from starlette.status import HTTP_401_UNAUTHORIZED
from database import models
from decorators.endpoints import authenticated_user_only
from graphql_schema.entities.resolvers.base import BaseMutationResolver
@@ -17,6 +20,9 @@ class CopilotQueries:
@strawberry.field()
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(
id,
user_id=info.context.user_id,