gambling.py

import random

MAX_LINES = 3
MAX_BET = 100
MIN_BET = 1

ROWS = 3
COLUMNS = 3

symbol_count = {
    "A": 2,
    "B": 4,
    "C": 6,
    "D": 8,
} 

symbol_value = {
    "A": 5,
    "B": 4,
    "C": 3,
    "D": 2,
} 

def check_winning(columns, lines, bet, values):
    winning = 0
    winning_lines = []
    for line in range(lines):
        symbol = columns[0][line]
        for column in columns:
            symbols_to_check = column[line]
            if symbol != symbols_to_check:
                break
        else:
            winning += values[symbol] * bet
            winning_lines.append(line)

    return winning, winning_lines

def get_slot_machine_spin(rows, cols, symbols):
    all_symbols = []
    for symbol, symbol_count in symbols.items():
        for _ in range(symbol_count):
            all_symbols.append(symbol)

    columns = []
    for _ in range(cols):
        column = []
        current_symbols = all_symbols[:]
    
        for _ in range(rows):
            value = random.choice(current_symbols)
            current_symbols.remove(value)
            column.append(value)
    
        columns.append(column)
    
    return columns

def print_slot_machine(columns):
    for row in range(len(columns[0])):
        for i, column in enumerate(columns):
            if i != len(columns) -1 :
                print(column[row], end= " | ")
            else:
                print(column[row], end="")
        print()

def deposit():
    while True:
        amount = input("What would you like to deposit? $")
        if amount.isdigit():
            amount = int(amount)
            if amount > 0:
                break
            else:
                print("You can't deposit negative amount!")
        else:
            print("Please enter a amount")
        
    return amount

def get_number_of_lines():
    while True:
        lines = input("Enter a number of lines to bet on (1-" + str(MAX_LINES) +  ")? ")
        if lines.isdigit():
            lines = int(lines)
            if 1 <= lines <= MAX_LINES:
                break
            else:
                print("Enter a valid number of lines")
        else:
            print("Please enter a number")
        
    return lines

def get_bet():
    while True:
        amount = input("What would you like to bet on each lines? $")
        if amount.isdigit():
            amount = int(amount)
            if MIN_BET <= amount <= MAX_BET:
                break
            else:
                print(f"Amount between ${MIN_BET} - ${MAX_BET}.")
        else:
            print("Please enter a amount")
        
    return amount

def spin(balance):
    lines = get_number_of_lines()

    while True:
        bet = get_bet()
        total_bet = bet * lines
        if total_bet > balance:
            print(f"You don't have enough to bet that amount, your current balance is ${balance}!")     
        else:
            break

    print(f"You are betting ${bet} on {lines} lines. And Total bet is equal to ${total_bet}.")

    slots = get_slot_machine_spin(ROWS, COLUMNS, symbol_count)
    print_slot_machine(slots)
    winnings, winnings_lines = check_winning(slots, lines, bet, symbol_value)

    print(f" You won ${winnings} ")
    print(" You won on lines: ", *winnings_lines)

    return winnings - total_bet

def main():
    balance = deposit()
    
    while True:
        print(f"Current Balance: ${balance}")
        answer = input("Press enter to play (q to quite)")
        
        if answer == 'q':
            break

        balance += spin(balance)

    print(f"You left with ${balance} ")

main()