147 lines
5.7 KiB
Python
147 lines
5.7 KiB
Python
import sys
|
|
|
|
import discord
|
|
import uuid
|
|
import datetime
|
|
|
|
from src.main.configs import Configs
|
|
from src.main.database_module.guess_dao import GuessDAO, GUESSED_NUMBER, MEMBER_ID, MEMBER_NAME
|
|
from src.main.database_module.play_dao import PlayDAO, PLAY_ID, CREATION_DATE
|
|
from src.main.db_session import DatabaseSession
|
|
from src.main.discord_module.leaderboard_config import LeaderboardConfig
|
|
|
|
play_dao = None
|
|
guess_dao = None
|
|
bot = discord.Client()
|
|
|
|
@bot.event
|
|
async def on_ready():
|
|
print('Logged in as')
|
|
print(bot.user.name)
|
|
print(bot.user.id)
|
|
print('------')
|
|
|
|
@bot.event
|
|
async def on_message(message):
|
|
if message.author == bot.user:
|
|
return
|
|
|
|
content = message.content
|
|
|
|
'''
|
|
Sets up the next set of guesses.
|
|
'''
|
|
if content.startswith('!ghostball'):
|
|
if play_dao.is_active_play():
|
|
await message.channel.send("There's already an active play. Could you close that one first, please?")
|
|
else:
|
|
play_object = {PLAY_ID : uuid.uuid4(), CREATION_DATE : datetime.datetime.now()}
|
|
play_dao.insert(play_object)
|
|
|
|
await message.channel.send( "@flappyball, pitch is in! Send me your guesses with a !guess command.")
|
|
|
|
if content.startswith("!guess"):
|
|
guess_value = __parse_guess__(content)
|
|
|
|
if not play_dao.is_active_play():
|
|
await message.channel.send( "Hey, there's no active play! Start one up first with !ghostball.")
|
|
else:
|
|
play = play_dao.get_active_play()
|
|
guess_object = {PLAY_ID: play['play_id'],
|
|
MEMBER_ID: str(message.author.id),
|
|
GUESSED_NUMBER: guess_value,
|
|
MEMBER_NAME: str(message.author.name)}
|
|
|
|
if guess_dao.insert(guess_object):
|
|
await message.add_reaction(emoji="\N{THUMBS UP SIGN}")
|
|
|
|
# Closes off the active play to be ready for the next set
|
|
if content.startswith('!resolve'):
|
|
# try:
|
|
pitch_value = __parse_resolve_play__(content)
|
|
if pitch_value is None:
|
|
await message.channel.send( "Hey " + "<@" + str(message.author.id) + ">, I'm not sure what you meant. "
|
|
"You need real, numeric, values for this command to work. "
|
|
"Use !resolve <pitch number> <swing_number> and try again.")
|
|
|
|
# Check if we have an active play
|
|
if not play_dao.is_active_play():
|
|
await message.channel.send( "You confused me. There's no active play so I have nothing to close!")
|
|
else:
|
|
play = play_dao.resolve_play(pitch_value)
|
|
guess_dao.set_differences(pitch_value, play['play_id'])
|
|
|
|
await message.channel.send(
|
|
"You got it boss. Closed this play, no further guesses allowed!")
|
|
|
|
# Likely due to too few parameters but could be any number of things
|
|
# except :
|
|
# await message.channel.send( "Hey " + "<@" + str(message.author.id) + ">, you confused me with that message. "
|
|
# "To close an active pitch, the proper command is !resolve <pitch number> <swing_number>. "
|
|
# "Use that format and try again, ok?")
|
|
|
|
if content.startswith('!leaderboard'):
|
|
leaderboard_config = __parse_leaderboard_message__(content)
|
|
|
|
if leaderboard_config.should_sort_by_pure_closest():
|
|
values = guess_dao.fetch_closest(10)
|
|
|
|
string_to_send = ''
|
|
for i, value in enumerate(values):
|
|
string_to_send += str(i + 1) + ': ' + value['member_name'] + ', ' + value['difference'] + '\n'
|
|
|
|
await message.channel.send( string_to_send)
|
|
|
|
elif leaderboard_config.should_sort_by_best_average():
|
|
pass
|
|
else:
|
|
await message.channel.send( "I don't understand that leaderboard command, sorry! I know it's a little confusing, so send me"
|
|
" a !help message if you want the full rundown for how to make this work!")
|
|
|
|
if content.startswith('!help'):
|
|
help_message = __get_help_message__()
|
|
recipient = await bot.fetch_user(message.author.id)
|
|
await recipient.send(help_message)
|
|
|
|
def __get_help_message__():
|
|
# Start message with person who asked for help
|
|
help_message = "Hey! I can be instructed to do any number of things! Use the following commands: \n" \
|
|
"!guess <NUMBER> --- This will add your guess to the currently active play. " \
|
|
"I will give you a thumbs up if everything worked!\n" \
|
|
"!ghostball --- Starts a new play. I'll let you know if this didn't work for some reason!\n" \
|
|
"!help --- You just asked for this. If you ask for it again, I'll repeat myself.\n" \
|
|
"!resolve <PITCH_NUMBER> --- Uses the pitch number and real swing number " \
|
|
"to figure out who was closest and ends the active play.\n" \
|
|
"<HELP MESSAGE NEEDS DOCUMENTATION FOR LEADERBOARD COMMAND! PING KALI IF YOU'RE ANGRY!>\n"
|
|
|
|
return help_message
|
|
|
|
|
|
def __parse_leaderboard_message__(message_content):
|
|
return LeaderboardConfig(message_content)
|
|
|
|
def __parse_guess__(message_content):
|
|
pieces = message_content.split(' ')
|
|
try:
|
|
return pieces[1]
|
|
except TypeError:
|
|
return None
|
|
|
|
def __parse_resolve_play__(message_content):
|
|
pieces = message_content.split(' ')
|
|
try:
|
|
return pieces[1]
|
|
except TypeError:
|
|
return None
|
|
|
|
if __name__ == '__main__':
|
|
args = sys.argv
|
|
token = args[1]
|
|
file_path = args[2]
|
|
|
|
configs = Configs(file_path)
|
|
databaseSession = DatabaseSession()
|
|
|
|
play_dao = PlayDAO()
|
|
guess_dao = GuessDAO()
|
|
bot.run(token) |