Generate Reading list from a csv list of recomendations.

constrain.py

import csv

HARD_MODE = 2
NORMAL_MODE = 3
recs_column = HARD_MODE
name_column = 0

from constraint import *
problem = Problem()

with open('recomendations.csv') as csv_file:
    csv_reader = csv.reader(csv_file, delimiter=',')
    line_count = 0
    for row in csv_reader:
        if line_count == 0:
            line_count += 1
            continue
        if line_count > 5:
            break
        square_name = row[name_column]
        recomendations = row[recs_column].split(',')
        print(square_name, recomendations)
        problem.addVariable(square_name, recomendations)
        line_count += 1

    print(f'Processed {line_count} lines.')

problem.addConstraint(AllDifferentConstraint())
print(problem.getSolutions())