mastermind.py

#!/usr/bin/env python3


def chiffres(i):
    return i // 100, (i // 10) % 10, i % 10


def bad(a, b, place, desordre, absent):
    aa = a
    a = chiffres(a)
    b = chiffres(b)

    if a[0] == b[0]:
        place -= 1
    if a[1] == b[1]:
        place -= 1
    if a[2] == b[2]:
        place -= 1
    if place != 0:
        return True

    if a[0] == b[1] or a[0] == b[2]:
        desordre -= 1
    if a[1] == b[0] or a[1] == b[2]:
        desordre -= 1
    if a[2] == b[0] or a[2] == b[1]:
        desordre -= 1
    if desordre != 0:
        return True

    return False


for i in range(111, 1000):

    if bad(i, 795, 0, 0, 3):
        continue
    if bad(i, 147, 0, 1, 0):
        continue
    if bad(i, 368, 1, 0, 0):
        continue
    if bad(i, 218, 0, 2, 0):
        continue
    if bad(i, 841, 1, 0, 0):
        continue
    if bad(i, 639, 0, 1, 0):
        continue

    print(i)