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])