Compile protobuf strings into Python module objects at runtime. — no files, no code generation step:
from proto_topy import ProtoModule
module = ProtoModule(
source="message Hello { optional string name = 1; }",
).compiled()
msg = module.py.Hello(name="world")
assert msg.name == "world"It is useful for programs needing to en/decode protobuf messages for which the definition is provided as a string at runtime.
pip install proto-topy
Prerequisite: proto-topy needs protoc to be installed. On macOS, a simple brew install protobuf shall suffice.
Adaptation of the protocolbuffers example from Google tutorial:
from io import BytesIO
import requests
from proto_topy import ProtoModule
example_source = requests.get(
"https://raw.githubusercontent.com/protocolbuffers/protobuf/main/"
"examples/addressbook.proto").text
module = ProtoModule(source=example_source).compiled()
# Serialize an address book
buffer = BytesIO()
address_book = module.py.AddressBook()
person = address_book.people.add()
person.id = 111
person.name = "A Name"
person.email = "a.name@mail.com"
phone_number = person.phones.add()
phone_number.number = "+1234567"
phone_number.type = module.py.Person.MOBILE
buffer.write(address_book.SerializeToString())
# Deserialize it back
buffer.seek(0)
address_book = module.py.AddressBook()
address_book.ParseFromString(buffer.read())
for person in address_book.people:
assert person.id == 111
assert person.name == "A Name"
assert person.email == "a.name@mail.com"When several definition strings need to be considered, use a ProtoCollection:
from proto_topy import ProtoModule, ProtoCollection
from google.protobuf.timestamp_pb2 import Timestamp
module1 = ProtoModule(
file_path="p1/p2/other2.proto",
source="""
syntax = "proto3";
import "google/protobuf/timestamp.proto";
message OtherThing2 {
google.protobuf.Timestamp created = 1;
}"""
)
module2 = ProtoModule(
file_path="p3/p4/test6.proto",
source="""
syntax = "proto3";
import "p1/p2/other2.proto";
message Test6 {
OtherThing2 foo = 1;
}"""
)
collection = ProtoCollection(module1, module2).compiled()
Test6 = collection.modules["p3/p4/test6.proto"].py.Test6
OtherThing2 = collection.modules["p1/p2/other2.proto"].py.OtherThing2
ts = Timestamp(seconds=1234567890)
thing = OtherThing2(created=ts)
msg = Test6(foo=thing)
assert msg.foo.created.seconds == 1234567890To decode a stream of contiguous protobuf messages of the same type, use DelimitedMessageFactory. Example:
from io import BytesIO
from proto_topy import ProtoModule, DelimitedMessageFactory
module = ProtoModule(
source="""
syntax = "proto3";
message TestInt { int32 val = 1; }
"""
).compiled()
integers = (module.py.TestInt(val=val) for val in range(10))
factory = DelimitedMessageFactory(BytesIO(), *integers)
factory.rewind()
for i, (offset, msg) in enumerate(factory.message_read(module.py.TestInt)):
assert msg.val == i