2020-12-22 06:56:33 +00:00
import discord , json , os , roman , games , asyncio
2020-12-20 00:08:09 +00:00
import database as db
2020-12-20 01:26:05 +00:00
import onomancer as ono
2020-12-20 00:08:09 +00:00
client = discord . Client ( )
2020-12-22 06:56:33 +00:00
gamesarray = [ ]
2020-12-23 01:20:58 +00:00
setupmessages = { }
2020-12-20 00:08:09 +00:00
def config ( ) :
if not os . path . exists ( " config.json " ) :
#generate default config
config_dic = {
" token " : " " ,
" owners " : [
0000
2020-12-20 01:26:05 +00:00
] ,
" prefix " : [ " m; " , " m! " ] ,
" soulscream channel id " : 0
2020-12-20 00:08:09 +00:00
}
with open ( " config.json " , " w " ) as config_file :
json . dump ( config_dic , config_file , indent = 4 )
print ( " please fill in bot token and any bot admin discord ids to the new config.json file! " )
quit ( )
else :
with open ( " config.json " ) as config_file :
return json . load ( config_file )
@client.event
async def on_ready ( ) :
2020-12-20 03:14:45 +00:00
db . initialcheck ( )
2020-12-20 00:08:09 +00:00
print ( f " logged in as { client . user } with token { config ( ) [ ' token ' ] } " )
2020-12-23 01:20:58 +00:00
@client.event
async def on_reaction_add ( reaction , user ) :
if reaction . message in setupmessages . keys ( ) :
game = setupmessages [ reaction . message ]
try :
if str ( reaction . emoji ) == " 🔼 " and not user == client . user :
new_player = games . player ( ono . get_stats ( db . get_user_player ( user ) [ " name " ] ) )
game . teams [ " away " ] . add_lineup ( new_player )
await reaction . message . channel . send ( f " { new_player } { new_player . star_string ( ' batting_stars ' ) } takes spot # { len ( game . teams [ ' away ' ] . lineup ) } on the away lineup. " )
elif str ( reaction . emoji ) == " 🔽 " and not user == client . user :
new_player = games . player ( ono . get_stats ( db . get_user_player ( user ) [ " name " ] ) )
game . teams [ " home " ] . add_lineup ( new_player )
await reaction . message . channel . send ( f " { new_player } { new_player . star_string ( ' batting_stars ' ) } takes spot # { len ( game . teams [ ' home ' ] . lineup ) } on the home lineup. " )
except :
await reaction . message . channel . send ( f " { user . display_name } , we can ' t find your idol. Maybe you don ' t have one yet? " )
2020-12-20 00:08:09 +00:00
@client.event
async def on_message ( msg ) :
2020-12-22 20:02:28 +00:00
if msg . author == client . user :
return
2020-12-20 05:13:23 +00:00
command_b = False
2020-12-20 01:26:05 +00:00
for prefix in config ( ) [ " prefix " ] :
if msg . content . startswith ( prefix ) :
2020-12-20 05:13:23 +00:00
command_b = True
command = msg . content . split ( prefix , 1 ) [ 1 ]
if not command_b :
2020-12-20 01:26:05 +00:00
return
2020-12-20 05:13:23 +00:00
if msg . author . id in config ( ) [ " owners " ] and command == " introduce " :
await introduce ( msg . channel )
elif msg . channel . id == config ( ) [ " soulscream channel id " ] :
try :
2020-12-20 12:25:25 +00:00
await msg . channel . send ( ono . get_scream ( msg . author . nick ) )
2020-12-20 06:31:48 +00:00
except TypeError or AttributeError :
2020-12-20 12:25:25 +00:00
await msg . channel . send ( ono . get_scream ( msg . author . name ) )
2020-12-20 06:31:48 +00:00
except AttributeError :
2020-12-20 12:25:25 +00:00
await msg . channel . send ( ono . get_scream ( msg . author . name ) )
2020-12-20 05:13:23 +00:00
2020-12-20 10:28:47 +00:00
elif command . startswith ( " roman " ) :
possible_int_string = command . split ( " " , 1 ) [ 1 ]
try :
await msg . channel . send ( roman . roman_convert ( possible_int_string ) )
except ValueError :
await msg . channel . send ( f " \" { possible_int_string } \" isn ' t an integer in Arabic numerals. " )
2020-12-20 12:25:25 +00:00
elif command . startswith ( " idolize " ) :
if ( command . startswith ( " idolizememe " ) ) :
meme = True
else :
meme = False
2020-12-22 20:15:18 +00:00
player_name = discord . utils . escape_mentions ( command . split ( " " , 1 ) [ 1 ] )
if len ( player_name ) > = 70 :
await msg . channel . send ( " That name is too long. Please keep it below 70 characters, for my sake and yours. " )
return
2020-12-23 23:02:55 +00:00
try :
player_json = ono . get_stats ( player_name )
db . designate_player ( msg . author , json . loads ( player_json ) )
if not meme :
await msg . channel . send ( f " { player_name } is now your idol. " )
else :
await msg . channel . send ( f " { player_name } is now { msg . author . display_name } ' s idol. " )
await msg . channel . send ( f " Reply if { player_name } is your idol also. " )
except :
await msg . channel . send ( " Something went wrong. Tell 16. " )
2020-12-20 12:25:25 +00:00
elif command == " showidol " :
try :
player_json = db . get_user_player ( msg . author )
embed = build_star_embed ( player_json )
embed . set_footer ( text = msg . author . display_name )
await msg . channel . send ( embed = embed )
except :
await msg . channel . send ( " We can ' t find your idol. Looked everywhere, too. " )
2020-12-24 09:51:38 +00:00
elif command . startswith ( " showplayer " ) :
player_name = json . loads ( ono . get_stats ( command . split ( " " , 1 ) [ 1 ] ) )
await msg . channel . send ( embed = build_star_embed ( player_name ) )
2020-12-21 09:46:12 +00:00
2020-12-24 10:18:29 +00:00
elif command . startswith ( " startgame \n " ) and msg . author . id in config ( ) [ " owners " ] :
try :
team1 = games . get_team ( command . split ( " \n " ) [ 1 ] )
team2 = games . get_team ( command . split ( " \n " ) [ 2 ] )
innings = int ( command . split ( " \n " ) [ 3 ] )
except IndexError :
await msg . channel . send ( " We need four lines: startgame, away team, home team, and the number of innings. " )
return
except :
await msg . channel . send ( " Something about that command tripped us up. Probably the number of innings at the end? " )
if team1 is not None and team2 is not None :
game = games . game ( msg . author . name , team1 , team2 , length = innings )
game_task = asyncio . create_task ( watch_game ( msg . channel , game ) )
await game_task
2020-12-22 06:56:33 +00:00
2020-12-23 01:20:58 +00:00
elif command . startswith ( " setupgame " ) and msg . author . id in config ( ) [ " owners " ] :
for game in gamesarray :
if game [ 0 ] . name == msg . author . name :
await msg . channel . send ( " There ' s already an active game with that name. " )
return
2020-12-24 09:51:38 +00:00
try :
inningmax = int ( command . split ( " setupgame " ) [ 1 ] )
except :
inningmax = 3
game_task = asyncio . create_task ( setup_game ( msg . channel , msg . author , games . game ( msg . author . name , games . team ( ) , games . team ( ) , length = inningmax ) ) )
2020-12-23 01:20:58 +00:00
await game_task
2020-12-24 09:51:38 +00:00
elif command . startswith ( " saveteam \n " ) and msg . author . id in config ( ) [ " owners " ] :
save_task = asyncio . create_task ( save_team_batch ( msg , command ) )
await save_task
2020-12-22 06:56:33 +00:00
2020-12-24 09:51:38 +00:00
elif command . startswith ( " showteam " ) :
team = games . get_team ( command . split ( " " , 1 ) [ 1 ] )
if team is not None :
await msg . channel . send ( embed = build_team_embed ( team ) )
else :
await msg . channel . send ( " Can ' t find that team, boss. Typo? " )
2020-12-21 09:46:12 +00:00
2020-12-20 05:17:05 +00:00
elif command == " credit " :
await msg . channel . send ( " Our avatar was graciously provided to us, with permission, by @HetreaSky on Twitter. " )
2020-12-20 05:13:23 +00:00
async def introduce ( channel ) :
text = """ **Your name, favorite team, and pronouns**: Matteo Prestige, CHST, they/them ***only.*** There ' s more than one of us up here, after all.
* * What are you majoring in ( wrong answers only ) * * : Economics .
2020-12-20 05:17:05 +00:00
* * Your favorite and least favorite beverage , without specifying which * * : Vanilla milkshakes , chocolate milkshakes .
2020-12-20 05:13:23 +00:00
* * Favorite non - Mild Low team * * : The Mills . We hope they ' re treating Ren alright.
* * If you were a current blaseball player , who would you be * * : We refuse to answer this question .
* * Your hobbies / interests * * : Minigolf , blaseball , felony insider trading .
2020-12-20 05:17:05 +00:00
Our avatar was graciously provided to us , with permission , by @HetreaSky on Twitter .
2020-12-20 05:13:23 +00:00
"""
await channel . send ( text )
2020-12-20 00:08:09 +00:00
2020-12-22 06:56:33 +00:00
async def start_game ( channel ) :
msg = await channel . send ( " Play ball! " )
await asyncio . sleep ( 4 )
newgame = games . debug_game ( )
gamesarray . append ( newgame )
while not newgame . over :
state = newgame . gamestate_update_full ( )
if not state . startswith ( " Game over " ) :
await msg . edit ( content = state )
2020-12-22 12:59:46 +00:00
await asyncio . sleep ( 3 )
2020-12-22 06:56:33 +00:00
await channel . send ( state )
gamesarray . pop ( )
2020-12-23 01:20:58 +00:00
async def setup_game ( channel , owner , newgame ) :
newgame . owner = owner
await channel . send ( f " Game sucessfully created! \n Start any commands for this game with ` { newgame . name } ` so I know who ' s talking about what. " )
await asyncio . sleep ( 1 )
await channel . send ( " Who ' s pitching for the away team? " )
def input ( msg ) :
return msg . content . startswith ( newgame . name ) and msg . channel == channel #if author or willing participant and in correct channel
while newgame . teams [ " home " ] . pitcher == None :
def nameinput ( msg ) :
return msg . content . startswith ( newgame . name ) and msg . channel == channel #if author or willing participant and in correct channel
while newgame . teams [ " away " ] . pitcher == None :
try :
namemsg = await client . wait_for ( ' message ' , check = input )
new_pitcher_name = discord . utils . escape_mentions ( namemsg . content . split ( f " { newgame . name } " ) [ 1 ] )
if len ( new_pitcher_name ) > 70 :
await channel . send ( " That player name is too long, chief. 70 or less. " )
else :
new_pitcher = games . player ( ono . get_stats ( new_pitcher_name ) )
newgame . teams [ " away " ] . set_pitcher ( new_pitcher )
await channel . send ( f " { new_pitcher } { new_pitcher . star_string ( ' pitching_stars ' ) } , pitching for the away team! \n Now, the home team ' s pitcher. Same dance, folks. " )
except NameError :
await channel . send ( " Uh. " )
try :
namemsg = await client . wait_for ( ' message ' , check = input )
new_pitcher_name = discord . utils . escape_mentions ( namemsg . content . split ( f " { newgame . name } " ) [ 1 ] )
if len ( new_pitcher_name ) > 70 :
await channel . send ( " That player name is too long, chief. 70 or less. " )
else :
new_pitcher = games . player ( ono . get_stats ( new_pitcher_name ) )
newgame . teams [ " home " ] . set_pitcher ( new_pitcher )
await channel . send ( f " And { new_pitcher } { new_pitcher . star_string ( ' pitching_stars ' ) } , pitching for the home team. " )
except :
await channel . send ( " Uh. " )
#pitchers assigned!
team_join_message = await channel . send ( f """ Now, the lineups! I need somewhere between 1 and 12 batters. Cloning helps a lot with this sort of thing.
React to this message with 🔼 to have your idol join the away team , or 🔽 to have them join the home team .
You can also enter names like you did for the pitchers , with a slight difference : ` away [ name ] ` or ` home [ name ] ` instead of just the name .
Creator , type ` { newgame . name } done ` to finalize lineups . """ )
await team_join_message . add_reaction ( " 🔼 " )
await team_join_message . add_reaction ( " 🔽 " )
setupmessages [ team_join_message ] = newgame
#emoji_task = asyncio.create_task(watch_for_reacts(team_join_message, ready, newgame))
#msg_task = asyncio.create_task(watch_for_messages(channel, ready, newgame))
#await asyncio.gather(
# watch_for_reacts(team_join_message, newgame),
# watch_for_messages(channel, newgame)
# )
def messagecheck ( msg ) :
return ( msg . content . startswith ( newgame . name ) ) and msg . channel == channel and msg . author != client . user
while not newgame . ready :
msg = await client . wait_for ( ' message ' , check = messagecheck )
new_player = None
if msg . author == newgame . owner and msg . content == f " { newgame . name } done " :
if newgame . teams [ ' home ' ] . finalize ( ) and newgame . teams [ ' away ' ] . finalize ( ) :
newgame . ready = True
break
else :
side = None
if msg . content . split ( f " { newgame . name } " ) [ 1 ] . split ( " " , 1 ) [ 0 ] == " home " :
side = " home "
elif msg . content . split ( f " { newgame . name } " ) [ 1 ] . split ( " " , 1 ) [ 0 ] == " away " :
side = " away "
if side is not None :
new_player_name = discord . utils . escape_mentions ( msg . content . split ( f " { newgame . name } " ) [ 1 ] . split ( " " , 1 ) [ 1 ] )
if len ( new_player_name ) > 70 :
await channel . send ( " That player name is too long, chief. 70 or less. " )
else :
new_player = games . player ( ono . get_stats ( new_player_name ) )
try :
if new_player is not None :
newgame . teams [ side ] . add_lineup ( new_player )
await channel . send ( f " { new_player } { new_player . star_string ( ' batting_stars ' ) } takes spot # { len ( newgame . teams [ side ] . lineup ) } on the { side } lineup. " )
except :
True
del setupmessages [ team_join_message ] #cleanup!
await channel . send ( " Name the away team, creator. " )
def ownercheck ( msg ) :
return msg . author == newgame . owner
while newgame . teams [ " home " ] . name == None :
while newgame . teams [ " away " ] . name == None :
newname = await client . wait_for ( ' message ' , check = ownercheck )
if len ( newname . content ) < 30 :
newgame . teams [ ' away ' ] . name = newname . content
await channel . send ( f " Stepping onto the field, the visitors: { newname . content } ! \n Finally, the home team, and we can begin. " )
else :
await channel . send ( " Hey, keep these to 30 characters or less please. Discord messages have to stay short. " )
newname = await client . wait_for ( ' message ' , check = ownercheck )
if len ( newname . content ) < 30 :
newgame . teams [ ' home ' ] . name = newname . content
await channel . send ( f " Next on the diamond, your home team: { newname . content } ! " )
else :
await channel . send ( " Hey, keep these to 30 characters or less please. Discord messages have to stay short. " )
await asyncio . sleep ( 3 )
await channel . send ( f " ** { newgame . teams [ ' away ' ] . name } at { newgame . teams [ ' home ' ] . name } ** " )
game_task = asyncio . create_task ( watch_game ( channel , newgame ) )
await game_task
async def watch_game ( channel , game ) :
2020-12-22 12:59:46 +00:00
blank_emoji = discord . utils . get ( client . emojis , id = 790899850295509053 )
empty_base = discord . utils . get ( client . emojis , id = 790899850395779074 )
first_base = discord . utils . get ( client . emojis , id = 790899850320543745 )
second_base = discord . utils . get ( client . emojis , id = 790900139656740865 )
third_base = discord . utils . get ( client . emojis , id = 790900156597403658 )
2020-12-24 09:51:38 +00:00
out_emoji = discord . utils . get ( client . emojis , id = 791578957241778226 )
in_emoji = discord . utils . get ( client . emojis , id = 791578957244792832 )
2020-12-22 12:59:46 +00:00
2020-12-23 01:20:58 +00:00
newgame = game
2020-12-22 12:59:46 +00:00
embed = await channel . send ( " Play ball! " )
2020-12-23 01:20:58 +00:00
msg_top = await channel . send ( f " { newgame . name } " )
msg_bot = await channel . send ( f " starting... " )
2020-12-22 12:59:46 +00:00
await asyncio . sleep ( 4 )
use_emoji_names = True
for game in gamesarray :
if game [ 1 ] :
use_emoji_names = False
gamesarray . append ( ( newgame , use_emoji_names ) )
while not newgame . over :
state = newgame . gamestate_update_full ( )
punc = " "
if newgame . last_update [ 0 ] [ " defender " ] != " " :
punc = " . "
new_embed = discord . Embed ( color = discord . Color . purple ( ) , title = f " { newgame . teams [ ' away ' ] . name } at { newgame . teams [ ' home ' ] . name } " )
new_embed . add_field ( name = newgame . teams [ ' away ' ] . name , value = newgame . teams [ ' away ' ] . score , inline = True )
new_embed . add_field ( name = newgame . teams [ ' home ' ] . name , value = newgame . teams [ ' home ' ] . score , inline = True )
if newgame . top_of_inning :
new_embed . add_field ( name = " Inning: " , value = f " 🔼 { newgame . inning } " , inline = True )
else :
new_embed . add_field ( name = " Inning: " , value = f " 🔽 { newgame . inning } " , inline = True )
2020-12-24 09:51:38 +00:00
new_embed . add_field ( name = " Outs: " , value = f " { str ( out_emoji ) * newgame . outs + str ( in_emoji ) * ( 2 - newgame . outs ) } " , inline = True )
2020-12-22 12:59:46 +00:00
new_embed . add_field ( name = " Pitcher: " , value = newgame . get_pitcher ( ) , inline = False )
new_embed . add_field ( name = " Batter: " , value = newgame . get_batter ( ) , inline = False )
updatestring = f " { newgame . last_update [ 0 ] [ ' batter ' ] } { newgame . last_update [ 0 ] [ ' text ' ] . value } { newgame . last_update [ 0 ] [ ' defender ' ] } { punc } "
if newgame . last_update [ 1 ] > 0 :
updatestring + = f " { newgame . last_update [ 1 ] } runs scored! "
new_embed . add_field ( name = " 🏏 " , value = updatestring , inline = False )
basemessage_t = str ( blank_emoji )
if newgame . bases [ 2 ] is not None :
basemessage_t + = str ( second_base )
else :
basemessage_t + = str ( empty_base )
basemessage_b = " "
if newgame . bases [ 3 ] is not None :
basemessage_b + = str ( third_base )
else :
basemessage_b + = str ( empty_base )
basemessage_b + = str ( blank_emoji )
if newgame . bases [ 1 ] is not None :
basemessage_b + = str ( first_base )
else :
basemessage_b + = str ( empty_base )
await embed . edit ( content = None , embed = new_embed )
2020-12-23 01:20:58 +00:00
await asyncio . sleep ( .5 )
2020-12-22 12:59:46 +00:00
await msg_top . edit ( content = basemessage_t )
await msg_bot . edit ( content = basemessage_b )
2020-12-23 01:20:58 +00:00
await asyncio . sleep ( 5 )
2020-12-22 12:59:46 +00:00
punc = " "
if newgame . last_update [ 0 ] [ " defender " ] != " " :
punc = " . "
final_embed = discord . Embed ( color = discord . Color . dark_purple ( ) , title = f " { newgame . teams [ ' away ' ] . name } at { newgame . teams [ ' home ' ] . name } " )
final_embed . add_field ( name = " Final score: " , value = f " { newgame . teams [ ' away ' ] . score } to { newgame . teams [ ' home ' ] . score } " )
final_embed . add_field ( name = " Last update: " , value = f " { newgame . last_update [ 0 ] [ ' batter ' ] } { newgame . last_update [ 0 ] [ ' text ' ] . value } { newgame . last_update [ 0 ] [ ' defender ' ] } { punc } " )
await embed . edit ( content = None , embed = final_embed )
if newgame . teams [ ' away ' ] . score > newgame . teams [ ' home ' ] . score :
await msg_top . edit ( content = f " Game over! \n { newgame . teams [ ' away ' ] . name } wins! " )
else :
await msg_top . edit ( content = f " Game over! \n { newgame . teams [ ' home ' ] . name } wins! " )
await msg_bot . delete ( )
gamesarray . pop ( gamesarray . index ( ( newgame , use_emoji_names ) ) ) #cleanup is important!
del newgame
2020-12-24 09:51:38 +00:00
def build_team_embed ( team ) :
embed = discord . Embed ( color = discord . Color . purple ( ) , title = team . name )
lineup_string = " "
for player in team . lineup :
lineup_string + = f " { player . name } { player . star_string ( ' batting_stars ' ) } \n "
embed . add_field ( name = " Pitcher: " , value = f " { team . pitcher . name } { team . pitcher . star_string ( ' pitching_stars ' ) } . " , inline = False )
embed . add_field ( name = " Lineup: " , value = lineup_string , inline = False )
embed . set_footer ( text = team . slogan )
return embed
2020-12-20 00:08:09 +00:00
2020-12-20 12:25:25 +00:00
def build_star_embed ( player_json ) :
starkeys = { " batting_stars " : " Batting " , " pitching_stars " : " Pitching " , " baserunning_stars " : " Baserunning " , " defense_stars " : " Defense " }
embed = discord . Embed ( color = discord . Color . purple ( ) , title = player_json [ " name " ] )
for key in starkeys . keys ( ) :
2020-12-20 19:05:31 +00:00
embedstring = " "
2020-12-20 12:25:25 +00:00
starstring = str ( player_json [ key ] )
2020-12-20 19:05:31 +00:00
if " .5 " in starstring :
2020-12-20 12:25:25 +00:00
starnum = int ( starstring [ 0 ] )
addhalf = True
else :
starnum = int ( player_json [ key ] )
addhalf = False
2020-12-21 05:23:02 +00:00
embedstring + = " ⭐ " * starnum
2020-12-20 12:25:25 +00:00
if addhalf :
embedstring + = " ✨ "
embed . add_field ( name = starkeys [ key ] , value = embedstring , inline = False )
return embed
2020-12-24 09:51:38 +00:00
async def save_team_batch ( message , command ) :
newteam = games . team ( )
#try:
roster = command . split ( " \n " , 1 ) [ 1 ] . split ( " \n " )
newteam . name = roster [ 0 ] #first line is team name
newteam . slogan = roster [ 1 ] #second line is slogan
for rosternum in range ( 2 , len ( roster ) - 1 ) :
if roster [ rosternum ] != " " :
newteam . add_lineup ( games . player ( ono . get_stats ( roster [ rosternum ] ) ) )
newteam . set_pitcher ( games . player ( ono . get_stats ( roster [ len ( roster ) - 1 ] ) ) ) #last line is pitcher name
2020-12-21 05:23:02 +00:00
2020-12-24 09:51:38 +00:00
if len ( newteam . name ) > 30 :
await message . send ( " Team names have to be less than 30 characters! Try again. " )
return
elif len ( newteam . slogan ) > 100 :
await message . send ( " We ' ve given you 100 characters for the slogan. Discord puts limits on us and thus, we put limits on you. C ' est la vie. " )
await message . channel . send ( embed = build_team_embed ( newteam ) )
checkmsg = await message . channel . send ( " Does this look good to you, boss? " )
await checkmsg . add_reaction ( " 👍 " )
await checkmsg . add_reaction ( " 👎 " )
def react_check ( react , user ) :
return user == message . author and react . message == checkmsg
try :
react , user = await client . wait_for ( ' reaction_add ' , timeout = 20.0 , check = react_check )
if react . emoji == " 👍 " :
await message . channel . send ( " You got it, chief. Saving now. " )
games . save_team ( newteam )
await message . channel . send ( " Saved! Thank you for flying Air Matteo. We hope you had a pleasant data entry. " )
return
elif react . emoji == " 👎 " :
await message . channel . send ( " Message received. Pumping brakes, turning this car around. Try again, chief. " )
return
except asyncio . TimeoutError :
await message . channel . send ( " Look, I don ' t have all day. 20 seconds is long enough, right? Try again. " )
return
#except:
#await message.channel.send("uh.")
2020-12-20 00:08:09 +00:00
client . run ( config ( ) [ " token " ] )
