import discord from discord import app_commands import aiohttp import asyncio from datetime import datetime, timezone from cash import user_cash user_bets = {} async def fetch_nhl_scores(): async with aiohttp.ClientSession() as session: async with session.get("https://nhl-score-api.herokuapp.com/api/scores/latest") as response: return await response.json() 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: " ) for game in games ] super().__init__(placeholder="Select a game", options=options) class TeamSelect(discord.ui.Select): def __init__(self, away_team, home_team): options = [ discord.SelectOption(label=away_team['teamName'], value=away_team['abbreviation']), discord.SelectOption(label=home_team['teamName'], value=home_team['abbreviation']) ] super().__init__(placeholder="Select a team to bet on", options=options) class BetModal(discord.ui.Modal, title="Place Your Bet"): bet_amount = discord.ui.TextInput(label="Bet Amount", placeholder="Enter bet amount") def __init__(self, team, user_id, game_data): super().__init__() self.team = team self.user_id = user_id self.game_data = game_data async def on_submit(self, interaction: discord.Interaction): try: bet_amount = int(self.bet_amount.value) if bet_amount <= 0: raise ValueError("Bet more than 0 dollars") if bet_amount > user_cash.get(self.user_id, 0): raise ValueError("You are too poor. Check your balance and try again") user_cash[self.user_id] -= bet_amount await interaction.response.send_message(f"Bet placed on {self.team} for ${bet_amount}") user = await interaction.client.fetch_user(self.user_id) embed = discord.Embed(title="Bet Placed", color=0x787878) 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=f"", inline=False) await user.send(embed=embed) if self.user_id not in user_bets: user_bets[self.user_id] = [] user_bets[self.user_id].append({ "team": self.team, "amount": bet_amount, "game_data": self.game_data }) asyncio.create_task(self.monitor_game(interaction, bet_amount)) except ValueError as e: await interaction.response.send_message(str(e), ephemeral=True) async def monitor_game(self, interaction, bet_amount): game_start = datetime.fromisoformat(self.game_data['startTime'].replace('Z', '+00:00')) await asyncio.sleep((game_start - datetime.now(timezone.utc)).total_seconds()) user = await interaction.client.fetch_user(self.user_id) await user.send(f"Your team {self.team} has started playing!") previous_scores = {self.game_data['teams']['away']['abbreviation']: 0, self.game_data['teams']['home']['abbreviation']: 0} while True: scores = await fetch_nhl_scores() game = next((g for g in scores['games'] if g['teams']['away']['abbreviation'] == self.game_data['teams']['away']['abbreviation'] and g['teams']['home']['abbreviation'] == self.game_data['teams']['home']['abbreviation']), None) if game is None or game['status']['state'] == 'PREVIEW': await asyncio.sleep(60) continue current_scores = game['scores'] for team in [self.game_data['teams']['away']['abbreviation'], self.game_data['teams']['home']['abbreviation']]: if current_scores[team] > previous_scores[team]: team_name = game['teams']['away']['teamName'] if team == game['teams']['away']['abbreviation'] else game['teams']['home']['teamName'] await user.send(f"{team_name} SCORED! Current score: {current_scores[self.game_data['teams']['away']['abbreviation']]} - {current_scores[self.game_data['teams']['home']['abbreviation']]}") previous_scores = current_scores.copy() if game['status']['state'] == 'FINAL': winner = self.game_data['teams']['away']['abbreviation'] if current_scores[self.game_data['teams']['away']['abbreviation']] > current_scores[self.game_data['teams']['home']['abbreviation']] else self.game_data['teams']['home']['abbreviation'] if winner == self.team: winnings = bet_amount * 2 user_cash[self.user_id] += winnings await user.send(f"Congratulations! Your team won. You won ${winnings}!") else: await 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(60) # Check for updates every minute class SportBetView(discord.ui.View): def __init__(self): super().__init__() @discord.ui.button(label="Bet Again", style=discord.ButtonStyle.primary) async def bet_again(self, interaction: discord.Interaction, button: discord.ui.Button): await show_game_selection(interaction) @discord.ui.button(label="View Bets", style=discord.ButtonStyle.secondary) async def view_bets(self, interaction: discord.Interaction, button: discord.ui.Button): await show_current_bets(interaction) async def show_game_selection(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 available for betting.") return view = discord.ui.View() game_select = GameSelect(upcoming_games) view.add_item(game_select) 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]) team_view = discord.ui.View() team_select = TeamSelect(selected_game['teams']['away'], selected_game['teams']['home']) team_view.add_item(team_select) 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] await interaction.response.send_modal(BetModal(selected_team, interaction.user.id, selected_game)) team_select.callback = team_callback game_select.callback = game_callback async def show_current_bets(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 active bets.") return embed = discord.Embed(title="Your Current Bets", color=0x787878) scores = await fetch_nhl_scores() for i, bet in enumerate(user_bets[user_id]): game = next((g for g in scores['games'] if g['teams']['away']['abbreviation'] == bet['game_data']['teams']['away']['abbreviation'] and g['teams']['home']['abbreviation'] == bet['game_data']['teams']['home']['abbreviation']), None) start_time = datetime.fromisoformat(bet['game_data']['startTime'].replace('Z', '+00:00')) current_time = datetime.now(timezone.utc) if current_time < start_time: status = f"Starts " score = "N/A" else: status = game['status']['state'] if game else "Unknown" score = f"{game['scores'][bet['game_data']['teams']['away']['abbreviation']]} - {game['scores'][bet['game_data']['teams']['home']['abbreviation']]}" if game and 'scores' in game else "Unknown" embed.add_field(name=f"Bet {i+1}", value=( f"Team: {bet['team']}\n" f"Amount: ${bet['amount']}\n" f"Game: {bet['game_data']['teams']['away']['teamName']} vs {bet['game_data']['teams']['home']['teamName']}\n" f"Status: {status}\n" f"Current Score: {score}\n" f"Start Time: " ), inline=False) view = discord.ui.View() cancel_select = discord.ui.Select(placeholder="Select a bet to cancel", options=[ discord.SelectOption(label=f"Bet {i+1}", value=str(i)) for i in range(len(user_bets[user_id])) ]) view.add_item(cancel_select) async def cancel_callback(interaction: discord.Interaction): if interaction.user.id != user_id: await interaction.response.send_message("You cannot cancel other users' bets.", ephemeral=True) return bet_index = int(cancel_select.values[0]) cancelled_bet = user_bets[user_id].pop(bet_index) user_cash[user_id] += cancelled_bet['amount'] 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) @app_commands.command(name="sportbet", description="Bet on sports games") async def sportbet(interaction: discord.Interaction): user_id = interaction.user.id if user_id in user_bets and user_bets[user_id]: view = SportBetView() await interaction.response.send_message("?", view=view) else: await show_game_selection(interaction)