diff options
author | Loo Rong Jie <loorongjie@gmail.com> | 2018-06-21 11:29:04 +0800 |
---|---|---|
committer | Jakob Buchgraber <buchgr@google.com> | 2018-07-09 16:28:56 +0200 |
commit | 6fd4e0edd4de22dec9eda13dc0b29214f2ca117e (patch) | |
tree | e05b713ab4eceb88e6686e34e2a98426b16009bc /third_party/protobuf/3.6.0/examples/add_person.py | |
parent | 868060b2925a98a22c4a2e760dd38da0bef6146c (diff) |
Update protobuf to 3.6.0. Fixes #5439
Diffstat (limited to 'third_party/protobuf/3.6.0/examples/add_person.py')
-rw-r--r-- | third_party/protobuf/3.6.0/examples/add_person.py | 63 |
1 files changed, 63 insertions, 0 deletions
diff --git a/third_party/protobuf/3.6.0/examples/add_person.py b/third_party/protobuf/3.6.0/examples/add_person.py new file mode 100644 index 0000000000..aa0fbca737 --- /dev/null +++ b/third_party/protobuf/3.6.0/examples/add_person.py @@ -0,0 +1,63 @@ +#! /usr/bin/env python + +# See README.txt for information and build instructions. + +import addressbook_pb2 +import sys + +try: + raw_input # Python 2 +except NameError: + raw_input = input # Python 3 + + +# 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.phones.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: + with open(sys.argv[1], "rb") as f: + address_book.ParseFromString(f.read()) +except IOError: + print(sys.argv[1] + ": File not found. Creating a new file.") + +# Add an address. +PromptForAddress(address_book.people.add()) + +# Write the new address book back to disk. +with open(sys.argv[1], "wb") as f: + f.write(address_book.SerializeToString()) |