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 TeamSelect(discord.ui.Select): | |
def __init__(self, teams): | |
options = [discord.SelectOption(label=f"{team['teamName']} ({team['abbreviation']})", value=team['abbreviation']) for team in teams] | |
super().__init__(placeholder="Select a team", 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 | |
embed = discord.Embed(title="Upcoming NHL Games", color=0x00ff00) | |
for game in upcoming_games: | |
away_team = game['teams']['away'] | |
home_team = game['teams']['home'] | |
embed.add_field(name=f"{away_team['teamName']} vs {home_team['teamName']}", | |
value=f"Start time: {game['startTime']}", inline=False) | |
view = discord.ui.View() | |
team_select = TeamSelect(sum([game['teams'].values() for game in upcoming_games], [])) | |
view.add_item(team_select) | |
await interaction.response.send_message(embed=embed, view=view) | |
async def team_callback(interaction: discord.Interaction): | |
selected_team = team_select.values[0] | |
game = next(game for game in upcoming_games if selected_team in [game['teams']['away']['abbreviation'], game['teams']['home']['abbreviation']]) | |
await interaction.response.send_modal(BetModal(selected_team, interaction.user.id, game)) | |
team_select.callback = team_callback |