📝 Open files with contextmanager

Author: veit

docs/modules/wc.py

@@ -6,9 +6,8 @@ def words_occur():
     # Prompt user for the name of the file to use.
     file_name = input("Enter the name of the file: ")
     # Open the file, read it and store its words in a list.
-    f = open(file_name, "r")
-    word_list = f.read().split()
-    f.close()
+    with open(file_name, "r") as f:
+        word_list = f.read().split()
     # Count the number of occurrences of each word in the file.
     occurs_dict = {}
     for word in word_list:

docs/modules/wcargparse.py

@@ -11,18 +11,17 @@ def words_occur():
     args = parser.parse_args()
     file_name = args.filename
     # Open the file, read it and store its words in a list.
-    f = open(file_name, "r")
-    word_list = f.read().split()
-    f.close()
+    with open(file_name, "r") as f:
+        word_list = f.read().split()
     # Count the number of occurrences of each word in the file.
     occurs_dict = {}
     for word in word_list:
         # increment the occurrences count for this word
         occurs_dict[word] = occurs_dict.get(word, 0) + 1
     # Print out the results.
     print(
-        "File %s has %d words (%d are unique)"
-        % (file_name, len(word_list), len(occurs_dict))
+        f"File {file_name} has {len(word_list)} words, "
+        f"{len(occurs_dict)} are unique:"
     )
     print(occurs_dict)
 

docs/modules/wcargv.py

@@ -8,9 +8,8 @@ def words_occur():
     # Prompt user for the name of the file to use.
     file_name = sys.argv.pop()
     # Open the file, read it and store its words in a list.
-    f = open(file_name, "r")
-    word_list = f.read().split()
-    f.close()
+    with open(file_name, "r") as f:
+        word_list = f.read().split()
     # Count the number of occurrences of each word in the file.
     occurs_dict = {}
     for word in word_list: