BaseballBot/GhostBallBot/game/game.py

258 lines
7.7 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)
2022-11-11 03:46:42 +00:00
# pylint: disable=no-member,not-an-iterable
"""
A Context Manager / State Machine that keeps track of
a single game instance (there should only be one) in a
Discord channel
"""
2022-09-24 02:41:03 +00:00
import uuid
2022-10-26 04:17:57 +00:00
import datetime
2022-09-24 02:41:03 +00:00
# import dateparser
2022-10-26 04:17:57 +00:00
from database.models import (
database,
GameModel as Game,
GuessModel as Guess,
PlayerModel as Player,
)
from game.guess import ProcessGuess
2022-09-24 02:41:03 +00:00
2022-10-25 01:05:21 +00:00
2022-10-26 04:17:57 +00:00
class GameManager:
2022-09-24 02:41:03 +00:00
"""
2022-10-25 01:00:39 +00:00
The game state class
2022-09-24 02:41:03 +00:00
2022-10-25 01:00:39 +00:00
This represents a game that exists in a channel
"""
def __init__(self):
2022-09-24 02:41:03 +00:00
# Only one game should run at at time
self.is_running = False
self.commands = [
("braveball", self.start),
("resolve", self.stop),
("guess", self.guess),
("points", self.points),
("help", self.help),
]
2022-09-24 02:41:03 +00:00
2022-10-26 04:17:57 +00:00
self.game = Game
2022-09-24 02:41:03 +00:00
# Discord message
self.message = None
# Discord client instance
self.discord = None
def __enter__(self):
"""
Allows use of `with Game() as game` for try/except statements
(https://peps.python.org/pep-0343/)
"""
database.connect()
return self
def __exit__(self, exception_type, exception_value, exception_traceback):
"""
2022-10-25 01:00:39 +00:00
Automagically close the database
when this class has ended execution
"""
database.close()
2022-09-24 02:41:03 +00:00
async def start(self):
"""
Start command - Starts a new game if there isn't already one running
"""
if self.is_running:
return await self.message.channel.send("A game is already running")
2022-09-24 02:41:03 +00:00
self.is_running = True
# game.pitch_value is unknown at the start of the game
self.game = Game.create(game_id=uuid.uuid4(), server_id=self.message.channel.id)
2022-09-24 02:41:03 +00:00
await self.message.channel.send("Send me your guesses with !guess <number>")
2022-09-24 02:41:03 +00:00
def __stop_args__(self):
2022-09-24 02:41:03 +00:00
pieces = self.message.content.split()
if len(pieces) == 2:
return pieces[1], False, None, None
if len(pieces) == 4:
2022-09-24 02:41:03 +00:00
return pieces[1], True, pieces[2], pieces[3]
return None, False, None, None
async def update_pitch_value(self):
2022-10-26 04:17:57 +00:00
"""Update game state database for closing arguments"""
2022-09-24 02:41:03 +00:00
# Determine arguments
pitch_value, has_batter, batter_id, batter_guess = self.__stop_args__()
2022-09-24 02:41:03 +00:00
if not pitch_value:
2022-10-25 01:00:39 +00:00
return await self.message.channel.send(
f"Invalid command <@{ str(self.message.author.id) }>!"
)
2022-09-24 02:41:03 +00:00
if has_batter:
player_id = batter_id[3:]
2022-10-26 04:17:57 +00:00
Guess.create(
2022-10-25 01:00:39 +00:00
game_id=self.game.game_id,
player_id=player_id,
player_name=self.discord.get_user(int(player_id).name),
guess=int(batter_guess),
2022-10-26 04:17:57 +00:00
).save()
2022-09-24 02:41:03 +00:00
# Save the pitch value
Game.update(
2022-10-26 04:17:57 +00:00
{
Game.pitch_value: pitch_value,
Game.date_ended: datetime.datetime.now(),
}
).where(Game.game_id == self.game.game_id).execute()
2022-10-26 04:17:57 +00:00
2022-10-27 05:18:12 +00:00
return int(pitch_value)
2022-10-26 04:17:57 +00:00
async def stop(self):
"""
Stop command - Stops the game if it is currently running,
saves the pitch value, and displays differences
"""
if not self.is_running:
return await self.message.channel.send("There is no game running")
# How many valid guesses got placed?
guess_count = (
2022-11-11 03:03:38 +00:00
Guess.select()
.join(Game)
2022-11-09 17:48:31 +00:00
.where((Guess.game.game_id == self.game.game_id) & (Guess.guess > 0))
.count()
)
2022-09-24 02:41:03 +00:00
# Discard the game if there weren't enough players
if guess_count < 3:
2022-10-26 04:17:57 +00:00
self.game = None
self.is_running = False
2022-11-11 03:46:42 +00:00
return await self.message.channel.send(
2022-10-26 04:17:57 +00:00
("Play closed!\n" + "However, there were not enough participants.")
)
message = (
"Closed this play! Here are the results\n"
+ "PLAYER | GUESS | DIFFERENCE | POINTS GAINED | TOTAL POINTS\n"
)
pitch_value = await self.update_pitch_value()
2022-11-11 03:03:38 +00:00
guess_processor = ProcessGuess(
game=self, pitch_value=pitch_value, message=message
)
2022-10-27 05:18:12 +00:00
2022-11-09 17:48:31 +00:00
(
message,
closest_player_id,
furthest_player_id,
) = guess_processor.process_guesses()
2022-10-26 04:17:57 +00:00
message += (
2022-10-27 05:18:12 +00:00
f"\nCongrats <@{closest_player_id}>! You were the closest!\n"
2022-10-26 04:17:57 +00:00
+ f"Sorry <@{furthest_player_id}>, you were way off"
)
await self.message.channel.send(message)
2022-09-24 02:41:03 +00:00
# stop and discard game
self.is_running = False
self.game = None
async def guess(self):
"""
Guess command - Allows the player to add a guess to the current
running game
"""
if not self.is_running:
return await self.message.channel.send("There is no game running")
2022-10-04 00:33:15 +00:00
value = int(self.message.content.split()[1])
if value < 1 or value > 1000:
2022-10-25 01:00:39 +00:00
return await self.message.channel.send(
"Invalid value. It must be between 1 and 1000 inclusive"
2022-10-25 01:00:39 +00:00
)
2022-10-04 00:33:15 +00:00
# Create player if they don't exist
player, _ = Player.get_or_create(
player_id=self.message.author.id, player_name=self.message.author.name
)
# Create the guess (or allow us to say update successful)
_, created = Guess.get_or_create(
2022-11-09 17:48:31 +00:00
guess_id=uuid.uuid4(), game_id=self.game.game_id, player_id=player.player_id
2022-10-04 00:33:15 +00:00
)
Guess.update({"guess": value}).where(
2022-11-11 03:46:42 +00:00
(Guess.game == self.game.game_id)
& (Guess.player_id == self.message.author.id)
).execute()
2022-10-27 05:18:12 +00:00
if created:
return await self.message.add_reaction("\N{THUMBS UP SIGN}")
2022-10-27 05:18:12 +00:00
return await self.message.channel.send(
f"<@{ str(self.message.author.id) }> your guess has been updated"
)
2022-10-04 00:33:15 +00:00
2022-09-24 02:41:03 +00:00
async def points(self):
"""
Points command - returns a table ordered from highest to lowest
of the players and their points
"""
2022-11-11 03:46:18 +00:00
message = (
"\nPlayers, who played recently, with their points highest to lowest\n\n"
2022-11-11 03:46:18 +00:00
)
message += "Player | Total Points | Last Played\n"
players = Player.select(
Player.player_name, Player.total_points, Player.last_update
).order_by(Player.last_update.desc(), Player.total_points.desc())
for player in players:
message += (
" | ".join(
[
player.player_name,
str(player.total_points),
str(player.last_update)[:-10],
]
)
+ "\n"
)
return await self.message.channel.send(message)
2022-09-24 02:41:03 +00:00
async def help(self):
2022-11-11 03:58:55 +00:00
"""help command - Sends a DM to the requesting user with available commands"""
help_message = (
"Braveball commands\n"
+ "!braveball - Start new game\n"
+ "!guess - While a game is running, add a guess"
+ " (or update an existing one) from 1-1000\n"
+ "!resolve <value> - 1-1000 to resolve the game\n"
+ " You can also add a batter's guess with: "
+ "!resolve <value> <discord id #> <guess>\n"
+ "!points - Shows a table of the most recent players, and their scores\n"
+ "!help - Shows this message"
)
2022-09-24 02:41:03 +00:00
recipient = await self.discord.fetch_user(self.message.author.id)
await recipient.send(help_message)