One more defense in style #19

2013-09-25 19:09:24 -07:00
parent 88a9cfaa3e
commit aa54d93746
1 changed files with 5 additions and 1 deletions
--- a/19-defensive/tf-19.py
+++ b/19-defensive/tf-19.py
@@ -64,7 +64,11 @@ def sort(word_freq):
 #
 # The main function
 #
-word_freqs = sort(frequencies(extract_words(sys.argv[1])))
+#if (len(sys.argv) < 2):
+#    filename = "../input.txt"
+#else:
+filename = sys.argv[1] if len(sys.argv) > 1 else "../input.txt"
+word_freqs = sort(frequencies(extract_words(filename)))

 for tf in word_freqs[0:25]:
    print tf[0], ' - ', tf[1]