Update sportbet.py

sportbet.py

@@ -2,11 +2,16 @@ import discord
 from discord import app_commands
 import aiohttp
 import asyncio
-from datetime import datetime, timezone
+from datetime import datetime, timezone, timedelta
 
 user_cash = {}
 user_bets = {}
 
+def save_database():
+    with open("database.txt", "w") as f:
+        for user_id, cash in user_cash.items():
+            f.write(f"{user_id} cash({cash})\n")
+
 def load_database():
     global user_cash
     try:
@@ -27,13 +32,27 @@ async def fetch_nhl_scores():
         async with session.get("https://nhl-score-api.herokuapp.com/api/scores/latest") as response:
             return await response.json()
 
+def format_time_difference(start_time):
+    now = datetime.now(timezone.utc)
+    start = datetime.fromisoformat(start_time.replace('Z', '+00:00'))
+    diff = start - now
+    
+    if diff < timedelta(0):
+        return "Game has already started"
+    elif diff < timedelta(hours=1):
+        return f"{diff.seconds // 60} minutes from now"
+    elif diff < timedelta(days=1):
+        return f"{diff.seconds // 3600} hours from now"
+    else:
+        return f"{diff.days} days from now"
+
 class GameSelect(discord.ui.Select):
     def __init__(self, games):
         options = [
             discord.SelectOption(
                 label=f"{game['teams']['away']['teamName']} vs {game['teams']['home']['teamName']}",
                 value=f"{game['teams']['away']['abbreviation']}_{game['teams']['home']['abbreviation']}",
-                description=f"Start time: {game['startTime']}"
+                description=f"Start: {format_time_difference(game['startTime'])}"
             ) for game in games
         ]
         super().__init__(placeholder="Select a game", options=options)
@@ -61,9 +80,10 @@ class BetModal(discord.ui.Modal, title="Place Your Bet"):
             if bet_amount <= 0:
                 raise ValueError("Bet more than 0 dollars")
             if bet_amount > user_cash.get(self.user_id, 0):
-                raise ValueError("poor")
+                raise ValueError("Insufficient funds")
 
             user_cash[self.user_id] -= bet_amount
+            save_database()
             await interaction.response.send_message(f"Bet placed on {self.team} for ${bet_amount}")
             
             user = await interaction.client.fetch_user(self.user_id)
@@ -71,7 +91,7 @@ class BetModal(discord.ui.Modal, title="Place Your Bet"):
             embed.add_field(name="Team", value=self.team, inline=False)
             embed.add_field(name="Amount", value=f"${bet_amount}", inline=False)
             embed.add_field(name="Game", value=f"{self.game_data['teams']['away']['teamName']} vs {self.game_data['teams']['home']['teamName']}", inline=False)
-            embed.add_field(name="Start Time", value=self.game_data['startTime'], inline=False)
+            embed.add_field(name="Start Time", value=format_time_difference(self.game_data['startTime']), inline=False)
             await user.send(embed=embed)
 
             if self.user_id not in user_bets:
@@ -100,29 +120,30 @@ class BetModal(discord.ui.Modal, title="Place Your Bet"):
                 if game['teams'][winner]['abbreviation'] == self.team:
                     winnings = bet_amount * 2
                     user_cash[self.user_id] += winnings
-                    await interaction.user.send(f"WOO YOUR TEAM WON you won ${winnings}!")
+                    save_database()
+                    await interaction.user.send(f"Congratulations! Your team won. You won ${winnings}!")
                 else:
-                    await interaction.user.send(f"Sorry, your team lost booo!")
+                    await interaction.user.send(f"Sorry, your team lost. Better luck next time!")
                 
                 user_bets[self.user_id] = [bet for bet in user_bets[self.user_id] if bet['game_data'] != self.game_data]
                 break
             
             await asyncio.sleep(300)
 
-@app_commands.command(name="sportbet", description="bet on sports game")
+@app_commands.command(name="sportbet", description="Bet on sports games")
 async def sportbet(interaction: discord.Interaction):
     scores = await fetch_nhl_scores()
     upcoming_games = [game for game in scores['games'] if game['status']['state'] == 'PREVIEW']
 
     if not upcoming_games:
-        await interaction.response.send_message("No games for betting.")
+        await interaction.response.send_message("No games available for betting.")
         return
 
     view = discord.ui.View()
     game_select = GameSelect(upcoming_games)
     view.add_item(game_select)
 
-    await interaction.response.send_message("game to bet on:", view=view)
+    await interaction.response.send_message("Select a game to bet on:", view=view)
 
     async def game_callback(interaction: discord.Interaction):
         selected_game = next(game for game in upcoming_games if f"{game['teams']['away']['abbreviation']}_{game['teams']['home']['abbreviation']}" == game_select.values[0])
@@ -131,7 +152,7 @@ async def sportbet(interaction: discord.Interaction):
         team_select = TeamSelect(selected_game['teams']['away'], selected_game['teams']['home'])
         team_view.add_item(team_select)
 
-        await interaction.response.edit_message(content="team to bet on:", view=team_view)
+        await interaction.response.edit_message(content="Select a team to bet on:", view=team_view)
 
         async def team_callback(interaction: discord.Interaction):
             selected_team = team_select.values[0]
@@ -141,11 +162,11 @@ async def sportbet(interaction: discord.Interaction):
 
     game_select.callback = game_callback
 
-@app_commands.command(name="currentbets", description="view your bets")
+@app_commands.command(name="currentbets", description="View your current bets")
 async def currentbets(interaction: discord.Interaction):
     user_id = interaction.user.id
     if user_id not in user_bets or not user_bets[user_id]:
-        await interaction.response.send_message("You have no bets.")
+        await interaction.response.send_message("You have no active bets.")
         return
 
     embed = discord.Embed(title="Your Current Bets", color=0x787878)
@@ -166,8 +187,27 @@ async def currentbets(interaction: discord.Interaction):
         bet_index = int(cancel_select.values[0])
         cancelled_bet = user_bets[user_id].pop(bet_index)
         user_cash[user_id] += cancelled_bet['amount']
+        save_database()
         await interaction.response.send_message(f"Bet cancelled. ${cancelled_bet['amount']} has been refunded.")
 
     cancel_select.callback = cancel_callback
 
-    await interaction.response.send_message(embed=embed, view=view)
+    await interaction.response.send_message(embed=embed, view=view)
+
+@app_commands.command(name="cash", description="Check your cash balance")
+async def cash(interaction: discord.Interaction):
+    user_id = interaction.user.id
+    balance = user_cash.get(user_id, 0)
+
+    if balance == 0:
+        user_cash[user_id] = 1000
+        balance = 1000
+        message = "You have no cash. Here's $1,000 to start!"
+    else:
+        message = f"Your current balance is ${balance:,}"
+
+    embed = discord.Embed(title="Cash Balance", description=message, color=0x787878)
+    embed.set_footer(text="Use /sportbet to bet your cash!")
+
+    await interaction.response.send_message(embed=embed)
+    save_database()