From e8aa678555513909dd6ee11acd40c7aa7c3da431 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Michal=20Kv=C3=A1=C4=8Dek?= Date: Thu, 2 Nov 2023 16:29:02 +0100 Subject: [PATCH] Dotazovani na detail kopilota jen po prihlaseni nebo jen s username --- src/graphql_schema/entities/copilot.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/src/graphql_schema/entities/copilot.py b/src/graphql_schema/entities/copilot.py index 64d1446..c8b4fa9 100644 --- a/src/graphql_schema/entities/copilot.py +++ b/src/graphql_schema/entities/copilot.py @@ -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,