add a filter for rebuild calls

ariadne_codegen/client_generators/result_types.py

@@ -155,10 +155,15 @@ def _get_operation_type_name(self, definition: ExecutableDefinitionNode) -> str:
         raise NotSupported(f"Not supported operation type: {definition}")
 
     def generate(self) -> ast.Module:
-        model_rebuild_calls = [
-            generate_expr(generate_method_call(class_def.name, MODEL_REBUILD_METHOD))
-            for class_def in self._class_defs
-        ]
+        model_rebuild_calls = []
+        for class_def in self._class_defs:
+            for node in ast.walk(class_def):
+                if isinstance(node, ast.Name) and '"' in node.id:
+                    call_expr = generate_expr(
+                        generate_method_call(class_def.name, MODEL_REBUILD_METHOD)
+                    )
+                    model_rebuild_calls.append(call_expr)
+
         module_body = (
             cast(List[ast.stmt], self._imports)
             + cast(List[ast.stmt], self._class_defs)

tests/main/clients/custom_base_client/expected_client/get_query_a.py

@@ -12,4 +12,3 @@ class GetQueryAQueryA(BaseModel):
 
 
 GetQueryA.model_rebuild()
-GetQueryAQueryA.model_rebuild()

tests/main/clients/custom_config_file/expected_client/test.py

@@ -5,6 +5,3 @@
 
 class Test(BaseModel):
     test_query: str = Field(alias="testQuery")
-
-
-Test.model_rebuild()

tests/main/clients/custom_files_names/expected_client/get_query_a.py

@@ -12,4 +12,3 @@ class GetQueryAQueryA(BaseModel):
 
 
 GetQueryA.model_rebuild()
-GetQueryAQueryA.model_rebuild()

tests/main/clients/custom_scalars/expected_client/get_a.py

@@ -19,4 +19,3 @@ class GetATestQuery(BaseModel):
 
 
 GetA.model_rebuild()
-GetATestQuery.model_rebuild()

tests/main/clients/example/expected_client/create_user.py

@@ -14,4 +14,3 @@ class CreateUserUserCreate(BaseModel):
 
 
 CreateUser.model_rebuild()
-CreateUserUserCreate.model_rebuild()

tests/main/clients/example/expected_client/get_users_counter.py

@@ -5,6 +5,3 @@
 
 class GetUsersCounter(BaseModel):
     users_counter: int = Field(alias="usersCounter")
-
-
-GetUsersCounter.model_rebuild()

tests/main/clients/example/expected_client/list_all_users.py

@@ -23,4 +23,3 @@ class ListAllUsersUsersLocation(BaseModel):
 
 ListAllUsers.model_rebuild()
 ListAllUsersUsers.model_rebuild()
-ListAllUsersUsersLocation.model_rebuild()

tests/main/clients/example/expected_client/list_users_by_country.py

@@ -16,4 +16,3 @@ class ListUsersByCountryUsers(BasicUser, UserPersonalData):
 
 
 ListUsersByCountry.model_rebuild()
-ListUsersByCountryUsers.model_rebuild()

tests/main/clients/example/expected_client/upload_file.py

@@ -5,6 +5,3 @@
 
 class UploadFile(BaseModel):
     file_upload: bool = Field(alias="fileUpload")
-
-
-UploadFile.model_rebuild()

tests/main/clients/extended_models/expected_client/fragments_with_mixins.py

@@ -19,5 +19,4 @@ class FragmentsWithMixinsQueryB(FragmentB, CommonMixin):
 
 
 FragmentsWithMixins.model_rebuild()
-FragmentsWithMixinsQueryA.model_rebuild()
-FragmentsWithMixinsQueryB.model_rebuild()
+FragmentsWithMixins.model_rebuild()

tests/main/clients/extended_models/expected_client/get_query_a.py

@@ -14,4 +14,3 @@ class GetQueryAQueryA(BaseModel, MixinA, CommonMixin):
 
 
 GetQueryA.model_rebuild()
-GetQueryAQueryA.model_rebuild()

tests/main/clients/extended_models/expected_client/get_query_a_with_fragment.py

@@ -3,6 +3,3 @@
 
 class GetQueryAWithFragment(GetQueryAFragment):
     pass
-
-
-GetQueryAWithFragment.model_rebuild()

tests/main/clients/extended_models/expected_client/get_query_b.py

@@ -14,4 +14,3 @@ class GetQueryBQueryB(BaseModel, MixinB, CommonMixin):
 
 
 GetQueryB.model_rebuild()
-GetQueryBQueryB.model_rebuild()

tests/main/clients/fragments_on_abstract_types/expected_client/query_with_fragment_on_sub_interface.py

@@ -22,5 +22,7 @@ class QueryWithFragmentOnSubInterfaceQueryInterfaceInterfaceA(FragmentA):
 
 
 QueryWithFragmentOnSubInterface.model_rebuild()
+QueryWithFragmentOnSubInterface.model_rebuild()
+QueryWithFragmentOnSubInterfaceQueryInterfaceBaseInterface.model_rebuild()
 QueryWithFragmentOnSubInterfaceQueryInterfaceBaseInterface.model_rebuild()
 QueryWithFragmentOnSubInterfaceQueryInterfaceInterfaceA.model_rebuild()

tests/main/clients/fragments_on_abstract_types/expected_client/query_with_fragment_on_sub_interface_with_inline_fragment.py

@@ -34,6 +34,8 @@ class QueryWithFragmentOnSubInterfaceWithInlineFragmentQueryInterfaceTypeA(BaseM
     another: str
 
 
+QueryWithFragmentOnSubInterfaceWithInlineFragment.model_rebuild()
+QueryWithFragmentOnSubInterfaceWithInlineFragment.model_rebuild()
 QueryWithFragmentOnSubInterfaceWithInlineFragment.model_rebuild()
 QueryWithFragmentOnSubInterfaceWithInlineFragmentQueryInterfaceBaseInterface.model_rebuild()
 QueryWithFragmentOnSubInterfaceWithInlineFragmentQueryInterfaceInterfaceA.model_rebuild()

tests/main/clients/fragments_on_abstract_types/expected_client/query_with_fragment_on_union_member.py

@@ -21,6 +21,7 @@ class QueryWithFragmentOnUnionMemberQueryUnionTypeB(FragmentB):
     typename__: Literal["TypeB"] = Field(alias="__typename")
 
 
+QueryWithFragmentOnUnionMember.model_rebuild()
 QueryWithFragmentOnUnionMember.model_rebuild()
 QueryWithFragmentOnUnionMemberQueryUnionTypeA.model_rebuild()
 QueryWithFragmentOnUnionMemberQueryUnionTypeB.model_rebuild()

tests/main/clients/inline_fragments/expected_client/interface_a.py

@@ -29,6 +29,9 @@ class InterfaceAQueryITypeB(BaseModel):
 
 
 InterfaceA.model_rebuild()
+InterfaceA.model_rebuild()
+InterfaceA.model_rebuild()
+InterfaceAQueryIInterface.model_rebuild()
 InterfaceAQueryIInterface.model_rebuild()
 InterfaceAQueryITypeA.model_rebuild()
 InterfaceAQueryITypeB.model_rebuild()

tests/main/clients/inline_fragments/expected_client/interface_b.py

@@ -23,5 +23,8 @@ class InterfaceBQueryITypeA(BaseModel):
 
 
 InterfaceB.model_rebuild()
+InterfaceB.model_rebuild()
+InterfaceBQueryIInterface.model_rebuild()
+InterfaceBQueryIInterface.model_rebuild()
 InterfaceBQueryIInterface.model_rebuild()
 InterfaceBQueryITypeA.model_rebuild()

tests/main/clients/inline_fragments/expected_client/interface_c.py

@@ -18,3 +18,6 @@ class InterfaceCQueryI(BaseModel):
 
 InterfaceC.model_rebuild()
 InterfaceCQueryI.model_rebuild()
+InterfaceCQueryI.model_rebuild()
+InterfaceCQueryI.model_rebuild()
+InterfaceCQueryI.model_rebuild()

tests/main/clients/inline_fragments/expected_client/interface_with_typename.py

@@ -18,3 +18,6 @@ class InterfaceWithTypenameQueryI(BaseModel):
 
 InterfaceWithTypename.model_rebuild()
 InterfaceWithTypenameQueryI.model_rebuild()
+InterfaceWithTypenameQueryI.model_rebuild()
+InterfaceWithTypenameQueryI.model_rebuild()
+InterfaceWithTypenameQueryI.model_rebuild()

tests/main/clients/inline_fragments/expected_client/list_interface.py

@@ -36,6 +36,9 @@ class ListInterfaceQueryListITypeB(BaseModel):
 
 
 ListInterface.model_rebuild()
+ListInterface.model_rebuild()
+ListInterface.model_rebuild()
+ListInterfaceQueryListIInterface.model_rebuild()
 ListInterfaceQueryListIInterface.model_rebuild()
 ListInterfaceQueryListITypeA.model_rebuild()
 ListInterfaceQueryListITypeB.model_rebuild()

tests/main/clients/inline_fragments/expected_client/list_union.py

@@ -34,6 +34,8 @@ class ListUnionQueryListUTypeC(BaseModel):
     typename__: Literal["TypeC"] = Field(alias="__typename")
 
 
+ListUnion.model_rebuild()
+ListUnion.model_rebuild()
 ListUnion.model_rebuild()
 ListUnionQueryListUTypeA.model_rebuild()
 ListUnionQueryListUTypeB.model_rebuild()

tests/main/clients/inline_fragments/expected_client/query_with_fragment_on_interface.py

@@ -31,6 +31,9 @@ class QueryWithFragmentOnInterfaceQueryITypeB(BaseModel):
 
 
 QueryWithFragmentOnInterface.model_rebuild()
+QueryWithFragmentOnInterface.model_rebuild()
+QueryWithFragmentOnInterface.model_rebuild()
+QueryWithFragmentOnInterfaceQueryIInterface.model_rebuild()
 QueryWithFragmentOnInterfaceQueryIInterface.model_rebuild()
 QueryWithFragmentOnInterfaceQueryITypeA.model_rebuild()
 QueryWithFragmentOnInterfaceQueryITypeB.model_rebuild()

tests/main/clients/inline_fragments/expected_client/query_with_fragment_on_query_with_interface.py

@@ -3,6 +3,3 @@
 
 class QueryWithFragmentOnQueryWithInterface(FragmentOnQueryWithInterface):
     pass
-
-
-QueryWithFragmentOnQueryWithInterface.model_rebuild()

tests/main/clients/inline_fragments/expected_client/query_with_fragment_on_query_with_union.py

@@ -3,6 +3,3 @@
 
 class QueryWithFragmentOnQueryWithUnion(FragmentOnQueryWithUnion):
     pass
-
-
-QueryWithFragmentOnQueryWithUnion.model_rebuild()

tests/main/clients/inline_fragments/expected_client/query_with_fragment_on_union.py

@@ -29,6 +29,8 @@ class QueryWithFragmentOnUnionQueryUTypeC(BaseModel):
     typename__: Literal["TypeC"] = Field(alias="__typename")
 
 
+QueryWithFragmentOnUnion.model_rebuild()
+QueryWithFragmentOnUnion.model_rebuild()
 QueryWithFragmentOnUnion.model_rebuild()
 QueryWithFragmentOnUnionQueryUTypeA.model_rebuild()
 QueryWithFragmentOnUnionQueryUTypeB.model_rebuild()

tests/main/clients/inline_fragments/expected_client/union_a.py

@@ -27,6 +27,8 @@ class UnionAQueryUTypeC(BaseModel):
     typename__: Literal["TypeC"] = Field(alias="__typename")
 
 
+UnionA.model_rebuild()
+UnionA.model_rebuild()
 UnionA.model_rebuild()
 UnionAQueryUTypeA.model_rebuild()
 UnionAQueryUTypeB.model_rebuild()

tests/main/clients/inline_fragments/expected_client/union_b.py

@@ -25,6 +25,8 @@ class UnionBQueryUTypeC(BaseModel):
     typename__: Literal["TypeC"] = Field(alias="__typename")
 
 
+UnionB.model_rebuild()
+UnionB.model_rebuild()
 UnionB.model_rebuild()
 UnionBQueryUTypeA.model_rebuild()
 UnionBQueryUTypeB.model_rebuild()

tests/main/clients/multiple_fragments/expected_client/example_query_1.py

@@ -13,4 +13,3 @@ class ExampleQuery1ExampleQuery(MinimalA):
 
 
 ExampleQuery1.model_rebuild()
-ExampleQuery1ExampleQuery.model_rebuild()

tests/main/clients/multiple_fragments/expected_client/example_query_2.py

@@ -13,4 +13,3 @@ class ExampleQuery2ExampleQuery(FullA):
 
 
 ExampleQuery2.model_rebuild()
-ExampleQuery2ExampleQuery.model_rebuild()

tests/main/clients/multiple_fragments/expected_client/example_query_3.py

@@ -13,4 +13,3 @@ class ExampleQuery3ExampleQuery(CompleteA):
 
 
 ExampleQuery3.model_rebuild()
-ExampleQuery3ExampleQuery.model_rebuild()

tests/main/clients/only_used_inputs_and_enums/expected_client/get_a.py

@@ -3,6 +3,3 @@
 
 class GetA(BaseModel):
     a: str
-
-
-GetA.model_rebuild()

tests/main/clients/only_used_inputs_and_enums/expected_client/get_a_2.py

@@ -3,6 +3,3 @@
 
 class GetA2(BaseModel):
     a: str
-
-
-GetA2.model_rebuild()

tests/main/clients/only_used_inputs_and_enums/expected_client/get_b.py

@@ -3,6 +3,3 @@
 
 class GetB(BaseModel):
     b: str
-
-
-GetB.model_rebuild()

tests/main/clients/only_used_inputs_and_enums/expected_client/get_d.py

@@ -3,6 +3,3 @@
 
 class GetD(BaseModel):
     d: str
-
-
-GetD.model_rebuild()

tests/main/clients/only_used_inputs_and_enums/expected_client/get_e.py

@@ -3,6 +3,3 @@
 
 class GetE(BaseModel):
     e: str
-
-
-GetE.model_rebuild()

tests/main/clients/only_used_inputs_and_enums/expected_client/get_f.py

@@ -11,4 +11,3 @@ class GetFF(BaseModel):
 
 
 GetF.model_rebuild()
-GetFF.model_rebuild()

tests/main/clients/only_used_inputs_and_enums/expected_client/get_g.py

@@ -11,4 +11,3 @@ class GetGG(FragmentG):
 
 
 GetG.model_rebuild()
-GetGG.model_rebuild()

tests/main/clients/operations/expected_client/c_mutation.py

@@ -5,6 +5,3 @@
 
 class CMutation(BaseModel):
     const_mutation: int = Field(alias="constMutation")
-
-
-CMutation.model_rebuild()

tests/main/clients/operations/expected_client/c_query.py

@@ -5,6 +5,3 @@
 
 class CQuery(BaseModel):
     const_query: str = Field(alias="constQuery")
-
-
-CQuery.model_rebuild()

tests/main/clients/operations/expected_client/c_subscription.py

@@ -5,6 +5,3 @@
 
 class CSubscription(BaseModel):
     const_subscription: float = Field(alias="constSubscription")
-
-
-CSubscription.model_rebuild()

tests/main/clients/operations/expected_client/get_a.py

@@ -18,4 +18,3 @@ class GetAAValueB(BaseModel):
 
 GetA.model_rebuild()
 GetAA.model_rebuild()
-GetAAValueB.model_rebuild()

tests/main/clients/operations/expected_client/get_a_with_fragment.py

@@ -19,4 +19,3 @@ class GetAWithFragmentAValueB(FragmentB):
 
 GetAWithFragment.model_rebuild()
 GetAWithFragmentA.model_rebuild()
-GetAWithFragmentAValueB.model_rebuild()

tests/main/clients/operations/expected_client/get_s.py

@@ -10,4 +10,3 @@ class GetSS(BaseModel):
 
 
 GetS.model_rebuild()
-GetSS.model_rebuild()

tests/main/clients/operations/expected_client/get_xyz.py

@@ -25,6 +25,8 @@ class GetXYZXyzTypeZ(BaseModel):
     typename__: Literal["TypeZ"] = Field(alias="__typename")
 
 
+GetXYZ.model_rebuild()
+GetXYZ.model_rebuild()
 GetXYZ.model_rebuild()
 GetXYZXyzTypeX.model_rebuild()
 GetXYZXyzTypeY.model_rebuild()

tests/main/clients/remote_schema/expected_client/test.py

@@ -7,6 +7,3 @@
 
 class Test(BaseModel):
     test_query: Optional[int] = Field(alias="testQuery")
-
-
-Test.model_rebuild()

tests/main/clients/shorter_results/expected_client/get_animal_by_name.py

@@ -30,6 +30,8 @@ class GetAnimalByNameAnimalByNameDog(BaseModel):
     puppies: int
 
 
+GetAnimalByName.model_rebuild()
+GetAnimalByName.model_rebuild()
 GetAnimalByName.model_rebuild()
 GetAnimalByNameAnimalByNameAnimal.model_rebuild()
 GetAnimalByNameAnimalByNameCat.model_rebuild()

tests/main/clients/shorter_results/expected_client/get_animal_fragment_with_extra.py

@@ -7,6 +7,3 @@
 
 class GetAnimalFragmentWithExtra(ListAnimalsFragment):
     list_string: List[str] = Field(alias="listString")
-
-
-GetAnimalFragmentWithExtra.model_rebuild()

tests/main/clients/shorter_results/expected_client/get_authenticated_user.py

@@ -11,4 +11,3 @@ class GetAuthenticatedUserMe(BaseModel):
 
 
 GetAuthenticatedUser.model_rebuild()
-GetAuthenticatedUserMe.model_rebuild()

tests/main/clients/shorter_results/expected_client/get_complex_scalar.py

@@ -10,6 +10,3 @@ class GetComplexScalar(BaseModel):
     just_complex_scalar: Annotated[
         ComplexScalar, BeforeValidator(parse_complex_scalar)
     ] = Field(alias="justComplexScalar")
-
-
-GetComplexScalar.model_rebuild()

tests/main/clients/shorter_results/expected_client/get_simple_scalar.py

@@ -6,6 +6,3 @@
 
 class GetSimpleScalar(BaseModel):
     just_simple_scalar: SimpleScalar = Field(alias="justSimpleScalar")
-
-
-GetSimpleScalar.model_rebuild()

tests/main/clients/shorter_results/expected_client/list_animals.py

@@ -35,6 +35,8 @@ class ListAnimalsListAnimalsDog(BaseModel):
     puppies: int
 
 
+ListAnimals.model_rebuild()
+ListAnimals.model_rebuild()
 ListAnimals.model_rebuild()
 ListAnimalsListAnimalsAnimal.model_rebuild()
 ListAnimalsListAnimalsCat.model_rebuild()

tests/main/clients/shorter_results/expected_client/list_strings_1.py

@@ -9,6 +9,3 @@ class ListStrings1(BaseModel):
     optional_list_optional_string: Optional[List[Optional[str]]] = Field(
         alias="optionalListOptionalString"
     )
-
-
-ListStrings1.model_rebuild()

tests/main/clients/shorter_results/expected_client/list_strings_2.py

@@ -7,6 +7,3 @@
 
 class ListStrings2(BaseModel):
     optional_list_string: Optional[List[str]] = Field(alias="optionalListString")
-
-
-ListStrings2.model_rebuild()

tests/main/clients/shorter_results/expected_client/list_strings_3.py

@@ -7,6 +7,3 @@
 
 class ListStrings3(BaseModel):
     list_optional_string: List[Optional[str]] = Field(alias="listOptionalString")
-
-
-ListStrings3.model_rebuild()