Home | History | Annotate | Download | only in examples
      1 #! /usr/bin/python
      2 
      3 # See README.txt for information and build instructions.
      4 
      5 import addressbook_pb2
      6 import sys
      7 
      8 # This function fills in a Person message based on user input.
      9 def PromptForAddress(person):
     10   person.id = int(raw_input("Enter person ID number: "))
     11   person.name = raw_input("Enter name: ")
     12 
     13   email = raw_input("Enter email address (blank for none): ")
     14   if email != "":
     15     person.email = email
     16 
     17   while True:
     18     number = raw_input("Enter a phone number (or leave blank to finish): ")
     19     if number == "":
     20       break
     21 
     22     phone_number = person.phone.add()
     23     phone_number.number = number
     24 
     25     type = raw_input("Is this a mobile, home, or work phone? ")
     26     if type == "mobile":
     27       phone_number.type = addressbook_pb2.Person.MOBILE
     28     elif type == "home":
     29       phone_number.type = addressbook_pb2.Person.HOME
     30     elif type == "work":
     31       phone_number.type = addressbook_pb2.Person.WORK
     32     else:
     33       print "Unknown phone type; leaving as default value."
     34 
     35 # Main procedure:  Reads the entire address book from a file,
     36 #   adds one person based on user input, then writes it back out to the same
     37 #   file.
     38 if len(sys.argv) != 2:
     39   print "Usage:", sys.argv[0], "ADDRESS_BOOK_FILE"
     40   sys.exit(-1)
     41 
     42 address_book = addressbook_pb2.AddressBook()
     43 
     44 # Read the existing address book.
     45 try:
     46   f = open(sys.argv[1], "rb")
     47   address_book.ParseFromString(f.read())
     48   f.close()
     49 except IOError:
     50   print sys.argv[1] + ": File not found.  Creating a new file."
     51 
     52 # Add an address.
     53 PromptForAddress(address_book.person.add())
     54 
     55 # Write the new address book back to disk.
     56 f = open(sys.argv[1], "wb")
     57 f.write(address_book.SerializeToString())
     58 f.close()
     59