diff options
author | Parth Kolekar <parth.kolekar@students.iiit.ac.in> | 2016-01-26 04:13:31 +0530 |
---|---|---|
committer | Parth Kolekar <parth.kolekar@students.iiit.ac.in> | 2016-01-26 04:13:31 +0530 |
commit | 21e1f1d2e99cf194185107fbbd15620b36d99e37 (patch) | |
tree | b6af1b33181286b35c11eefc7dd407a1830a367d /examples | |
parent | 60f7fc51fb01ac98fd64d9372e7547f5ab267ce5 (diff) |
Updated python examples to use with.
Diffstat (limited to 'examples')
-rwxr-xr-x | examples/add_person.py | 10 | ||||
-rwxr-xr-x | examples/list_people.py | 5 |
2 files changed, 6 insertions, 9 deletions
diff --git a/examples/add_person.py b/examples/add_person.py index fd81c982..0b698579 100755 --- a/examples/add_person.py +++ b/examples/add_person.py @@ -43,9 +43,8 @@ address_book = addressbook_pb2.AddressBook() # Read the existing address book. try: - f = open(sys.argv[1], "rb") - address_book.ParseFromString(f.read()) - f.close() + 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." @@ -53,6 +52,5 @@ except IOError: PromptForAddress(address_book.people.add()) # Write the new address book back to disk. -f = open(sys.argv[1], "wb") -f.write(address_book.SerializeToString()) -f.close() +with open(sys.argv[1], "wb") as f: + f.write(address_book.SerializeToString()) diff --git a/examples/list_people.py b/examples/list_people.py index 755de901..f131872d 100755 --- a/examples/list_people.py +++ b/examples/list_people.py @@ -31,8 +31,7 @@ if len(sys.argv) != 2: address_book = addressbook_pb2.AddressBook() # Read the existing address book. -f = open(sys.argv[1], "rb") -address_book.ParseFromString(f.read()) -f.close() +with open(sys.argv[1], "rb") as f: + address_book.ParseFromString(f.read()) ListPeople(address_book) |