Slot machine

slot-machine/README.md

@@ -0,0 +1,11 @@
+## What it does
+
+A fully functional terminal-based slot machine game. The player deposits
+money, chooses how many lines to bet on (up to 3), sets a bet amount per
+line, and spins. The machine generates a random 3x3 grid of symbols and
+pays out based on matching symbols across each line. The game tracks
+balance across multiple spins until the player quits.
+
+## Requirements
+
+None — uses only built-in random module

slot-machine/slot_machine.py

@@ -0,0 +1,162 @@
+import random
+
+MAX_LINES = 3
+MAX_BET = 10
+MIN_BET = 1
+
+ROWS = 3
+COLS = 3
+
+symbol_count = {
+    "A": 2,
+    "B": 4,
+    "C": 6,
+    "D": 8
+}
+
+symbol_value = {
+    "A": 5,
+    "B": 4,
+    "C": 3,
+    "D": 2
+}
+
+
+def deposit():
+    while True:
+        amount = input("What would you like to deposit? $")
+
+        if amount.isdigit():
+            amount = int(amount)
+
+            if amount > 0:
+                return amount
+
+        print("Please enter a valid amount.")
+
+
+def get_number_of_lines():
+    while True:
+        lines = input(f"Enter the number of lines to bet on (1-{MAX_LINES}): ")
+
+        if lines.isdigit():
+            lines = int(lines)
+
+            if 1 <= lines <= MAX_LINES:
+                return lines
+
+        print("Enter a valid number.")
+
+
+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:
+                return amount
+
+        print(f"Amount must be between ${MIN_BET} and ${MAX_BET}.")
+
+
+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):
+        current_symbols = all_symbols[:]
+        column = []
+
+        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 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 != symbol_to_check:
+                break
+        else:
+            winnings += values[symbol] * bet
+            winning_lines.append(line + 1)
+
+    return winnings, winning_lines
+
+
+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 balance. Current balance: ${balance}")
+        else:
+            break
+
+    print(f"\nYou are betting ${bet} on {lines} lines.")
+    print(f"Total bet: ${total_bet}\n")
+
+    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"\nYou won ${winnings}.")
+    print(f"Winning lines: {winning_lines}")
+
+    return winnings - total_bet
+
+
+def main():
+    balance = deposit()
+
+    while True:
+        print(f"\nCurrent balance: ${balance}")
+
+        answer = input("Press Enter to spin (q to quit): ")
+
+        if answer.lower() == "q":
+            break
+
+        balance += spin(balance)
+
+    print(f"\nYou left with ${balance}")
+
+
+main()