combinator.py

combinator.py

FRUIT = "fruit"
VEGGIE = "veggie"

class Product:
    def __init__(self, name, kind):
        self.name = name
        self.kind = kind

    def __repr__(self):
        return "(name: %s / kind: %s)" % (self.name, self.kind)

# Collection of all products.
products = [
    Product("apple", FRUIT),
    Product("tangerine", FRUIT),
    Product("grape", FRUIT),
    Product("cucumber", VEGGIE),
    Product("sweet pea", VEGGIE),
    Product("celary", VEGGIE),
    Product("tomato", VEGGIE),
]

# Tuples of templates and tags that they implement.
templates = [
    ["i love {{fruit}}", [FRUIT]],
    ["i love {{fruit}} but only {{fruit}}", [FRUIT, FRUIT]],
    ["i love {{fruit}} and sometimes {{veggie}}", [FRUIT, VEGGIE]],
]

# Create dictionary of products by type.
templateDict = {
    FRUIT: list(filter(lambda p: p.kind == FRUIT, products)),
    VEGGIE: list(filter(lambda p: p.kind == VEGGIE, products)),
}

def combinator(arr):
    n = len(arr)
    indices = [0 for i in range(n)] 
    combset = []
    while (1):
        combs = []

        for i in range(n):
            combs.append(arr[i][indices[i]])

        combset.append(combs)

        next = n - 1
        while (next >= 0 and (indices[next] + 1 >= len(arr[next]))):
            next-=1

        if (next < 0):
            return combset

        indices[next] += 1

        for i in range(next + 1, n):
            indices[i] = 0

    return combset

for tpl in templates:
    columns = []

    for type in tpl[1]:
        productsForColumn = templateDict[type]
        columns.append(templateDict[type])

    for col in combinator(columns):
        print(tpl[0] + " - " + "".join(map(repr, col)))