Update preferred method for reading/writing messages from files

This commit is contained in:
Jason Paryani
2013-09-01 02:13:19 -07:00
parent eb9e7511d0
commit 6fcdf841e4
7 changed files with 103 additions and 124 deletions

View File

@@ -5,10 +5,9 @@ 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)
def writeAddressBook(file):
addresses = addressbook.AddressBook.newMessage()
people = addresses.init('people', 2)
alice = people[0]
alice.id = 123
@@ -30,14 +29,13 @@ def writeAddressBook(fd):
bobPhones[1].type = 'work'
bob.employment.unemployed = None
capnp.writePackedMessageToFd(fd, message)
addresses.writeTo(file)
def printAddressBook(fd):
message = capnp.PackedFdMessageReader(f.fileno())
addressBook = message.getRoot(addressbook.AddressBook)
def printAddressBook(file):
addresses = addressbook.AddressBook.readFrom(file)
for person in addressBook.people:
for person in addresses.people:
print(person.name, ':', person.email)
for phone in person.phones:
print(phone.type, ':', phone.number)
@@ -58,7 +56,7 @@ def printAddressBook(fd):
if __name__ == '__main__':
f = open('example', 'w')
writeAddressBook(f.fileno())
writeAddressBook(f)
f = open('example', 'r')
printAddressBook(f.fileno())
printAddressBook(f)