Skip to content

Commit ab9030c

Browse files
author
John Lyu
committed
add docstring for input field
1 parent 55fab7f commit ab9030c

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

ariadne_codegen/client_generators/input_types.py

+7-2
Original file line numberDiff line numberDiff line change
@@ -161,8 +161,9 @@ def _parse_input_definition(
161161
base_names=[BASE_MODEL_CLASS_NAME],
162162
description=definition.description,
163163
)
164-
165-
for lineno, (org_name, field) in enumerate(definition.fields.items(), start=1):
164+
lineno = 0
165+
for org_name, field in definition.fields.items():
166+
lineno += 1
166167
name = process_name(
167168
org_name,
168169
convert_to_snake_case=self.convert_to_snake_case,
@@ -192,6 +193,10 @@ def _parse_input_definition(
192193
field_implementation, input_field=field, field_name=org_name
193194
)
194195
class_def.body.append(field_implementation)
196+
if field.description:
197+
lineno += 1
198+
docstring = ast.Expr(value=ast.Constant(value=field.description))
199+
class_def.body.append(docstring)
195200
self._save_dependencies(root_type=definition.name, field_type=field_type)
196201

197202
if self.plugin_manager:

0 commit comments

Comments
 (0)