Коровы и быки. Секретный уровень

import random as r
 
hist = []
 
 
def random_number():
    d = [i for i in range(10)]
    r.shuffle(d)
    conc = d[:4]
    if conc[0] == 0:
        conc[0] = conc[1]
        conc[1] = 0
    return conc
 
 
def ret_mass(num):
    num = list(str(num))
    return list(map(int, num))
 
 
def get_mass(num):
    mass_zag, mass_try = random_number(), ret_mass(num)
    bull, cow = 0, 0
    for i, j in zip(mass_zag, mass_try):
        if j in mass_zag:
            if j == i:
                bull += 1
            else:
                cow += 1
    return bull, cow

from random import choice
 
def first(res, mask, que):
    c, b = divmod(mask, 10)
    result = []
    for elem in res:
        if len(set(elem) & set(que)) == c:
            m = sum(elem[i] == que[i] for i in range(4))
            if m == b:
                result.append(elem)
    return result
 
def other(res, que):
    result = []
    cnt = 0
    for e in ms:
        r = first(res, e, que)
        if len(r) > cnt:
            result = r.copy()
            cnt = len(r)
            m = e
    return result, m
 
 
masks = (10, 20, 21, 30)
ms = [10, 11, 20, 21, 22, 30, 31, 32, 33, 40, 41, 42, 43, 44]
res = [str(i) for i in range(1000, 10000) if len(set(str(i)))==4]
 
q = input()
mask = choice(masks)
res = first(res, mask, q)
print(mask)
while mask != 44:
    q = input()
    res, mask = other(res, q)
    print(mask)
    # print(res)

# Импорт модулей
import random 
 
# Глобальные списки
digits = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
 
 
# Функция для генерации случайного числа
def generateNum():
    return random.sample(digits, 4)
 
    
# Функция для подсчета быков и коров
def numOfBullsCows(number, guess):
    bull_cow = [0, 0]
    guesslist = [int(x) for x in str(guess)]
    # Цикл для поиска быков и коров
    for i, j in zip(number, guesslist):
        # Есть ли цифра из ввода в искомом числе
        if j in number:
            # Если место цифры ввода и цифры искомого числа совпадают
            if j == i:
                bull_cow[0] += 1
            # Если место цифры ввода и цифры искомого числа не совпадают
            else:
                bull_cow[1] += 1
    return bull_cow
 
 
# Генерация числа
number = generateNum()
 
# Модификация для услонения игры
# Пользователь вводит число попыток, за которое он должен угадать число
# Если попытки истрачиваютя, пользователь проигрывает
tries = int(input('Введите кол-во попыток: '))
print()
k = 0
 
# Игровой процесс 
while tries > 0:
    guess = int(input("Введите число: "))
    bull_cow = numOfBullsCows(number, guess)
    print(f"{bull_cow[0]} быков, {bull_cow[1]} коров\n")
    tries -= 1
    k += 1
    if bull_cow[0] == 4:
        print(f"Вы угадали за {k} попыток")
        break
else:
    print(f"Вы истратили все свои попытки. Число было {number}")

import random
 
def num_to_digits(num):
    return [int(digit) for digit in str(num)]
 
def count_cows_bulls(secret_num, guess_num):
    secret_digits = num_to_digits(secret_num)
    guess_digits = num_to_digits(guess_num)
    bulls = sum([secret == guess for secret, guess in zip(secret_digits, guess_digits)])
    cows = len(set(secret_digits) & set(guess_digits)) - bulls
    return cows, bulls
 
def is_valid_new_secret(history, new_secret):
    for guess, cows, bulls in history:
        new_cows, new_bulls = count_cows_bulls(new_secret, guess)
        if new_cows != cows or new_bulls != bulls:
            return False
    return True
 
def generate_new_secret(history):
    while True:
        new_secret = random.randint(1000, 9999)
        if is_valid_new_secret(history, new_secret):
            return new_secret
 
def play_game():
    history = []
    secret_num = random.randint(1000, 9999)
    print("Welcome to Cows and Bulls! I have a 4 digit number. Can you guess it?")
    while True:
        guess_num = int(input("Enter your guess: "))
        while guess_num < 1000 or guess_num > 9999:
            print("Please enter a 4 digit number.")
            guess_num = int(input("Enter your guess: "))
        cows, bulls = count_cows_bulls(secret_num, guess_num)
        history.append((guess_num, cows, bulls))
        print(f"Cows: {cows}, Bulls: {bulls}")
        if bulls == 4:
            print("Congratulations! You guessed the number.")
            play_again = input("Do you want to play again? (yes/no): ")
            if play_again.lower() != 'yes':
                break
            else:
                history = []
                secret_num = random.randint(1000, 9999)
 
play_game()