from __future__ import print_function
import addressbook_pb2 as addressbook
import os

print = lambda *x: x


def writeAddressBook(fd):
    addressBook = addressbook.AddressBook()

    alice = addressBook.person.add()
    alice.id = 123
    alice.name = 'Alice'
    alice.email = 'alice@example.com'
    alicePhones = [alice.phone.add()]
    alicePhones[0].number = "555-1212"
    alicePhones[0].type = addressbook.Person.MOBILE

    bob = addressBook.person.add()
    bob.id = 456
    bob.name = 'Bob'
    bob.email = 'bob@example.com'
    bobPhones = [bob.phone.add(), bob.phone.add()]
    bobPhones[0].number = "555-4567"
    bobPhones[0].type = addressbook.Person.HOME
    bobPhones[1].number = "555-7654"
    bobPhones[1].type = addressbook.Person.WORK

    message_string = addressBook.SerializeToString()
    fd.write(message_string)


def printAddressBook(fd):
    addressBook = addressbook.AddressBook()
    addressBook.ParseFromString(fd.read())

    for person in addressBook.person:
        print(person.name, ':', person.email)
        for phone in person.phone:
            print(phone.type, ':', phone.number)
        print()


if __name__ == '__main__':
    for i in range(10000):
        f = open('example', 'w')
        writeAddressBook(f)

        f = open('example', 'r')
        printAddressBook(f)

os.remove('example')