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:
parent
0f4b5100d9
commit
43770a8efc
44
bot.py
44
bot.py
@ -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
|
||||||
|
Loading…
Reference in New Issue
Block a user