Moznost nahravat fotky copilotem

src/graphql_schema/entities/copilot.py

@@ -1,5 +1,6 @@
 from typing import List, Optional
 import strawberry
+from graphql import GraphQLError
 from strawberry.types import Info
 from database import models
 from decorators.error_logging import error_logging
@@ -27,12 +28,24 @@ class CopilotQueries:
         id: Optional[int] = None,
         url_slug: Optional[str] = None,
         pilot_username: Optional[str] = None,
+        upload_token: Optional[str] = None,
+        upload_flight_slug: Optional[str] = None,
         public: Optional[bool] = False
     ) -> Copilot:
-        filter_params = get_detail_filters(id, url_slug)
+        filter_params = {}
+        if id:
+            filter_params['object_id'] = id
+        if url_slug is not None:
+            filter_params['url_slug'] = url_slug
+        if upload_token and upload_flight_slug:
+            filter_params['upload_token'] = upload_token
+            filter_params['upload_flight_slug'] = upload_flight_slug
         if pilot_username:
             filter_params['pilot_username'] = pilot_username
 
+        if not filter_params:
+            raise GraphQLError(f"Invalid identification supplied: {filter_params}")
+
         return await CopilotQueryResolver().get_one(
             user_id=info.context.user_id,
             only_public=public,