diff options
-rwxr-xr-x | main.py | 22 |
1 files changed, 14 insertions, 8 deletions
@@ -2,20 +2,26 @@ import sys try: - word_num = int(sys.argv[1]) + word_num = int(sys.argv[1]) except IndexError: - word_num = None + word_num = None WORDLIST_FILE_PATH = "corncob_lowercase.txt" with open(WORDLIST_FILE_PATH, "r") as wordlist_file: - valid_words = set([word[0].upper() + word.rstrip()[1:] for word in wordlist_file.readlines() if word.startswith("d")]) + valid_words = set( + [ + word[0].upper() + word.rstrip()[1:] + for word in wordlist_file.readlines() + if word.startswith("d") + ] + ) sys.stdout.write("Dungeons & Dragons") for word in valid_words: - if word_num is not None: - if word_num <= 0: - break - word_num -= 1 - sys.stdout.write(" & " + word) + if word_num is not None: + if word_num <= 0: + break + word_num -= 1 + sys.stdout.write(" & " + word) print() |