Given a set of letters and a length, finds all words known to hunspell dictionary

4i1m.py

#!/usr/bin/env python3
import argparse
import os
import sys

from itertools import permutations

import hunspell


DESCRIPTION = """
Given a set of letters and a length, finds all words known to hunspell
dictionary.
"""


DICT_DIR = '/usr/share/hunspell'


def main():
    parser = argparse.ArgumentParser()
    parser.description = DESCRIPTION

    parser.add_argument('-d', '--dict', default='fr',
                        help='Dictionary to use, defaults to "fr"')
    parser.add_argument('letters', help='Letters allowed in the word')
    parser.add_argument('length', type=int,
                        help='Number of letters in the word to find')

    args = parser.parse_args()

    hobj = hunspell.HunSpell(os.path.join(DICT_DIR, args.dict + '.dic'),
                             os.path.join(DICT_DIR, args.dict + '.aff'))

    words = set()
    for lst in permutations(sorted(args.letters), args.length):
        word = ''.join(lst)
        if word in words:
            continue
        if hobj.spell(word):
            words.add(word)
            print(word)

    return 0


if __name__ == '__main__':
    sys.exit(main())
# vi: ts=4 sw=4 et