Spaces:
Building
Building
import discord | |
from discord import app_commands | |
import aiohttp | |
import asyncio | |
from datetime import datetime, timezone | |
user_cash = {} | |
def load_database(): | |
global user_cash | |
try: | |
with open("database.txt", "r") as f: | |
for line in f: | |
parts = line.strip().split() | |
if len(parts) == 2 and parts[1].startswith("cash(") and parts[1].endswith(")"): | |
user_id = int(parts[0]) | |
cash = int(parts[1][5:-1]) | |
user_cash[user_id] = cash | |
except FileNotFoundError: | |
print("No database found. Creating a new one.") | |
load_database() | |
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: {game['startTime']}" | |
) 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("poor") | |
user_cash[self.user_id] -= bet_amount | |
await interaction.response.send_message(f"Bet placed on {self.team} for ${bet_amount}") | |
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()) | |
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['status']['state'] == 'FINAL': | |
winner = 'away' if game['scores']['away'] > game['scores']['home'] else 'home' | |
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}!") | |
else: | |
await interaction.user.send(f"Sorry, your team lost booo!") | |
break | |
await asyncio.sleep(300) # Check every 5 minutes | |
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.") | |
return | |
view = discord.ui.View() | |
game_select = GameSelect(upcoming_games) | |
view.add_item(game_select) | |
await interaction.response.send_message("select a game:", 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 | |
# Add this command to your command tree | |
tree.add_command(sportbet) |