From 7d9d6a7cf05034c1021a602299c4713b2ce7bb36 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Rafa=C5=82=20Pito=C5=84?= Date: Mon, 29 Jan 2024 12:41:09 +0100 Subject: [PATCH] Format code with lastest black --- ariadne_codegen/client_generators/client.py | 18 +++++++++--------- .../client_generators/result_fields.py | 6 +++--- .../expected_client/union_a.py | 6 +++--- .../expected_client/union_b.py | 6 +++--- .../shorter_results/expected_client/client.py | 8 ++------ 5 files changed, 20 insertions(+), 24 deletions(-) diff --git a/ariadne_codegen/client_generators/client.py b/ariadne_codegen/client_generators/client.py index 77f193ca..6319da52 100644 --- a/ariadne_codegen/client_generators/client.py +++ b/ariadne_codegen/client_generators/client.py @@ -141,15 +141,15 @@ def add_method( raise NotSupported( "Subscriptions are only available when using async client." ) - method_def: Union[ - ast.FunctionDef, ast.AsyncFunctionDef - ] = self._generate_subscription_method_def( - name=name, - operation_name=operation_name, - return_type=return_type, - arguments=arguments, - arguments_dict=arguments_dict, - operation_str=operation_str, + method_def: Union[ast.FunctionDef, ast.AsyncFunctionDef] = ( + self._generate_subscription_method_def( + name=name, + operation_name=operation_name, + return_type=return_type, + arguments=arguments, + arguments_dict=arguments_dict, + operation_str=operation_str, + ) ) elif async_: method_def = self._generate_async_method( diff --git a/ariadne_codegen/client_generators/result_fields.py b/ariadne_codegen/client_generators/result_fields.py index 8b188a88..e0c377e3 100644 --- a/ariadne_codegen/client_generators/result_fields.py +++ b/ariadne_codegen/client_generators/result_fields.py @@ -89,9 +89,9 @@ def parse_operation_field( schema=schema, field_node=field, custom_scalars=custom_scalars if custom_scalars else {}, - fragments_definitions=fragments_definitions - if fragments_definitions - else {}, + fragments_definitions=( + fragments_definitions if fragments_definitions else {} + ), ) ) diff --git a/tests/main/clients/inline_fragments/expected_client/union_a.py b/tests/main/clients/inline_fragments/expected_client/union_a.py index 6f24e4a7..ae379c11 100644 --- a/tests/main/clients/inline_fragments/expected_client/union_a.py +++ b/tests/main/clients/inline_fragments/expected_client/union_a.py @@ -6,9 +6,9 @@ class UnionA(BaseModel): - query_u: Union[ - "UnionAQueryUTypeA", "UnionAQueryUTypeB", "UnionAQueryUTypeC" - ] = Field(alias="queryU", discriminator="typename__") + query_u: Union["UnionAQueryUTypeA", "UnionAQueryUTypeB", "UnionAQueryUTypeC"] = ( + Field(alias="queryU", discriminator="typename__") + ) class UnionAQueryUTypeA(BaseModel): diff --git a/tests/main/clients/inline_fragments/expected_client/union_b.py b/tests/main/clients/inline_fragments/expected_client/union_b.py index cde21a53..5200a741 100644 --- a/tests/main/clients/inline_fragments/expected_client/union_b.py +++ b/tests/main/clients/inline_fragments/expected_client/union_b.py @@ -6,9 +6,9 @@ class UnionB(BaseModel): - query_u: Union[ - "UnionBQueryUTypeA", "UnionBQueryUTypeB", "UnionBQueryUTypeC" - ] = Field(alias="queryU", discriminator="typename__") + query_u: Union["UnionBQueryUTypeA", "UnionBQueryUTypeB", "UnionBQueryUTypeC"] = ( + Field(alias="queryU", discriminator="typename__") + ) class UnionBQueryUTypeA(BaseModel): diff --git a/tests/main/clients/shorter_results/expected_client/client.py b/tests/main/clients/shorter_results/expected_client/client.py index f14484db..9cc2b0f5 100644 --- a/tests/main/clients/shorter_results/expected_client/client.py +++ b/tests/main/clients/shorter_results/expected_client/client.py @@ -133,9 +133,7 @@ async def list_type_a( data = self.get_data(response) return ListTypeA.model_validate(data).list_optional_type_a - async def get_animal_by_name( - self, name: str, **kwargs: Any - ) -> Union[ + async def get_animal_by_name(self, name: str, **kwargs: Any) -> Union[ GetAnimalByNameAnimalByNameAnimal, GetAnimalByNameAnimalByNameCat, GetAnimalByNameAnimalByNameDog, @@ -163,9 +161,7 @@ async def get_animal_by_name( data = self.get_data(response) return GetAnimalByName.model_validate(data).animal_by_name - async def list_animals( - self, **kwargs: Any - ) -> List[ + async def list_animals(self, **kwargs: Any) -> List[ Union[ ListAnimalsListAnimalsAnimal, ListAnimalsListAnimalsCat,