1
0
mirror of https://github.com/MrDetonia/Maki.git synced 2024-11-22 20:00:00 -05:00

Added QR code generato, v0.16.0

This commit is contained in:
Zac Herd 2016-05-25 13:10:24 +01:00
parent 0f4b5100d9
commit 43770a8efc

44
bot.py
View File

@ -12,6 +12,8 @@ import asyncio
import os import os
import io import io
import sys import sys
import shlex
import subprocess
import time import time
import datetime import datetime
import random import random
@ -31,7 +33,7 @@ from secret import token
name = "Maki" name = "Maki"
# bot version # bot version
version = "v0.15.2" version = "v0.16.0"
# 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
@ -50,6 +52,7 @@ My commands are:
.sayy <msg> - say something a e s t h e t i c a l l y .sayy <msg> - say something a e s t h e t i c a l l y
.markov <user> - generate sentence using markov chains over a user's chat history .markov <user> - generate sentence using markov chains over a user's chat history
.roll <x>d<y> - roll x number of y sided dice .roll <x>d<y> - roll x number of y sided dice
.qr <msg> - generate a QR code
```""" ```"""
# IDs of admin users # IDs of admin users
@ -271,6 +274,45 @@ def on_message(message):
else: else:
response = 'you did it wrong!' response = 'you did it wrong!'
elif message.content.startswith('.qr '):
# generate QR code - DANGEROUS, CHECK CAREFULLY HERE
tmp = message.content[4:]
# send typing signal to discord
for attempt in range(5):
try:
yield from client.send_typing(message.channel)
except discord.errors.HTTPException:
continue
else:
break
else:
print('ERROR: Failed to send typing signal to discord after 5 attempts')
# make sure there are no nasty characters
msg = re.sub(r'[^a-zA-Z0-9_ -]', '', tmp, 0)
# echo message
cmd = 'echo "\'' + msg + '\'"'
args = shlex.split(cmd)
echo = subprocess.Popen(args, stdout=subprocess.PIPE)
# generate QR code
cmd = 'qrencode -t png -o -'
args = shlex.split(cmd)
qr = subprocess.Popen(args, stdin=echo.stdout, stdout=subprocess.PIPE)
# upload file with curl and get URL
cmd = 'curl -F upload=@- https://w1r3.net'
args = shlex.split(cmd)
out = subprocess.check_output(args, stdin=qr.stdout)
# run piped commands
echo.wait()
# send response
response = out.decode('utf-8').strip()
# Stuff that happens when message is not a bot command: # Stuff that happens when message is not a bot command:
else: else:
# log each message against users # log each message against users