zed/BaseballBot
1
0
Fork 0
forked from c0de/BaseballBot
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Format with Black

Browse Source
This commit is contained in:
c0de 2022-10-24 20:00:39 -05:00
parent 7d684d8b9d
commit 07ecd2248c
4 changed files with 52 additions and 38 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
GhostBallBot/database/models.py
View File

@ -8,16 +8,18 @@ import datetime
from peewee import *
# User can provide path to database, or it will be put next to main.py
DATABASE = os.environ.get('database_path', os.getcwd() + '/ghostball.db')
DATABASE = os.environ.get("database_path", os.getcwd() + "/ghostball.db")
database = SqliteDatabase(DATABASE)
class BaseModel(Model):
"""All of our models will inherit this class
and use the same database connection"""
and use the same database connection"""
class Meta:
database = database
class GameModel(BaseModel):
game_id = UUIDField(primary_key=True)
@ -27,10 +29,11 @@ class GameModel(BaseModel):
date_created = DateTimeField(default=datetime.datetime.now)
date_ended = DateTimeField(null=True)
class GuessModel(BaseModel):
player_id = IntegerField(primary_key=True)
game_id = ForeignKeyField(GameModel, backref="guesses")
game_id = ForeignKeyField(GameModel, backref="guesses")
player_name = CharField()
guess = IntegerField()

6
GhostBallBot/discord_client/client.py
View File

@ -7,11 +7,11 @@ import sys
import discord
# Import game functions
sys.path.append('..')
sys.path.append("..")
import game
class GhostBallClient(discord.Client):
class GhostBallClient(discord.Client):
def __init__(self, *args, **kwargs):
super(GhostBallClient, self).__init__(*args, **kwargs)
@ -31,7 +31,7 @@ class GhostBallClient(discord.Client):
await message.channel.send("pong")
# Game commands
if message.content.startswith('!'):
if message.content.startswith("!"):
firstword = message.content[1:].split()[0]
# Determine if the first word is a command, and run it

69
GhostBallBot/game.py
View File

@ -8,29 +8,30 @@ import dateparser
from database.models import database, GameModel, GuessModel
class Game:
"""
The game state class
The game state class
This represents a game that exists in a channel
This represents a game that exists in a channel
"""
def __enter__(self):
"""
Allows use of `with Game as game` for try/except statements
Allows use of `with Game as game` for try/except statements
We are using this instead of __init__, they work very similar
to each other (https://peps.python.org/pep-0343/)
We are using this instead of __init__, they work very similar
to each other (https://peps.python.org/pep-0343/)
"""
# 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,
"ghostball": self.start,
"resolve": self.stop,
"guess": self.guess,
"points": self.points,
"help": self.help,
}
self.game = GameModel
@ -46,21 +47,24 @@ class Game:
def __exit__(self, exception_type, exception_value, exception_traceback):
"""
Automagically close the database
when this class has ended execution
Automagically close the database
when this class has ended execution
"""
database.close()
async def check_is_running(method, start_new_game=True):
"""
Decorator that determines if the game is running or not
Decorator that determines if the game is running or not
"""
async def wrapper(self):
if self.is_running and start_new_game:
return await self.message.channel.send("A game is already running")
elif not self.is_running and not start_new_game:
return await self.message.channel.send("There is no game running to add guesses to")
return await self.message.channel.send(
"There is no game running to add guesses to"
)
await method(self)
@ -72,11 +76,12 @@ class Game:
# 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
game_id=uuid.uuid4(), server_id=self.message.guild.id
)
await self.message.send("@flappy ball, pitch is in! Send me your guesses with !guess <number>")
await self.message.send(
"@flappy ball, pitch is in! Send me your guesses with !guess <number>"
)
def __stopArgs__(self):
pieces = self.message.content.split()
@ -93,23 +98,27 @@ class Game:
# 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) }>!")
return await self.message.channel.send(
f"Invalid command <@{ str(self.message.author.id) }>!"
)
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)
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})
self.game.update({"pitch_value": pitch_value})
# TODO: Determine differences
await self.message.channel.send("Difference calculation is not currently available")
await self.message.channel.send(
"Difference calculation is not currently available"
)
# stop and discard game
self.is_running = False
@ -119,13 +128,15 @@ class Game:
async def guess(self):
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")
return await self.message.channel.send(
f"Invalid value. It must be between 1 and 1000 inclusive"
)
GuessModel.create(
game_id = self.game.game_id,
player_id = self.message.author.id,
player_name = self.message.author.name,
guess = value
game_id=self.game.game_id,
player_id=self.message.author.id,
player_name=self.message.author.name,
guess=value,
)
return await self.message.add_reaction(emoji="\N{THUMBS UP SIGN}")

4
GhostBallBot/main.py
View File

@ -8,9 +8,9 @@ from pathlib import Path
from discord_client.client import GhostBallClient
from database.models import DATABASE, database, create_models
if __name__ == '__main__':
if __name__ == "__main__":
client = GhostBallClient(intents=None)
client.run(os.environ.get('discord_token'))
client.run(os.environ.get("discord_token"))
# Set up the database if we haven't already
if not os.path.exists(DATABASE):