Spaces:
Building
Building
| 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: <t:{int(datetime.fromisoformat(game['startTime'].replace('Z', '+00:00')).timestamp())}:R>" | |
| ) 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"<t:{int(datetime.fromisoformat(self.game_data['startTime'].replace('Z', '+00:00')).timestamp())}:R>", 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__() | |
| async def bet_again(self, interaction: discord.Interaction, button: discord.ui.Button): | |
| await show_game_selection(interaction) | |
| 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 <t:{int(start_time.timestamp())}:R>" | |
| 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: <t:{int(start_time.timestamp())}:F>" | |
| ), 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) | |
| 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) | |