add_person.py 1.71 KB
Newer Older
1
#! /usr/bin/env python
temporal's avatar
temporal committed
2 3 4 5 6 7

# See README.txt for information and build instructions.

import addressbook_pb2
import sys

8 9 10 11 12 13
try:
  raw_input          # Python 2
except NameError:
  raw_input = input  # Python 3


temporal's avatar
temporal committed
14 15 16 17 18 19 20 21 22 23 24 25 26 27
# This function fills in a Person message based on user input.
def PromptForAddress(person):
  person.id = int(raw_input("Enter person ID number: "))
  person.name = raw_input("Enter name: ")

  email = raw_input("Enter email address (blank for none): ")
  if email != "":
    person.email = email

  while True:
    number = raw_input("Enter a phone number (or leave blank to finish): ")
    if number == "":
      break

Jan Tattermusch's avatar
Jan Tattermusch committed
28
    phone_number = person.phones.add()
temporal's avatar
temporal committed
29 30 31 32 33 34 35 36 37 38
    phone_number.number = number

    type = raw_input("Is this a mobile, home, or work phone? ")
    if type == "mobile":
      phone_number.type = addressbook_pb2.Person.MOBILE
    elif type == "home":
      phone_number.type = addressbook_pb2.Person.HOME
    elif type == "work":
      phone_number.type = addressbook_pb2.Person.WORK
    else:
39 40
      print("Unknown phone type; leaving as default value.")

temporal's avatar
temporal committed
41 42 43 44 45

# Main procedure:  Reads the entire address book from a file,
#   adds one person based on user input, then writes it back out to the same
#   file.
if len(sys.argv) != 2:
46
  print("Usage:", sys.argv[0], "ADDRESS_BOOK_FILE")
temporal's avatar
temporal committed
47 48 49 50 51 52
  sys.exit(-1)

address_book = addressbook_pb2.AddressBook()

# Read the existing address book.
try:
53 54
  with open(sys.argv[1], "rb") as f:
    address_book.ParseFromString(f.read())
temporal's avatar
temporal committed
55
except IOError:
56
  print(sys.argv[1] + ": File not found.  Creating a new file.")
temporal's avatar
temporal committed
57 58

# Add an address.
59
PromptForAddress(address_book.people.add())
temporal's avatar
temporal committed
60 61

# Write the new address book back to disk.
62 63
with open(sys.argv[1], "wb") as f:
  f.write(address_book.SerializeToString())