diff --git a/protobuf_to_pydantic/gen_model.py b/protobuf_to_pydantic/gen_model.py index 5a2f34b3..71bf6a6b 100644 --- a/protobuf_to_pydantic/gen_model.py +++ b/protobuf_to_pydantic/gen_model.py @@ -432,7 +432,8 @@ def _protobuf_field_type_is_type_enum_handler(self, field_dataclass: FieldDataCl else: enum_class_dict = {v.name: v.number for v in protobuf_field.enum_type.values} _class_name = protobuf_field.enum_type.name - _class_doc = "" + _class_doc = f"Enumeration {protobuf_field.enum_type.name}:\n"\ + + ("\n".join([ f"- {v.name} = {v.number}" for v in protobuf_field.enum_type.values])) if field_dataclass.descriptor.file.name != protobuf_field.enum_type.file.name: _class_name = replace_file_name_to_class_name(protobuf_field.enum_type.file.name) + _class_name _class_doc = (