wordscape/candidate_cache.py

import itertools
from dictionary import Dictionary, DICTS


def make_key(dictionary_name, letters):
    return (dictionary_name, frozenset(letters))


def candidates(letters, dictionary: Dictionary, min=2):
    permutations = itertools.chain.from_iterable(
        map(lambda r: itertools.permutations(letters, r), range(min, len(letters) + 1))
    )
    return dictionary.filter(map("".join, permutations))


class CandidateCache:
    def __init__(self, maxsize=5):
        self.maxsize = maxsize
        self.cache = {}
        self.keys = []

    def get(self, dictionary_name, letters):
        key = make_key(dictionary_name, letters)
        try:
            return self.cache[key]
        except KeyError:
            data = candidates(letters, DICTS[dictionary_name])
            self.create(key, data)
            return data

    def create(self, key, data):
        if len(self.keys) > self.maxsize:
            key = self.keys.pop()
            del self.cache[key]
        self.keys.insert(0, key)
        self.cache[key] = data