From 7073f33dd61135982fdd0377b3b16187c5877d00 Mon Sep 17 00:00:00 2001 From: amj2 Date: Wed, 20 Dec 2023 12:20:58 +0000 Subject: [PATCH] fix: handle empty proto messages --- protobuf2pydantic/biz.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/protobuf2pydantic/biz.py b/protobuf2pydantic/biz.py index 80ca738..123afd4 100644 --- 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])