|
@@ -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())
|