pycapnp/benchmark/addressbook.capnp.orphan.py
Madhava Jay fc617142c0 Fixed issue where pickle helpers didnt use new from_bytes with context
- Added with context to benchmarks that use with_bytes
2022-05-24 11:38:27 +10:00

50 lines
1.3 KiB
Python

import os
import capnp
this_dir = os.path.dirname(__file__)
addressbook = capnp.load(os.path.join(this_dir, "addressbook.capnp"))
print = lambda *x: x
def writeAddressBook():
addressBook = addressbook.AddressBook.new_message()
people = addressBook.init_resizable_list("people")
alice = people.add()
alice.id = 123
alice.name = "Alice"
alice.email = "alice@example.com"
alicePhones = alice.init("phones", 1)
alicePhones[0].number = "555-1212"
alicePhones[0].type = "mobile"
bob = people.add()
bob.id = 456
bob.name = "Bob"
bob.email = "bob@example.com"
bobPhones = bob.init("phones", 2)
bobPhones[0].number = "555-4567"
bobPhones[0].type = "home"
bobPhones[1].number = "555-7654"
bobPhones[1].type = "work"
people.finish()
msg_bytes = addressBook.to_bytes()
return msg_bytes
def printAddressBook(msg_bytes):
with addressbook.AddressBook.from_bytes(msg_bytes) as addressBook:
for person in addressBook.people:
print(person.name, ":", person.email)
for phone in person.phones:
print(phone.type, ":", phone.number)
print()
if __name__ == "__main__":
for i in range(10000):
msg_bytes = writeAddressBook()
printAddressBook(msg_bytes)