MrDetonia/Maki
1
0
Fork 0
mirror of https://github.com/MrDetonia/Maki.git synced 2024-11-22 20:00:00 -05:00
Code Issues Releases Wiki Activity

Added message retries to avoid 502 errors, v0.10.2

Browse Source
This commit is contained in:
Zac Herd 2016-04-03 16:54:00 +01:00
parent 0ede83abff
commit 15fe9e6e03
1 changed files with 43 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

63
bot.py
View File

@ -24,7 +24,7 @@ from secret import email,pwd
name = "Maki" name = "Maki"
# bot version # bot version
version = "v0.10.1" version = "v0.10.2"
# text shown by .help command # text shown by .help command
helptext = """I am a bot written in Python by MrDetonia helptext = """I am a bot written in Python by MrDetonia
@ -157,12 +157,21 @@ def on_message(message):
# do not parse own messages or private messages # do not parse own messages or private messages
if message.author != client.user and type(message.channel) is not discord.PrivateChannel: if message.author != client.user and type(message.channel) is not discord.PrivateChannel:
# response to send to channel
response = ''
# send any messages we have for author: # send any messages we have for author:
if message.author.name in tells: if message.author.name in tells:
yield from client.send_message(message.channel, 'Hey ' + message.author.name + ', I have messages for you!')
for msg in tells[message.author.name]: for msg in tells[message.author.name]:
for attempt in range(5):
try:
yield from client.send_message(message.author, msg[0] + ' says "' + msg[1] + '"') yield from client.send_message(message.author, msg[0] + ' says "' + msg[1] + '"')
except HTTPException:
continue
else:
break
else:
print('ERROR: Failed to send private message after 5 attempts')
# delete this user's entry # delete this user's entry
del tells[message.author.name] del tells[message.author.name]
@ -174,15 +183,15 @@ def on_message(message):
# parse messages for commands # parse messages for commands
if message.content.startswith('.bots') or message.content.startswith('.version'): if message.content.startswith('.bots') or message.content.startswith('.version'):
# print bot info # print bot info
yield from client.send_message(message.channel, 'I am ' + name + ', a Discord bot by MrDetonia | ' + version + ' | Python 3.4 | discord.py ' + discord.__version__) response = 'I am ' + name + ', a Discord bot by MrDetonia | ' + version + ' | Python 3.4 | discord.py ' + discord.__version__
elif message.content.startswith('.help'): elif message.content.startswith('.help'):
# print command list # print command list
yield from client.send_message(message.channel, helptext) response = helptext
elif message.content.startswith('.upskirt'): elif message.content.startswith('.upskirt'):
# link to source code # link to source code
yield from client.send_message(message.channel, 'No, don\'t look at my pantsu! Baka! <https://27b-a.xyz:55555/mrdetonia/Maki>') response = 'No, don\'t look at my pantsu! Baka! <https://27b-a.xyz:55555/mrdetonia/Maki>'
elif message.content.startswith('.die'): elif message.content.startswith('.die'):
if message.author.id in admins: if message.author.id in admins:
@ -191,30 +200,30 @@ def on_message(message):
yield from client.logout() yield from client.logout()
else: else:
# user not admin, refuse # user not admin, refuse
yield from client.send_message(message.channel, 'Don\'t be so rude! >:(') response = 'Don\'t be so rude! >:('
elif message.content.startswith('.whoami'): elif message.content.startswith('.whoami'):
# show info about user # show info about user
yield from client.send_message(message.channel, 'User: ' + message.author.name + ' ID: ' + message.author.id + ' Discriminator: ' + message.author.discriminator + '\nAccount Created: ' + strfromdt(message.author.created_at)) response = 'User: ' + message.author.name + ' ID: ' + message.author.id + ' Discriminator: ' + message.author.discriminator + '\nAccount Created: ' + strfromdt(message.author.created_at)
elif message.content.startswith('.whois'): elif message.content.startswith('.whois'):
# show info about another user # show info about another user
tmp = message.content[7:] tmp = message.content[7:]
if tmp in users: if tmp in users:
user = message.server.get_member(users[tmp]) user = message.server.get_member(users[tmp])
yield from client.send_message(message.channel, 'User: ' + user.name + ' ID: ' + user.id + ' Discriminator: ' + user.discriminator + '\nAccount Created: ' + strfromdt(user.created_at)) response = 'User: ' + user.name + ' ID: ' + user.id + ' Discriminator: ' + user.discriminator + '\nAccount Created: ' + strfromdt(user.created_at)
else: else:
yield from client.send_message(message.channel, 'I haven\'t seen ' + tmp + ' yet! :(') response = 'I haven\'t seen ' + tmp + ' yet! :('
elif message.content.startswith('.welcome'): elif message.content.startswith('.welcome'):
# manage welcome messages # manage welcome messages
if message.author.id in admins: if message.author.id in admins:
tmp = message.content[9:].split(' ',1) tmp = message.content[9:].split(' ',1)
welcomes[tmp[0]] = tmp[1] welcomes[tmp[0]] = tmp[1]
yield from client.send_message(message.channel, 'Okay, I will now greet ' + tmp[0] + ' with "' + tmp[1] + '"') response = 'Okay, I will now greet ' + tmp[0] + ' with "' + tmp[1] + '"'
else: else:
welcomes[message.author.name] = message.content[9:] welcomes[message.author.name] = message.content[9:]
yield from client.send_message(message.channel, 'Okay, I will now greet ' + message.author.name + ' with "' + message.content[9:] + '"') response = 'Okay, I will now greet ' + message.author.name + ' with "' + message.content[9:] + '"'
# save welcomes # save welcomes
with open('welcomes.json', 'w') as fp: with open('welcomes.json', 'w') as fp:
@ -225,13 +234,13 @@ def on_message(message):
target = message.content[6:] target = message.content[6:]
if target in history: if target in history:
# user logged, print last message and time # user logged, print last message and time
yield from client.send_message(message.channel, 'user ' + target + ' was last seen saying "' + history[target][0] + '" at ' + strfromdt(dtfromts(history[target][1]))) response = 'user ' + target + ' was last seen saying "' + history[target][0] + '" at ' + strfromdt(dtfromts(history[target][1]))
elif target == 'Maki': elif target == 'Maki':
# Maki doesn't need to be .seen # Maki doesn't need to be .seen
yield from client.send_message(message.channel, 'I\'m right here!') response = 'I\'m right here!'
else: else:
# user not logged # user not logged
yield from client.send_message(message.channel, 'user not seen yet') response = 'user not seen yet'
elif message.content.startswith('.tell'): elif message.content.startswith('.tell'):
# store message to tell user # store message to tell user
@ -246,13 +255,13 @@ def on_message(message):
json.dump(tells, fp) json.dump(tells, fp)
# let user know message is ready # let user know message is ready
yield from client.send_message(message.channel, 'Okay ' + message.author.name + ', I\'ll tell ' + tmp[0] + ' when I next see them!') response = 'Okay ' + message.author.name + ', I\'ll tell ' + tmp[0] + ' when I next see them!'
elif message.content.startswith('.say'): elif message.content.startswith('.say'):
# delete calling message for effect # delete calling message for effect
yield from client.delete_message(message) yield from client.delete_message(message)
# echo message # echo message
yield from client.send_message(message.channel, message.content[5:]) response = message.content[5:]
elif message.content.startswith('.markov'): elif message.content.startswith('.markov'):
# generate a markov chain sentence based on the user's chat history # generate a markov chain sentence based on the user's chat history
@ -260,11 +269,11 @@ def on_message(message):
if os.path.isfile('./markovs/' + users[tmp]): if os.path.isfile('./markovs/' + users[tmp]):
mc = markov.Markov(open('./markovs/' + users[tmp])) mc = markov.Markov(open('./markovs/' + users[tmp]))
try: try:
yield from client.send_message(message.channel, mc.generate_text()) response = mc.generate_text()
except KeyError: except KeyError:
yield from client.send_message(message.channel, 'Something went wrong :( Maybe you haven\'t spoken enough yet?') response = 'Something went wrong :( Maybe you haven\'t spoken enough yet?'
else: else:
yield from client.send_message(message.channel, 'I haven\'t seen that user speak yet!') response = 'I haven\'t seen that user speak yet!'
# Stuff that happens when message is not a bot command: # Stuff that happens when message is not a bot command:
else: else:
@ -280,18 +289,30 @@ def on_message(message):
# Ben meme trackers # Ben meme trackers
if '/ck/' in message.content and message.author.name == "Ben.H": if '/ck/' in message.content and message.author.name == "Ben.H":
bentrack['ck'] += 1 bentrack['ck'] += 1
yield from client.send_message(message.channel, 'I have seen Ben reference /ck/ ' + str(bentrack['ck']) + ' times now.') response = 'I have seen Ben reference /ck/ ' + str(bentrack['ck']) + ' times now.'
# save count # save count
with open('bentrack.json', 'w') as fp: with open('bentrack.json', 'w') as fp:
json.dump(bentrack, fp) json.dump(bentrack, fp)
elif '/fit/' in message.content and message.author.name == "Ben.H": elif '/fit/' in message.content and message.author.name == "Ben.H":
bentrack['fit'] += 1 bentrack['fit'] += 1
yield from client.send_message(message.channel, 'I have seen Ben reference /fit/ ' + str(bentrack['fit']) + ' times now.') response = 'I have seen Ben reference /fit/ ' + str(bentrack['fit']) + ' times now.'
# save count # save count
with open('bentrack.json', 'w') as fp: with open('bentrack.json', 'w') as fp:
json.dump(bentrack, fp) json.dump(bentrack, fp)
# send response to channel if needed:
if response is not '':
for attempt in range(5):
try:
yield from client.send_message(message.channel, response)
except HTTPException:
continue
else:
break
else:
print('ERROR: Failed to send message to discord after 5 attempts')
# Run the client # Run the client
client.run(email, pwd) client.run(email, pwd)