Skip to content

Commit 98214bf

Browse files
committed
Fix input tyhpe generate: add "pass" for an empty input type class def
1 parent e8458ec commit 98214bf

File tree

2 files changed

+16
-0
lines changed

2 files changed

+16
-0
lines changed

ariadne_codegen/client_generators/input_types.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
generate_method_call,
2020
generate_module,
2121
generate_name,
22+
generate_pass,
2223
generate_pydantic_field,
2324
model_has_forward_refs,
2425
)
@@ -199,6 +200,9 @@ def _parse_input_definition(
199200
class_def.body.append(docstring)
200201
self._save_dependencies(root_type=definition.name, field_type=field_type)
201202

203+
if not class_def.body:
204+
class_def.body.append(generate_pass())
205+
202206
if self.plugin_manager:
203207
class_def = self.plugin_manager.generate_input_class(
204208
class_def, input_type=definition

tests/client_generators/input_types_generator/test_parsing_inputs.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,8 @@
2626
input CustomInput2 {
2727
field: Int!
2828
}
29+
30+
input CustomInput3
2931
""",
3032
[
3133
ast.ClassDef(
@@ -81,6 +83,16 @@
8183
],
8284
type_params=[],
8385
),
86+
ast.ClassDef(
87+
name="CustomInput3",
88+
bases=[ast.Name(id=BASE_MODEL_CLASS_NAME)],
89+
keywords=[],
90+
decorator_list=[],
91+
body=[
92+
ast.Pass()
93+
],
94+
type_params=[],
95+
),
8496
],
8597
)
8698
],

0 commit comments

Comments
 (0)