Mastermind.py

import random
import os

TORANGE = "\033[1;30;33m"
TYELLOW = "\033[1;33m"
TGREEN = "\33[1;92m"
TNOEFFECT = "\033[0m"
TBLACKBG = "\33[40m"
TWHITEBG = "\33[107m"

play_again = "y"
guess = ""
all_pegs = [
    "R",
    "O",
    "Y",
    "G",
    "B",
    "P",
    "S",
    "L",
    "M",
    "C",
    "V",
    "T",
]
all_colours = [
    "(R)ed",
    "(O)range",
    "(Y)ellow",
    "(G)reen",
    "(B)lue",
    "(P)urple",
    "(S)ilver",
    "(L)ime",
    "(M)agenta",
    "(C)yan",
    "(V)iolet",
    "(T)urquoise",
]


def split(word):
    global char
    return [char for char in word]


print(TGREEN + "Welcome to MASTERMIND!" + TNOEFFECT)

pegs_list = []
solution_pegs = []
colours = []
player1_wins = 0
player2_wins = 0
player_wins = 0
bot_wins = 0
player1_name = ""
player2_name = ""

game_mode = input(
    "\nWhich game mode would you like to play: Player vs Player (1) or Player vs Bot (2): "
).strip()
while game_mode not in ("1", "2"):
    game_mode = input(
        "Please enter either 1 or 2 (according to the game mode you wish to play)\n\nWhich game mode would you like to play: Player vs Player (1) or Player vs Bot (2): "
    ).strip()

pegs_num = input(
    "\nHow many pegs did you want to be selected? There will be 2 more possible colours. "
    "(Standard: 4, Min: 2, Max: 10): "
).strip()
while not pegs_num.isdigit() or int(pegs_num) not in range(2, 11):
    pegs_num = input(
        "Please enter a whole number between 2 and 10 (inclusive of)\n\n"
        "How many pegs did you want to be selected? (Standard: 4, Min: 2, Max: 10): "
    ).strip()

pegs_num = int(pegs_num)
max_guesses = (int(pegs_num) * 3) - 2

for x in range(0, pegs_num + 2):
    pegs_list.append(all_pegs[x])
    colours.append(all_colours[x])

pegs = "".join(pegs_list)
colours = ", ".join(colours)

if game_mode == "1":
    player1_name = input("\nPlayer 1 - select your name: ").strip()
    while not player1_name.isalpha():
        player1_name = input(
            "Please enter your name\n\nPlayer 1 - select your name: "
        ).strip()
    player2_name = input("Player 2 - select your name: ").strip()
    while not player2_name.isalpha():
        player2_name = input(
            "Please enter your name\n\nPlayer 1 - select your name: "
        ).strip()

    peg_chooser_num = random.randint(1, 2)

    first_run = True

    while play_again == "y":
        if peg_chooser_num == 1:
            peg_chooser = player1_name
            peg_guesser = player2_name
        else:
            peg_chooser = player2_name
            peg_guesser = player1_name
        if first_run:
            first_run = False
            print(
                f"\n{peg_chooser} will start as the peg selector, and {peg_guesser} will start as the guesser."
            )
        print(
            f"\n{peg_chooser} will select {pegs_num} coloured pegs from the following colours:\n{colours}\n"
            f"\nBut {peg_guesser} will only have {max_guesses} guesses."
            f"\nIf {peg_guesser} can't guess the pegs in {max_guesses} guesses, then {peg_chooser} wins!\n\n"
            "Ready? Then let's go...\n"
        )

        guess_num = 0
        guesses = 0

        successful = False
        while not successful:
            solution_pegs = (
                input(
                    f"{peg_guesser}, look away.\nOk, {peg_chooser} - Choose the pegs (screen will be cleared after pegs are successfully chosen)\n> "
                )
                .strip()
                .upper()
            )
            successful = True
            if solution_pegs == "" or len(solution_pegs) != pegs_num:
                successful = False
            for char in solution_pegs:
                if char not in pegs_list:
                    successful = False
            if not successful:
                example = "".join(random.choices(pegs_list, k=int(pegs_num)))
                print(
                    TYELLOW
                    + f"Please only use the characters '{pegs}' (Not Case Sensitive)\n"
                    + TNOEFFECT
                    + f"Example: {example} ({pegs_num} Pegs Needed)\n"
                )

        solution = "".join(solution_pegs)
        os.system("cls||clear")

        print(
            f"\nOK, {peg_chooser} has chosen the pegs. {peg_guesser} - What's your guess?",
            end="",
        )

        while guesses < max_guesses:
            guess_num += 1

            successful = False
            while not successful:
                if guess_num == max_guesses:
                    guess = input(f"\n{guess_num} - Last Guess! > ").upper().strip()
                else:
                    guess = input(f"\n{guess_num}> ").upper().strip()
                successful = True
                if guess == "" or len(guess) != pegs_num:
                    successful = False
                for char in guess:
                    if char not in pegs_list:
                        successful = False
                if not successful:
                    example = "".join(random.choices(pegs_list, k=int(pegs_num)))
                    print(
                        TYELLOW
                        + f"Please only use the characters '{pegs}' (Not Case Sensitive)\n"
                        + TNOEFFECT
                        + f"Example guess: {example} ({pegs_num} Pegs Needed)"
                    )

            guesses += 1
            results = [guess + " ="]

            temp_guess = split(guess)
            temp_solution = split(solution)

            for x in range(pegs_num):
                if temp_solution[x] == temp_guess[x]:
                    results.append(" " + TBLACKBG + "  " + TNOEFFECT)
                    temp_guess[x] = "*"
                    temp_solution[x] = "#"

            for x in range(pegs_num):
                if (temp_guess[x] in temp_solution) and (
                    temp_guess[x] not in temp_solution[x]
                ):
                    results.append(" " + TWHITEBG + "  " + TNOEFFECT)
                    solution_x = temp_solution.index(temp_guess[x])
                    temp_solution[solution_x] = "#"
                    temp_guess[x] = "*"

            print(*results, sep="")

            if guess == solution:
                if guesses == 1:
                    print(TGREEN + "\nOutstanding! You got it first try!" + TNOEFFECT)
                else:
                    print(
                        TGREEN
                        + f"\nWell done! You got it. It took you {guesses} guesses."
                        + TNOEFFECT
                    )
                if peg_chooser_num == 1:
                    player2_wins += 1
                else:
                    player1_wins += 1
                break

            if guesses == max_guesses and guess != solution:
                print(TORANGE + f"You lost! The solution was {solution}" + TNOEFFECT)
                if peg_chooser_num == 1:
                    player1_wins += 1
                else:
                    player2_wins += 1

        print(
            f"\n{player1_name} Wins: {player1_wins}\n{player2_name} Wins: {player2_wins}\n"
        )
        play_again = input("Play Again? (Y or N)\n> ").lower().strip()
        while play_again not in ("y", "n"):
            play_again = (
                input("Please enter Y or N\n\nPlay Again? (Y or N)\n> ").lower().strip()
            )

        peg_chooser_num = -peg_chooser_num + 3

else:
    while play_again == "y":
        print(
            f"\nI will select {pegs_num} coloured pegs from the following colours:\n{colours}\n"
            f"\nBut you only have {max_guesses} guesses."
            f"\nIf you can't guess the pegs in {max_guesses} guesses, then I win!\n\n"
            "Ready? Then let's go..."
        )

        guess_num = 0
        guesses = 0

        solution_pegs = random.choices(pegs, k=int(pegs_num))
        solution = "".join(solution_pegs)

        print("\nOK, I've chosen my colours. What's your guess?", end="")

        while guesses < max_guesses:
            guess_num += 1

            successful = False
            while not successful:
                if guess_num == max_guesses:
                    guess = input(f"\n{guess_num} - Last Guess! > ").upper().strip()
                else:
                    guess = input(f"\n{guess_num}> ").upper().strip()
                successful = True
                if guess == "" or len(guess) != pegs_num:
                    successful = False
                for char in guess:
                    if char not in pegs_list:
                        successful = False
                if not successful:
                    example = "".join(random.choices(pegs_list, k=int(pegs_num)))
                    print(
                        TYELLOW
                        + f"Please only use the characters '{pegs}' (Not Case Sensitive)\n"
                        + TNOEFFECT
                        + f"Example guess: {example} ({pegs_num} Pegs Needed)"
                    )

            guesses += 1
            results = [guess + " ="]

            temp_guess = split(guess)
            temp_solution = split(solution)

            for x in range(pegs_num):
                if temp_solution[x] == temp_guess[x]:
                    results.append(" " + TBLACKBG + "  " + TNOEFFECT)
                    temp_guess[x] = "*"
                    temp_solution[x] = "#"

            for x in range(pegs_num):
                if (temp_guess[x] in temp_solution) and (
                    temp_guess[x] not in temp_solution[x]
                ):
                    results.append(" " + TWHITEBG + "  " + TNOEFFECT)
                    solution_x = temp_solution.index(temp_guess[x])
                    temp_solution[solution_x] = "#"
                    temp_guess[x] = "*"

            print(*results, sep="")

            if guess == solution:
                if guesses == 1:
                    print(TGREEN + "\nOutstanding! You got it first try!" + TNOEFFECT)
                else:
                    print(
                        TGREEN
                        + f"\nWell done! You got it. It took you {guesses} guesses."
                        + TNOEFFECT
                    )
                player_wins += 1
                break

            if guesses == max_guesses and guess != solution:
                print(TORANGE + f"You lost! The solution was {solution}" + TNOEFFECT)
                bot_wins += 1

        print(f"\nPlayer Wins: {player_wins}\nBot Wins: {bot_wins}\n")
        play_again = input("Play Again? (Y or N)\n> ").lower().strip()
        while play_again not in ("y", "n"):
            play_again = (
                input("Please enter Y or N\n\nPlay Again? (Y or N)\n> ").lower().strip()
            )

if game_mode == "1":
    if player1_wins == 1:
        p1_win_wins = "win"
    else:
        p1_win_wins = "wins"
    if player2_wins == 1:
        p2_win_wins = "win"
    else:
        p2_win_wins = "wins"

    if player1_wins == player2_wins:
        print(f"\nWell done! You drew, with {player1_wins} wins each!")
    elif player1_wins > player2_wins:
        print(
            f"\nWell done! {player1_name} won, having {player1_wins} {p1_win_wins} compared to {player2_name}'s {player2_wins} {p2_win_wins}."
        )
    else:
        print(
            f"\nWell done! {player2_name} won, having {player2_wins} {p2_win_wins} compared to {player1_name}'s {player1_wins} {p1_win_wins}."
        )
else:
    if player_wins == 1:
        p_win_wins = "win"
    else:
        p_win_wins = "wins"
    if bot_wins == 1:
        bot_win_wins = "win"
    else:
        bot_win_wins = "wins"

    if player_wins == bot_wins:
        print(
            f"\nWell done! You drew with the bot, both getting {player_wins} {p_win_wins} each!"
        )
    elif player_wins > bot_wins:
        print(
            f"\nCongratulations! You won, having {player_wins} {p_win_wins} compared to bots {bot_wins} {bot_win_wins}."
        )
    else:
        print(
            f"\nBetter luck next time! You lost, only winning {player_wins} games compared to the bots {bot_wins} {bot_win_wins}."
        )

input("\nPlease enter to close")