from constraint import *

problem = Problem()
problem.addVariables(range(0, 9), range(1, 9 + 1))
problem.addConstraint(AllDifferentConstraint(), range(0, 9))
problem.addConstraint(ExactSumConstraint(15), [0, 4, 8])
problem.addConstraint(ExactSumConstraint(15), [2, 4, 6])
for row in range(3):
	problem.addConstraint(ExactSumConstraint(15),
						  [row * 3 + i for i in range(3)])
for col in range(3):
	problem.addConstraint(ExactSumConstraint(15),
						  [col + 3 * i for i in range(3)])
solutions = problem.getSolution()
print(solutions)