newproject.py

import random

MAX_LINES =3
MAX_BET = 100
MIN_BET = 1

ROWS = 3
COLS =3

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

symbol_values ={
    "A": 5,
    "B": 4,
    "C": 3,
    "D": 6
}

def check_winnings(columns, lines, bet, values):
    winnings = 0
    winning_lines =[ ]
    for line in range(lines):
        symbol = columns[0][line]
        for column in columns:
            symbol_to_check = column[line]
            if symbol in symbol_to_check:
                break
        else:
            winnings += values[symbol] * bet
            winning_lines.append(line + 1)
    
    return winnings, 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('enter amount you wish to deposit $')
        if amount.isdigit ():
            amount = int(amount)
            if amount > 0:
                break
            else:
                print ('enter amount greater than zero')
        else:
            print('enter a digit kindly ')
    return amount

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


def get_bet():
    while True:
        amount = input('what would you like to bet on each line $')
        if amount.isdigit ():
            amount = int(amount)
            if MIN_BET <=amount <=MAX_BET:
                break
            else:
                print (f'amount must be between ${MIN_BET} - ${MAX_BET}' )
        else:
            print('enter a digit kindly ')
    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 do not have enough money to bet on, your current balance is ${balance} ")
        else:
            break
    print(f"your betting ${bet} on {lines} lines. Total bet equals: ${total_bet}")
    
    slots = get_slot_machine_spin(ROWS, COLS, symbol_count)
    print_slot_machine(slots)
    winnings, winnings_lines = check_winnings(slots, lines, bet, symbol_values)
    print(f"you won ${winnings}")
    print(f"you won on lines:", *winnings_lines)
    return (winnings) - (total_bet)
    
def main ():
    balance = deposit()
    while True:
        print(f"your current balance is ${balance}")
        answer = input("press enter to play (q to quit). ")
        if answer == "q":
            break
        balance += spin(balance )
    print(f"your left with ${balance}")

main ()