timesbee.py

#!/usr/bin/python3
#
# Count the number of possible New York Times Bee puzzles candidates.
# Sef Kloninger, 19-April-2020
#
# Takes a list of words on stdin. On MacOS, you can run like this:
#   pv /usr/share/dict/words | ./timesbee.py > timesbee.out

import sys

puzzlemap = {}
rejected = {}
considered = 0
pangrams = 0
vowels_set = set(['a', 'e', 'i', 'o', 'u', 'y'])


def rejection_tally(reason):
    if reason not in rejected.keys():
        rejected[reason] = 0
    rejected[reason] = rejected[reason] + 1


def valid_word(word, letter_set):
    valid = True
    if word[0] >= 'A' and word[0] <= 'Z':
        rejection_tally('proper noun')
        valid = False
    if len(letter_set) != 7:
        rejection_tally('not seven letters')
        valid = False
    if len(letter_set.intersection(vowels_set)) > 2:
        rejection_tally('too many vowels')
        valid = False
    return valid


def plural(count, noun):
    str = "{} {}".format(count, noun)
    if count == 1:
        return str
    else:
        return str+"s"


# Build a map of all puzzles
# key = seven letters for the puzzle
# value = list of all pangrams given those letters to work with.
for word in sys.stdin:
    word = word.strip()
    considered = considered + 1
    letter_set = set()
    for char in list(word):
        letter_set.add(char.lower())
    if not valid_word(word, letter_set):
        continue
    pangrams = pangrams + 1
    letters = ''.join(sorted(list(letter_set)))
    if letters not in puzzlemap.keys():
        puzzlemap[letters] = []
    puzzlemap[letters].append(word)

# Pivot by pangram count
# key = number of pangrams found
# value = list of tuples, each tuple is (letters, pangram list) from puzzlemap
pangram_counts = {}
for (letters, wordlist) in puzzlemap.items():
    count = len(wordlist)
    if count not in pangram_counts.keys():
        pangram_counts[count] = []
    found_tuple = (letters, wordlist)
    pangram_counts[count].append(found_tuple)

# Print out all we found, grouped by counts
for (count, pangrams_list) in sorted(pangram_counts.items()):
    print("----", plural(count, "pangram"), "----")
    for pangram_tuple in sorted(pangrams_list):
        print(pangram_tuple[0], sorted(pangram_tuple[1]))

print("---- found: %d puzzles, %d pangrams, of %d words considered"
      % (len(puzzlemap), pangrams, considered))
print("---- words rejected because:", rejected)