project(slot).py

import random


MAX_LINES = 3  # Capital letter denotes the constant
MAX_BET = 100
MIN_BET = 1

ROWS = 3
COLS = 3

symbol_count = { # Making a dictionary of symbols
  "A" : 2,
  "B" : 4,
  "C" : 6,
  "D" : 8
}

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

def check_winnings(columns, lines, bet, values):
    winnings = 0
    winning_lines = []
    max_rows = len(columns[0])  

    lines = min(lines, max_rows)  
    
    for line in range(lines):
        symbol = columns[0][line]
        is_winning = True
        for column in columns:
            if column[line] != symbol:
                is_winning = False
                break
        if is_winning:
            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():
        all_symbols.extend([symbol] * symbol_count)  
    
    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): # in order to print after spinning the slot machine
  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():#in order to deposit the amount before playing
  while True:
    amount = input("What would you like to deposit ? $")
    if amount.isdigit():
      amount = int(amount)
      if amount > 0:
        break
      else:
        print("Amount must be greater than zero.")
    else:
      print("Please enter a number.")
  return amount


def get_number_of_lines():# to define the no of lines used in slot machine 
  while True:
    lines = input("Enter the 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 the valid number of lines .")
    else:
      print("Please enter a number.")
  return lines


def get_bet(): # to bet in the slot machine in each line 
  while True :
    bet = input("What would like to bet on each line ? $")
    if bet.isdigit():
      bet = int(bet)
      if MIN_BET <= bet <= MAX_BET:
        break
      else:
        print(f"Bet must be between ${MIN_BET} - ${MAX_BET}.")
    else:
      print("Please enter a number. ")
  return bet


def spin(balance):
  lines = get_number_of_lines()
  while True:
    bet = get_bet()
    total_bet = lines * bet


    if total_bet > balance:
      print(f"You don't have enough to bet the amount, your current balance is ${balance} ")
    else:
      break


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

  slots = get_slot_machine_spin(ROWS, COLS, symbol_count)
  print_slot_machine(slots)
  winnings, winning_lines = check_winnings(slots, lines, bet, symbol_value)
  print(f"Your winnings are ${winnings}.")
  print(f"Your won on lines:", *winning_lines)
  return winnings - total_bet



def main():
  balance = deposit()
  while True:
    print(f"current balance is ${balance}")
    answer = input("press enter to spin (q to quit).") 
    if answer == 'q':
      break
    balance += spin(balance)
  print(f"You left with ${balance}")




main()