mirror of
https://github.com/capnproto/pycapnp.git
synced 2025-03-04 16:35:04 +01:00
64 lines
1.8 KiB
Python
64 lines
1.8 KiB
Python
from __future__ import print_function
|
|
import os
|
|
import capnp
|
|
|
|
this_dir = os.path.dirname(__file__)
|
|
addressbook = capnp.load(os.path.join(this_dir, 'addressbook.capnp'))
|
|
|
|
def writeAddressBook(fd):
|
|
message = capnp.MallocMessageBuilder()
|
|
addressBook = message.initRoot(addressbook.AddressBook)
|
|
people = addressBook.init('people', 2)
|
|
|
|
alice = people[0]
|
|
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'
|
|
alice.employment.school = "MIT"
|
|
|
|
bob = people[1]
|
|
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'
|
|
bob.employment.unemployed = None # This is definitely bad, syntax will change at some point
|
|
|
|
capnp.writePackedMessageToFd(fd, message)
|
|
|
|
|
|
def printAddressBook(fd):
|
|
message = capnp.PackedFdMessageReader(f.fileno())
|
|
addressBook = message.getRoot(addressbook.AddressBook)
|
|
|
|
for person in addressBook.people:
|
|
print(person.name, ':', person.email)
|
|
for phone in person.phones:
|
|
print(phone.type, ':', phone.number)
|
|
|
|
which = person.employment.which()
|
|
print(which)
|
|
|
|
if which == 'unemployed':
|
|
print('unemployed')
|
|
elif which == 'employer':
|
|
print('employer:', person.employment.employer)
|
|
elif which == 'school':
|
|
print('student at:', person.employment.school)
|
|
elif which == 'selfEmployed':
|
|
print('self employed')
|
|
print()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
f = open('example', 'w')
|
|
writeAddressBook(f.fileno())
|
|
|
|
f = open('example', 'r')
|
|
printAddressBook(f.fileno())
|