add model_rebuild_calls for result_types.py

ariadne_codegen/client_generators/result_types.py

@@ -38,6 +38,8 @@
     generate_module,
     generate_pass,
     generate_pydantic_field,
+    generate_method_call,
+    generate_expr,
 )
 from ..exceptions import NotSupported, ParsingError
 from ..plugins.manager import PluginManager
@@ -61,6 +63,7 @@
     TYPENAME_FIELD_NAME,
     TYPING_MODULE,
     UNION,
+    MODEL_REBUILD_METHOD,
 )
 from .result_fields import FieldContext, is_union, parse_operation_field
 from .scalars import ScalarData, generate_scalar_imports
@@ -152,8 +155,14 @@ def _get_operation_type_name(self, definition: ExecutableDefinitionNode) -> str:
         raise NotSupported(f"Not supported operation type: {definition}")
 
     def generate(self) -> ast.Module:
-        module_body = cast(List[ast.stmt], self._imports) + cast(
-            List[ast.stmt], self._class_defs
+        model_rebuild_calls = [
+            generate_expr(generate_method_call(class_def.name, MODEL_REBUILD_METHOD))
+            for class_def in self._class_defs
+        ]
+        module_body = (
+            cast(List[ast.stmt], self._imports)
+            + cast(List[ast.stmt], self._class_defs)
+            + cast(List[ast.stmt], model_rebuild_calls)
         )
 
         module = generate_module(module_body)

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

@@ -9,3 +9,7 @@ class GetQueryA(BaseModel):
 
 class GetQueryAQueryA(BaseModel):
     field_a: int = Field(alias="fieldA")
+
+
+GetQueryA.model_rebuild()
+GetQueryAQueryA.model_rebuild()

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

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

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

@@ -9,3 +9,7 @@ class GetQueryA(BaseModel):
 
 class GetQueryAQueryA(BaseModel):
     field_a: int = Field(alias="fieldA")
+
+
+GetQueryA.model_rebuild()
+GetQueryAQueryA.model_rebuild()

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

@@ -16,3 +16,7 @@ class GetATestQuery(BaseModel):
     code: Annotated[Code, BeforeValidator(parse_code)]
     id: int
     other: Any
+
+
+GetA.model_rebuild()
+GetATestQuery.model_rebuild()

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

@@ -11,3 +11,7 @@ class CreateUser(BaseModel):
 
 class CreateUserUserCreate(BaseModel):
     id: str
+
+
+CreateUser.model_rebuild()
+CreateUserUserCreate.model_rebuild()

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

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

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

@@ -19,3 +19,8 @@ class ListAllUsersUsers(BaseModel):
 
 class ListAllUsersUsersLocation(BaseModel):
     country: Optional[str]
+
+
+ListAllUsers.model_rebuild()
+ListAllUsersUsers.model_rebuild()
+ListAllUsersUsersLocation.model_rebuild()

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

@@ -13,3 +13,7 @@ class ListUsersByCountry(BaseModel):
 
 class ListUsersByCountryUsers(BasicUser, UserPersonalData):
     favourite_color: Optional[Color] = Field(alias="favouriteColor")
+
+
+ListUsersByCountry.model_rebuild()
+ListUsersByCountryUsers.model_rebuild()

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

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

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

@@ -16,3 +16,8 @@ class FragmentsWithMixinsQueryA(FragmentA, CommonMixin):
 
 class FragmentsWithMixinsQueryB(FragmentB, CommonMixin):
     pass
+
+
+FragmentsWithMixins.model_rebuild()
+FragmentsWithMixinsQueryA.model_rebuild()
+FragmentsWithMixinsQueryB.model_rebuild()

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

@@ -11,3 +11,7 @@ class GetQueryA(BaseModel):
 
 class GetQueryAQueryA(BaseModel, MixinA, CommonMixin):
     field_a: int = Field(alias="fieldA")
+
+
+GetQueryA.model_rebuild()
+GetQueryAQueryA.model_rebuild()

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

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

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

@@ -11,3 +11,7 @@ class GetQueryB(BaseModel):
 
 class GetQueryBQueryB(BaseModel, MixinB, CommonMixin):
     field_b: str = Field(alias="fieldB")
+
+
+GetQueryB.model_rebuild()
+GetQueryBQueryB.model_rebuild()

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

@@ -19,3 +19,8 @@ class QueryWithFragmentOnSubInterfaceQueryInterfaceBaseInterface(BaseModel):
 
 class QueryWithFragmentOnSubInterfaceQueryInterfaceInterfaceA(FragmentA):
     typename__: Literal["InterfaceA"] = Field(alias="__typename")
+
+
+QueryWithFragmentOnSubInterface.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

@@ -32,3 +32,9 @@ class QueryWithFragmentOnSubInterfaceWithInlineFragmentQueryInterfaceTypeA(BaseM
     id: str
     value_a: str = Field(alias="valueA")
     another: str
+
+
+QueryWithFragmentOnSubInterfaceWithInlineFragment.model_rebuild()
+QueryWithFragmentOnSubInterfaceWithInlineFragmentQueryInterfaceBaseInterface.model_rebuild()
+QueryWithFragmentOnSubInterfaceWithInlineFragmentQueryInterfaceInterfaceA.model_rebuild()
+QueryWithFragmentOnSubInterfaceWithInlineFragmentQueryInterfaceTypeA.model_rebuild()

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

@@ -19,3 +19,8 @@ class QueryWithFragmentOnUnionMemberQueryUnionTypeA(BaseModel):
 
 class QueryWithFragmentOnUnionMemberQueryUnionTypeB(FragmentB):
     typename__: Literal["TypeB"] = Field(alias="__typename")
+
+
+QueryWithFragmentOnUnionMember.model_rebuild()
+QueryWithFragmentOnUnionMemberQueryUnionTypeA.model_rebuild()
+QueryWithFragmentOnUnionMemberQueryUnionTypeB.model_rebuild()

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

@@ -26,3 +26,9 @@ class InterfaceAQueryITypeB(BaseModel):
     typename__: Literal["TypeB"] = Field(alias="__typename")
     id: str
     field_b: str = Field(alias="fieldB")
+
+
+InterfaceA.model_rebuild()
+InterfaceAQueryIInterface.model_rebuild()
+InterfaceAQueryITypeA.model_rebuild()
+InterfaceAQueryITypeB.model_rebuild()

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

@@ -20,3 +20,8 @@ class InterfaceBQueryITypeA(BaseModel):
     typename__: Literal["TypeA"] = Field(alias="__typename")
     id: str
     field_a: str = Field(alias="fieldA")
+
+
+InterfaceB.model_rebuild()
+InterfaceBQueryIInterface.model_rebuild()
+InterfaceBQueryITypeA.model_rebuild()

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

@@ -14,3 +14,7 @@ class InterfaceCQueryI(BaseModel):
         alias="__typename"
     )
     id: str
+
+
+InterfaceC.model_rebuild()
+InterfaceCQueryI.model_rebuild()

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

@@ -14,3 +14,7 @@ class InterfaceWithTypenameQueryI(BaseModel):
         alias="__typename"
     )
     id: str
+
+
+InterfaceWithTypename.model_rebuild()
+InterfaceWithTypenameQueryI.model_rebuild()

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

@@ -33,3 +33,9 @@ class ListInterfaceQueryListITypeB(BaseModel):
     typename__: Literal["TypeB"] = Field(alias="__typename")
     id: str
     field_b: str = Field(alias="fieldB")
+
+
+ListInterface.model_rebuild()
+ListInterfaceQueryListIInterface.model_rebuild()
+ListInterfaceQueryListITypeA.model_rebuild()
+ListInterfaceQueryListITypeB.model_rebuild()

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

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

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

@@ -28,3 +28,9 @@ class QueryWithFragmentOnInterfaceQueryITypeB(BaseModel):
     typename__: Literal["TypeB"] = Field(alias="__typename")
     id: str
     field_b: str = Field(alias="fieldB")
+
+
+QueryWithFragmentOnInterface.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,3 +3,6 @@
 
 class QueryWithFragmentOnQueryWithInterface(FragmentOnQueryWithInterface):
     pass
+
+
+QueryWithFragmentOnQueryWithInterface.model_rebuild()

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

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

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

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

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

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

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

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

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

@@ -10,3 +10,7 @@ class ExampleQuery1(BaseModel):
 
 class ExampleQuery1ExampleQuery(MinimalA):
     value: str
+
+
+ExampleQuery1.model_rebuild()
+ExampleQuery1ExampleQuery.model_rebuild()

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

@@ -10,3 +10,7 @@ class ExampleQuery2(BaseModel):
 
 class ExampleQuery2ExampleQuery(FullA):
     pass
+
+
+ExampleQuery2.model_rebuild()
+ExampleQuery2ExampleQuery.model_rebuild()

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

@@ -10,3 +10,7 @@ class ExampleQuery3(BaseModel):
 
 class ExampleQuery3ExampleQuery(CompleteA):
     pass
+
+
+ExampleQuery3.model_rebuild()
+ExampleQuery3ExampleQuery.model_rebuild()

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

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

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

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

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

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

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

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

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

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

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

@@ -8,3 +8,7 @@ class GetF(BaseModel):
 
 class GetFF(BaseModel):
     val: EnumF
+
+
+GetF.model_rebuild()
+GetFF.model_rebuild()

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

@@ -8,3 +8,7 @@ class GetG(BaseModel):
 
 class GetGG(FragmentG):
     pass
+
+
+GetG.model_rebuild()
+GetGG.model_rebuild()

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

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

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

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

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

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

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

@@ -14,3 +14,8 @@ class GetAA(BaseModel):
 
 class GetAAValueB(BaseModel):
     value: str
+
+
+GetA.model_rebuild()
+GetAA.model_rebuild()
+GetAAValueB.model_rebuild()

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

@@ -15,3 +15,8 @@ class GetAWithFragmentA(BaseModel):
 
 class GetAWithFragmentAValueB(FragmentB):
     pass
+
+
+GetAWithFragment.model_rebuild()
+GetAWithFragmentA.model_rebuild()
+GetAWithFragmentAValueB.model_rebuild()

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

@@ -7,3 +7,7 @@ class GetS(BaseModel):
 
 class GetSS(BaseModel):
     id: int
+
+
+GetS.model_rebuild()
+GetSS.model_rebuild()