Ver Fonte

Updated python examples to use with.

Parth Kolekar há 9 anos atrás
pai
commit
21e1f1d2e9
2 ficheiros alterados com 6 adições e 9 exclusões
  1. 4 6
      examples/add_person.py
  2. 2 3
      examples/list_people.py

+ 4 - 6
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())

+ 2 - 3
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)