Index: third_party/protobuf/examples/add_person.py |
diff --git a/third_party/protobuf/examples/add_person.py b/third_party/protobuf/examples/add_person.py |
index fd81c982110941cc46476208c66ab1d60099e5f3..0b6985792d7a5fcef545a44d085d929dab989c4e 100755 |
--- a/third_party/protobuf/examples/add_person.py |
+++ b/third_party/protobuf/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()) |