game.py

# Masterminid game using python 
import random 

COLORS = ["R", "G","B","Y","W","O"]
TRIES = 10
CODE_LENGTH = 4

#Generating the code
def generate_code():
    code = []

    for _ in range(CODE_LENGTH):
        color = random.choice(COLORS)
        code.append(color)

    return code
#Allow the users to guess the code
def guess_code():
    while True:
        guess = input("Guess: ").upper().split(" ")  #.split creates a list by removing the spaces provided by the user as an input. 
        if len(guess) != CODE_LENGTH:
            print(f"You must guess {CODE_LENGTH} colors.")
            continue
        for color in guess:
            if color not in COLORS:
                print(f"invalid color: {color}. Try again")
        else:
            break

    return guess
#Check the code
def check_code(guess, real_code):
    colors_count = {}
    correct_pos = 0
    incorrect_pos = 0

    for color in real_code:
        if color not in colors_count:
            colors_count[color] = 0
        colors_count[color] +=1
    
    for guess_color, real_color in zip(guess, real_code): #zip gives the list of the given arguments. 
        if guess_color == real_color:
            correct_pos +=1
            colors_count[guess_color] -=1

    for guess_color, real_color in zip(guess, real_code):
        if guess_color in colors_count and colors_count[guess_color]>0:
            incorrect_pos +=1
            colors_count[guess_color] -=1

    return correct_pos, incorrect_pos

def game():
    print(f"Welcome to mastermind, you have {TRIES} to guess the code...")
    print("The valid colors are", *COLORS)
    code = generate_code()
    for attempts in range(1, TRIES+1):
        guess = guess_code()
        correct_pos, incorrect_pos = check_code(guess, code)

        if correct_pos == CODE_LENGTH:
            print("You guessed the code in {attempts} tries!")
            break

        print(f"Correct Positions: {correct_pos} | Incorrect Positions: {incorrect_pos}")

    else:
        print("You ran out of tries, the code was:", *code)

if __name__ == "__main__":
    game()