Merge pull request geekcomputers#347 from Brikaa/master

geekcomputers · web-flow · commit bdf7c758f4be · 2018-06-28T07:24:18.000+01:00
Added error exception
diff --git a/CountMillionCharacters-2.0.py b/CountMillionCharacters-2.0.py
@@ -1,9 +1,8 @@
-"""Get the number of each character in any given text.
 
+"""Get the number of each character in any given text.
 Inputs:
 A txt file -- You will be asked for an input file. Simply input the name
 of the txt file in which you have the desired text.
-
 """
 
 import pprint
@@ -13,12 +12,16 @@
 def main():
 
     file_input = input('File Name: ')
-
-    with open(file_input, 'r') as info:
-        count = collections.Counter(info.read().upper())
+    try:
+        with open(file_input, 'r') as info:
+            count = collections.Counter(info.read().upper())
+    except FileNotFoundError:
+        print("Please enter a valid file name.")
+        main()
 
     value = pprint.pformat(count)
     print(value)
+    exit()
 
 
 if __name__ == "__main__":
