Merge pull request #7 from aidanjakes/main

Handle empty proto messages
Ed-XCF · Dec 21, 2023 · dab989b · dab989b
2 parents d184c11 + 7073f33
commit dab989b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/protobuf2pydantic/biz.py b/protobuf2pydantic/biz.py
@@ -92,7 +92,10 @@ def convert_field(level: int, field: FieldDescriptor) -> str:
 
 def msg2pydantic(level: int, msg: Descriptor) -> str:
  class_statement = f"{tab * level}class {msg.name}(BaseModel):"
- field_statements = map(partial(convert_field, level), msg.fields)
+ if msg.fields:
+ field_statements = map(partial(convert_field, level), msg.fields)
+ else:
+ field_statements = [tab + "pass"]
  return linesep.join([class_statement, *field_statements])