botbotbotbot / sportbet.py
coollsd's picture
Update sportbet.py
82c48f2 verified
raw
history blame
5.4 kB
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
@app_commands.command(name="sportbet", description="Bet on NHL 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.")
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)