diff options
author | Jon Skeet <skeet@pobox.com> | 2008-08-14 20:33:26 +0100 |
---|---|---|
committer | unknown <Jon@.(none)> | 2008-08-14 20:33:26 +0100 |
commit | f92c545f493dce6e5a2e378e8268c1e7b0e55a4d (patch) | |
tree | 7c2a4a78ca01c82d165c9ece810be052ff6cc691 /examples/add_person.py | |
download | protobuf-f92c545f493dce6e5a2e378e8268c1e7b0e55a4d.tar.gz protobuf-f92c545f493dce6e5a2e378e8268c1e7b0e55a4d.tar.bz2 protobuf-f92c545f493dce6e5a2e378e8268c1e7b0e55a4d.zip |
Initial commit of C# code developed before installing Git.
committer: Jon Skeet <skeet@pobox.com>
Diffstat (limited to 'examples/add_person.py')
-rwxr-xr-x | examples/add_person.py | 58 |
1 files changed, 58 insertions, 0 deletions
diff --git a/examples/add_person.py b/examples/add_person.py new file mode 100755 index 00000000..78e56966 --- /dev/null +++ b/examples/add_person.py @@ -0,0 +1,58 @@ +#! /usr/bin/python + +# See README.txt for information and build instructions. + +import addressbook_pb2 +import sys + +# 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 + + phone_number = person.phone.add() + 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: + print "Unknown phone type; leaving as default value." + +# 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: + print "Usage:", sys.argv[0], "ADDRESS_BOOK_FILE" + sys.exit(-1) + +address_book = addressbook_pb2.AddressBook() + +# Read the existing address book. +try: + f = open(sys.argv[1], "rb") + address_book.ParseFromString(f.read()) + f.close() +except IOError: + print sys.argv[1] + ": File not found. Creating a new file." + +# Add an address. +PromptForAddress(address_book.person.add()) + +# Write the new address book back to disk. +f = open(sys.argv[1], "wb") +f.write(address_book.SerializeToString()) +f.close() |