Skip to content

Commit 92d721a

Browse files
author
John Lyu
committed
add field docstring to custom_fields class
1 parent ab9030c commit 92d721a

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

ariadne_codegen/client_generators/custom_fields.py

+8-3
Original file line numberDiff line numberDiff line change
@@ -141,9 +141,9 @@ def _generate_class_def_body(
141141
class_def = generate_class_def(
142142
name=class_name, base_names=base_names, description=description
143143
)
144-
for lineno, (org_name, field) in enumerate(
145-
self._get_combined_fields(definition).items(), start=1
146-
):
144+
lineno = 0
145+
for org_name, field in self._get_combined_fields(definition).items():
146+
lineno += 1
147147
name = process_name(
148148
org_name, convert_to_snake_case=self.convert_to_snake_case
149149
)
@@ -158,6 +158,11 @@ def _generate_class_def_body(
158158
name, field_name, org_name, field, method_required, lineno
159159
)
160160
)
161+
if field.description:
162+
lineno += 1
163+
docstring = ast.Expr(value=ast.Constant(field.description), lineno=lineno)
164+
class_def.body.append(docstring)
165+
161166

162167
class_def.body.append(
163168
self._generate_fields_method(

0 commit comments

Comments
 (0)