stigg-api-client-v2 0.693.0__py3-none-any.whl → 5.9.0__py3-none-any.whl

stigg/generated/base_client.py

@@ -1,17 +1,17 @@
-# Generated by ariadne-codegen on 2023-12-07 19:22
+# Generated by ariadne-codegen
 
 import json
-from typing import Any, Dict, Optional, TypeVar, cast
+from typing import IO, Any, Dict, List, Optional, Tuple, TypeVar, cast
 
 import httpx
 from pydantic import BaseModel
-from pydantic.json import pydantic_encoder
+from pydantic_core import to_jsonable_python
 
-from .base_model import UNSET
+from .base_model import UNSET, Upload
 from .exceptions import (
     GraphQLClientGraphQLMultiError,
     GraphQLClientHttpError,
-    GraphQlClientInvalidResponseError,
+    GraphQLClientInvalidResponseError,
 )
 
 Self = TypeVar("Self", bound="BaseClient")
@@ -41,15 +41,32 @@ class BaseClient:
         self.http_client.close()
 
     def execute(
-        self, query: str, variables: Optional[Dict[str, Any]] = None
+        self,
+        query: str,
+        operation_name: Optional[str] = None,
+        variables: Optional[Dict[str, Any]] = None,
+        **kwargs: Any,
     ) -> httpx.Response:
-        payload: Dict[str, Any] = {"query": query}
-        if variables:
-            payload["variables"] = self._convert_dict_to_json_serializable(variables)
-        content = json.dumps(payload, default=pydantic_encoder)
-        return self.http_client.post(url=self.url, content=content)
+        processed_variables, files, files_map = self._process_variables(variables)
+
+        if files and files_map:
+            return self._execute_multipart(
+                query=query,
+                operation_name=operation_name,
+                variables=processed_variables,
+                files=files,
+                files_map=files_map,
+                **kwargs,
+            )
 
-    def get_data(self, response: httpx.Response) -> dict[str, Any]:
+        return self._execute_json(
+            query=query,
+            operation_name=operation_name,
+            variables=processed_variables,
+            **kwargs,
+        )
+
+    def get_data(self, response: httpx.Response) -> Dict[str, Any]:
         if not response.is_success:
             raise GraphQLClientHttpError(
                 status_code=response.status_code, response=response
@@ -58,12 +75,14 @@ class BaseClient:
         try:
             response_json = response.json()
         except ValueError as exc:
-            raise GraphQlClientInvalidResponseError(response=response) from exc
+            raise GraphQLClientInvalidResponseError(response=response) from exc
 
-        if (not isinstance(response_json, dict)) or ("data" not in response_json):
-            raise GraphQlClientInvalidResponseError(response=response)
+        if (not isinstance(response_json, dict)) or (
+            "data" not in response_json and "errors" not in response_json
+        ):
+            raise GraphQLClientInvalidResponseError(response=response)
 
-        data = response_json["data"]
+        data = response_json.get("data")
         errors = response_json.get("errors")
 
         if errors:
@@ -71,14 +90,18 @@ class BaseClient:
                 errors_dicts=errors, data=data
             )
 
-        return cast(dict[str, Any], data)
+        return cast(Dict[str, Any], data)
 
-    def _convert_value(self, value: Any) -> Any:
-        if isinstance(value, BaseModel):
-            return value.dict(by_alias=True, exclude_unset=True)
-        if isinstance(value, list):
-            return [self._convert_value(item) for item in value]
-        return value
+    def _process_variables(
+        self, variables: Optional[Dict[str, Any]]
+    ) -> Tuple[
+        Dict[str, Any], Dict[str, Tuple[str, IO[bytes], str]], Dict[str, List[str]]
+    ]:
+        if not variables:
+            return {}, {}, {}
+
+        serializable_variables = self._convert_dict_to_json_serializable(variables)
+        return self._get_files_from_variables(serializable_variables)
 
     def _convert_dict_to_json_serializable(
         self, dict_: Dict[str, Any]
@@ -88,3 +111,101 @@ class BaseClient:
             for key, value in dict_.items()
             if value is not UNSET
         }
+
+    def _convert_value(self, value: Any) -> Any:
+        if isinstance(value, BaseModel):
+            return value.model_dump(by_alias=True, exclude_unset=True)
+        if isinstance(value, list):
+            return [self._convert_value(item) for item in value]
+        return value
+
+    def _get_files_from_variables(
+        self, variables: Dict[str, Any]
+    ) -> Tuple[
+        Dict[str, Any], Dict[str, Tuple[str, IO[bytes], str]], Dict[str, List[str]]
+    ]:
+        files_map: Dict[str, List[str]] = {}
+        files_list: List[Upload] = []
+
+        def separate_files(path: str, obj: Any) -> Any:
+            if isinstance(obj, list):
+                nulled_list = []
+                for index, value in enumerate(obj):
+                    value = separate_files(f"{path}.{index}", value)
+                    nulled_list.append(value)
+                return nulled_list
+
+            if isinstance(obj, dict):
+                nulled_dict = {}
+                for key, value in obj.items():
+                    value = separate_files(f"{path}.{key}", value)
+                    nulled_dict[key] = value
+                return nulled_dict
+
+            if isinstance(obj, Upload):
+                if obj in files_list:
+                    file_index = files_list.index(obj)
+                    files_map[str(file_index)].append(path)
+                else:
+                    file_index = len(files_list)
+                    files_list.append(obj)
+                    files_map[str(file_index)] = [path]
+                return None
+
+            return obj
+
+        nulled_variables = separate_files("variables", variables)
+        files: Dict[str, Tuple[str, IO[bytes], str]] = {
+            str(i): (file_.filename, cast(IO[bytes], file_.content), file_.content_type)
+            for i, file_ in enumerate(files_list)
+        }
+        return nulled_variables, files, files_map
+
+    def _execute_multipart(
+        self,
+        query: str,
+        operation_name: Optional[str],
+        variables: Dict[str, Any],
+        files: Dict[str, Tuple[str, IO[bytes], str]],
+        files_map: Dict[str, List[str]],
+        **kwargs: Any,
+    ) -> httpx.Response:
+        data = {
+            "operations": json.dumps(
+                {
+                    "query": query,
+                    "operationName": operation_name,
+                    "variables": variables,
+                },
+                default=to_jsonable_python,
+            ),
+            "map": json.dumps(files_map, default=to_jsonable_python),
+        }
+
+        return self.http_client.post(url=self.url, data=data, files=files, **kwargs)
+
+    def _execute_json(
+        self,
+        query: str,
+        operation_name: Optional[str],
+        variables: Dict[str, Any],
+        **kwargs: Any,
+    ) -> httpx.Response:
+        headers: Dict[str, str] = {"Content-Type": "application/json"}
+        headers.update(kwargs.get("headers", {}))
+
+        merged_kwargs: Dict[str, Any] = kwargs.copy()
+        merged_kwargs["headers"] = headers
+
+        return self.http_client.post(
+            url=self.url,
+            content=json.dumps(
+                {
+                    "query": query,
+                    "operationName": operation_name,
+                    "variables": variables,
+                },
+                default=to_jsonable_python,
+            ),
+            **merged_kwargs,
+        )

stigg/generated/base_model.py

@@ -1,12 +1,8 @@
-# Generated by ariadne-codegen on 2023-12-07 19:22
+# Generated by ariadne-codegen
 
-from typing import Any, Dict, Type, Union, get_args, get_origin
+from io import IOBase
 
-from pydantic import BaseModel as PydanticBaseModel
-from pydantic.class_validators import validator
-from pydantic.fields import ModelField
-
-from .scalars import SCALARS_PARSE_FUNCTIONS, SCALARS_SERIALIZE_FUNCTIONS
+from pydantic import BaseModel as PydanticBaseModel, ConfigDict
 
 
 class UnsetType:
@@ -18,43 +14,16 @@ UNSET = UnsetType()
 
 
 class BaseModel(PydanticBaseModel):
-    class Config:
-        allow_population_by_field_name = True
-        validate_assignment = True
-        arbitrary_types_allowed = True
-
-    # pylint: disable=no-self-argument
-    @validator("*", pre=True)
-    def parse_custom_scalars(cls, value: Any, field: ModelField) -> Any:
-        return cls._parse_custom_scalar_value(value, field.annotation)
-
-    @classmethod
-    def _parse_custom_scalar_value(cls, value: Any, type_: Type[Any]) -> Any:
-        origin = get_origin(type_)
-        args = get_args(type_)
-        if origin is list and isinstance(value, list):
-            return [cls._parse_custom_scalar_value(item, args[0]) for item in value]
-
-        if origin is Union and type(None) in args:
-            sub_type: Any = list(filter(None, args))[0]
-            return cls._parse_custom_scalar_value(value, sub_type)
-
-        decode = SCALARS_PARSE_FUNCTIONS.get(type_)
-        if value and decode and callable(decode):
-            return decode(value)
-
-        return value
-
-    def dict(self, **kwargs: Any) -> Dict[str, Any]:
-        dict_ = super().dict(**kwargs)
-        return {key: self._serialize_value(value) for key, value in dict_.items()}
-
-    def _serialize_value(self, value: Any) -> Any:
-        serialize = SCALARS_SERIALIZE_FUNCTIONS.get(type(value))
-        if serialize and callable(serialize):
-            return serialize(value)
-
-        if isinstance(value, list):
-            return [self._serialize_value(item) for item in value]
-
-        return value
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        arbitrary_types_allowed=True,
+        protected_namespaces=(),
+    )
+
+
+class Upload:
+    def __init__(self, filename: str, content: IOBase, content_type: str):
+        self.filename = filename
+        self.content = content
+        self.content_type = content_type

stigg/generated/cancel_subscription.py

@@ -1,4 +1,4 @@
-# Generated by ariadne-codegen on 2023-12-07 19:22
+# Generated by ariadne-codegen
 # Source: operations.graphql
 
 from pydantic import Field
@@ -17,5 +17,4 @@ class CancelSubscriptionCancelSubscription(SlimSubscriptionFragment):
     pass
 
 
-CancelSubscription.update_forward_refs()
-CancelSubscriptionCancelSubscription.update_forward_refs()
+CancelSubscription.model_rebuild()

stigg/generated/cancel_subscription_updates.py

@@ -1,4 +1,4 @@
-# Generated by ariadne-codegen on 2023-12-07 19:22
+# Generated by ariadne-codegen
 # Source: operations.graphql
 
 from pydantic import Field
@@ -8,6 +8,3 @@ from .base_model import BaseModel
 
 class CancelSubscriptionUpdates(BaseModel):
     cancel_schedule: str = Field(alias="cancelSchedule")
-
-
-CancelSubscriptionUpdates.update_forward_refs()