BaseballBot/GhostBallBot/game.py

138 lines
3.9 KiB
Python
Raw Normal View History

2022-09-24 02:41:03 +00:00
#!/usr/bin/env python3
# Copyright 2022 - c0de <c0de@c0de.dev>
# Licensed under the MIT License (https://opensource.org/licenses/MIT)
import uuid
import dateparser
2022-10-04 01:08:27 +00:00
from database.models import database, GameModel, GuessModel
2022-09-24 02:41:03 +00:00
class Game:
"""
The game state class
This represents a game that exists in a channel
"""
def __init__(self):
# Only one game should run at at time
self.is_running = False
self.commands = {
'ghostball': self.start,
'resolve': self.stop,
'guess': self.guess,
'points': self.points,
'help': self.help,
}
2022-10-04 00:47:55 +00:00
self.game = GameModel
2022-09-24 02:41:03 +00:00
# Discord message
self.message = None
# Discord client instance
self.discord = None
async def start(self):
if self.is_running:
return await self.message.channel.send("A game is already running")
database.connect()
self.is_running = True
# game.pitch_value is unknown at the start of the game
self.game = GameModel.create(
game_id = uuid.uuid4(),
server_id = self.message.guild.id
)
database.close()
await self.message.send("@flappy ball, pitch is in! Send me your guesses with !guess <number>")
def __stopArgs__(self):
pieces = self.message.content.split()
if len(pieces) == 2:
return pieces[1], False, None, None
elif len(pieces) == 4:
return pieces[1], True, pieces[2], pieces[3]
return None, False, None, None
async def stop(self):
if not self.is_running:
return await self.message.channel.send("There is no game running to resolve")
# Determine arguments
pitch_value, has_batter, batter_id, batter_guess = self.__stopArgs__()
if not pitch_value:
return await self.message.channel.send(f"Invalid command <@{ str(self.message.author.id) }>!")
database.connect()
if has_batter:
player_id = batter_id[3:]
GuessModel.create(
game_id = self.game.game_id,
player_id = player_id,
player_name = self.discord.get_user(int(player_id).name),
guess = int(batter_guess)
)
# Save the pitch value
self.game.update({'pitch_value': pitch_value})
# TODO: Determine differences
await self.message.channel.send("Difference calculation is not currently available")
2022-09-24 02:41:03 +00:00
# stop and discard game
self.is_running = False
self.game = None
database.close()
async def guess(self):
2022-10-04 00:33:15 +00:00
if not self.is_running:
return await self.message.channel.send("There is no game running to add guesses to")
value = int(self.message.content.split()[1])
if value < 1 or value > 1000:
return await self.message.channel.send(f"Invalid value. It must be between 1 and 1000 inclusive")
2022-09-24 02:41:03 +00:00
database.connect()
2022-10-04 00:33:15 +00:00
GuessModel.create(
game_id = self.game.game_id,
player_id = self.message.author.id,
player_name = self.message.author.name,
guess = value
)
2022-09-24 02:41:03 +00:00
database.close()
2022-10-04 00:33:15 +00:00
return await self.message.add_reaction(emoji="\N{THUMBS UP SIGN}")
2022-09-24 02:41:03 +00:00
async def points(self):
# database.connect()
2022-09-24 02:41:03 +00:00
value = self.message.content.split()
try:
if len(value) > 1:
timestamp = dateparser.parse(value[1])
except:
return await self.message.channel.send("Invalid timestamp. Try again")
# database.close()
return await self.message.channel.send("Sorry, not implemented yet")
2022-09-24 02:41:03 +00:00
async def help(self):
# TODO: Add help message
help_message = "help"
recipient = await self.discord.fetch_user(self.message.author.id)
await recipient.send(help_message)