CreateSudokus

import itertools
import numpy as np
import random
import time

def possible(num,row,col, printReason=False):
    global sudoku

    # check row
    for i in range(9):
        if sudoku[row, i] == num:
            if printReason:
                print(f"{num} tested at {row},{col} --> already in the same row (at col {i})")
            return False
    # check column
    for i in range(9):
        if sudoku[i, col] == num:
            if printReason:
                print(f"{num} tested at {row},{col} --> already in the same col (at row {i})")
            return False
    # check square
    row0 = (row//3)*3
    col0 = (col//3)*3
    for y in range(row0, row0+3):
        for x in range(col0, col0+3):
            if sudoku[y,x] == num:
                if printReason:
                    print(f"{num} tested at {row},{col} --> already in the same square (at square with upper left corner {row0}, {col0})")
                return False
    return True

def solveSudoku(random_nums=True, onlyFirstSolution=False, sleep_sec=0.5, suppress_print=False):
    ## make settings global
    ### default: random_nums=True, onlyFirstSolution=False, sleep_sec=0.5, suppress_print=False
    global s1
    global s2
    global s3
    global s4

    global sudoku
    global counter

    for row in range(9):
        for col in range(9):
            if sudoku[row,col] == 0:
                nums1to9 = np.arange(1,10)
                if random_nums:
                    random.shuffle(nums1to9)
                for num in nums1to9:
                    if possible(num, row, col):
                        sudoku[row,col] = num
                        solveSudoku(random_nums=s1, onlyFirstSolution=s2, sleep_sec=s3, suppress_print=s4)
                        sudoku[row,col] = 0 # try new possibilities (non-unique Sudoku)
                return # backtracking (empty field, but no number fits) --> does not print out the "wrong" Sudoku

    counter += 1
    time.sleep(sleep_sec)

    if not suppress_print:
        # Check if Sudoku is the same
        if counter == 1:
            try:
                if np.array_equal(first_sudoku, sudoku):
                    print("Sudoku is not solvable/over-constrained!")
                    return
            except:
                print("Define the variable first_sudoku.")

    if not suppress_print:
        print(f"Solution #{counter}:\n")
        print(sudoku, "\n")

    if onlyFirstSolution:
        return sudoku

    ## PREVENT CPU OVERLOAD ##
    if counter > 100:
        return

    return sudoku

def createSudoku():
    ## make settings global
    ### default: random_nums=True, onlyFirstSolution=False, sleep_sec=0.5, suppress_print=False
    global s1
    global s2
    global s3
    global s4

    global sudoku
    global first_sudoku
    global counter

    sudoku = np.zeros((9,9))
    first_sudoku = sudoku.copy()
    counter = 0
    s1 = True
    s2 = True
    s3 = 0.5
    s4 = True
    sudoku = solveSudoku(random_nums=s1, onlyFirstSolution=s2, sleep_sec=s3, suppress_print=s4)
    print("---")
    print(sudoku)
    print("----")
    ## user selects difficulty level
    while True:
        numClues_input = input("Choose difficulty level:\n1) Extremely easy\n2) Easy\n3) Medium\n4) Difficult\n5) Evil\n\n--> ")
        if numClues_input.lower() in ["1", "1)", "one", "eins", "i", "extremely easy"]:
            numClues = random.randint(47,80)
            break
        elif numClues_input.lower() in ["2", "2)", "two", "zwei", "ii", "easy"]:
            numClues = random.randint(36,46)
            break
        elif numClues_input.lower() in ["3", "3)", "three", "drei", "iii", "medium"]:
            numClues = random.randint(32,35)
            break
        elif numClues_input.lower() in ["4", "4)", "four", "vier", "iv", "difficult"]:
            numClues = random.randint(28,31)
            break
        elif numClues_input.lower() in ["5", "5)", "five", "fünf", "v", "evil"]:
            numClues = random.randint(17,27)
            break
        print("\nSelect difficulty level...\n")

    print(f"\nYou have selected difficulty level {numClues_input}. Therefore you will get {numClues} clues.")

    numRemoveClues = 81-numClues

    ## remove up to 81-17=64 clues until there is a unique solution
    allCoordinates = list(itertools.product(range(0,9), repeat=2))

    while_counter = 0
    start = time.time()
    while True:
        removeCoordinates = random.choices(allCoordinates, k=numRemoveClues)
        for coord in removeCoordinates:
            row = coord[0]
            col = coord[1]
            sudoku[row, col] = 0
        counter = 0
        first_sudoku = sudoku.copy()
        s2 = False # 'tis only needed to change to multiple solutions
        sudoku = solveSudoku(random_nums=s1, onlyFirstSolution=s2, sleep_sec=s3, suppress_print=s4)
        if counter == 1:
            return sudoku

        while_counter += 1
        if while_counter in [10,100,1000,10000,100000,1000000]:
            print(f"{while_counter} Sudokus with {numClues} clues have been found so far, but none of them is unique...")
        stop = time.time()
        elapsed_time = stop-start
        if elapsed_time > 120:
            print("Time elapsed has exceed the maximum waiting time of 120s.")
            break