Merge pull request #28 from Kokiix/main

feat: Slots game

Author: NeedToLookAtMyPictures

casino/games/__init__.py

@@ -1,3 +1,3 @@
-from . import blackjack
+from . import blackjack, slots
 
-__all__ = ["blackjack"]
+__all__ = ["blackjack", "slots"]

casino/games/slots/__init__.py

@@ -0,0 +1,3 @@
+from .slots import play_slots
+
+__all__ = ["play_slots"]

casino/games/slots/slots.py

@@ -0,0 +1,136 @@
+import random
+import time
+
+from casino.card_assets import assign_card_art
+from casino.types import Card
+from casino.utils import clear_screen, cprint, cinput, display_topbar
+
+SLOTS_HEADER = """
+┌───────────────────────────────┐
+│         ♠ S L O T S ♠         │
+└───────────────────────────────┘
+"""
+HEADER_OPTIONS = {"header": SLOTS_HEADER,
+                  "margin": 1,}
+
+ITEMS = {"all": "abcd", "low_val": "abc", "high_val": "d"}
+BET_PROMPT = """
+How much would you like to bet?
+Matching a | b | c  : x1.5
+Matching d          : x5.0
+"""
+INVALID_BET_MSG = "That's not a valid bet."
+MIN_BET_AMT = 10
+MIN_BET_MSG = f"Each pay line requires at least ${MIN_BET_AMT}."
+
+SEC_BTWN_SPIN = 0.1
+TOTAL_SPINS = 10
+WIN_PROB = 0.2
+HI_VAL_PROB = 0.05
+
+# Currently 1 pay line, goal is to have several and:
+# - implement pattern patching for wins across lines
+# - be able to bet across lines independently
+# then maybe special lines
+
+def play_slots(account) -> None:
+    take_new_bet = True
+    bet_amount = 0
+    while True:
+        clear_screen()
+        display_topbar(account, **HEADER_OPTIONS)
+        if take_new_bet or bet_amount > account.balance:
+            bet_amount = get_bet_amount(account)
+            take_new_bet = False
+
+        spin_animation(account)
+        clear_screen()
+        display_topbar(account, **HEADER_OPTIONS)
+
+        # Display final spin result
+        rng = random.random()
+        if rng <= WIN_PROB:
+            money_gain = 0
+            if rng <= HI_VAL_PROB:
+                win_item = ITEMS["high_val"][random.randint(0, len(ITEMS["high_val"]) - 1)]
+                money_gain = bet_amount * 5
+            else:
+                win_item = ITEMS["low_val"][random.randint(0, len(ITEMS["low_val"]) - 1)]
+                money_gain = bet_amount * 1.5
+
+            account.deposit(money_gain)
+            clear_screen()
+            display_topbar(account, **HEADER_OPTIONS)
+            print_spin(win_item, win_item, win_item)
+            cprint(f"MATCH: +${money_gain}")
+        else:
+            final_items = (get_rand_item(), get_rand_item(), get_rand_item())
+            while len(set(final_items)) == 1:
+                final_items = (get_rand_item(), get_rand_item(), get_rand_item())
+            clear_screen()
+            account.withdraw(bet_amount)
+            display_topbar(account, **HEADER_OPTIONS)
+            print_spin(*final_items)
+            cprint(f"NO MATCH: -${bet_amount}")
+
+        # Choose what to do after spin
+        choice = get_player_choice()
+        if choice["quit"]:
+            return
+        if choice["change_bet"]:
+            take_new_bet = True
+
+def get_bet_amount(account):
+    if account.balance < MIN_BET_AMT:
+        raise RuntimeError("PLAYER OUT OF MONEY IDK WHAT TO DO")
+
+    while True:
+        bet_str = cinput(BET_PROMPT).strip()
+        try:
+            bet = int(bet_str)
+            if bet < MIN_BET_AMT:
+                clear_screen()
+                display_topbar(account, **HEADER_OPTIONS)
+                cprint(MIN_BET_MSG)
+                continue
+            return bet
+        except ValueError:
+            clear_screen()
+            display_topbar(account, **HEADER_OPTIONS)
+            cprint(INVALID_BET_MSG)
+    
+def get_player_choice():
+    player_input = ""
+    while player_input not in "rRqQcC" or player_input == "":
+        player_input = cinput("[R]espin [C]hange Bet [Q]uit")
+        
+    if player_input in "qQ": 
+        clear_screen()
+        return {"quit": True}
+    elif player_input in "rR":
+        return {"quit": False, "change_bet": False}
+    else:
+        return {"quit": False, "change_bet": True}
+
+def spin_animation(account, total_spins = TOTAL_SPINS, sec_btwn_spins = SEC_BTWN_SPIN) -> None:
+    for _ in range(total_spins):
+        clear_screen()
+        display_topbar(account, **HEADER_OPTIONS)
+        print_spin(get_rand_item(), get_rand_item(), get_rand_item())
+        time.sleep(sec_btwn_spins)
+
+def get_rand_item() -> str:
+    return ITEMS["all"][random.randint(0, len(ITEMS["all"]) - 1)];
+
+def print_spin(item1, item2, item3) -> None:
+    cprint(f"""
+    ┌────────────────────────┐
+    |                        |
+    |                        |
+    |                        |
+    |{item1:^8}{item2:^8}{item3:^8}|
+    |                        |
+    |                        |
+    |                        |
+    └────────────────────────┘
+    """)

casino/main.py

@@ -4,6 +4,8 @@
 from casino.accounts import Account
 from casino.games.blackjack import play_blackjack
 from casino.utils import cprint, cinput, clear_screen, display_topbar
+from casino.games.slots import play_slots
+from casino.utils import cprint, cinput, clear_screen
 
 CASINO_HEADER = """
 ┌──────────────────────────────────────┐
@@ -24,6 +26,7 @@
 # To add a new game, just add a handler function to GAME_HANDLERS
 GAME_HANDLERS: dict[str, Callable[[Account], None]] = {
     "blackjack": play_blackjack,
+    "slots": play_slots,
 }
 games = list(GAME_HANDLERS.keys())
 

casino/main3.py

@@ -0,0 +1,23 @@
+from casino.games.blackjack import play_blackjack
+from casino.games.slots import play_slots
+from casino.utils import cprint, cinput, clear_screen
+
+CASINO_HEADER = """
+┌──────────────────────────────────────┐
+│   ♦ T E R M I N A L  C A S I N O ♦   │
+└──────────────────────────────────────┘
+"""
+
+ENTER_OR_QUIT_PROMPT = "[E]nter   [Q]uit: \n"
+INVALID_CHOICE_PROMPT = "\nInvalid input. Please try again. \n\n"
+GAME_CHOICE_PROMPT = "Please choose a game to play:\n"
+
+games = ["blackjack", "slots"]
+
+
+def main():
+    play_slots()
+
+
+if __name__ == "__main__":
+    main()