XZZ/worker/madoka.py

46 lines
1.5 KiB
Python
Raw Normal View History

2020-08-05 21:03:48 +08:00
from zzcore import StdAns
import sqlite3
from random import randint, choice
n = {'madoka': '鹿目圆', 'homura': '晓美焰', 'sayaka': '美树沙耶香',
'kyoko': '佐仓杏子', 'mami': '巴麻美', 'qb': '丘比'}
def getQuote(name='random'):
2020-08-05 21:32:35 +08:00
conn = sqlite3.connect('./data/madoka/quote.db')
2020-08-05 21:03:48 +08:00
cursor = conn.cursor()
msg = ''
try:
if name == 'random':
id = randint(1, 49)
cursor.execute('SELECT text FROM quote WHERE id=?', (id,))
res = cursor.fetchall()
msg += res[0][0]
else:
cursor.execute('SELECT text FROM quote WHERE name=?', (n[name],))
res = cursor.fetchall()
msg += choice(res)[0]
except Exception:
pass
finally:
cursor.close()
conn.close()
return msg
class Ans(StdAns):
def GETMSG(self):
try:
if len(self.parms) == 1:
return getQuote()
if len(self.parms) == 2:
if self.parms[1].lower() in ['madoka', 'homura', 'sayaka', 'kyoko', 'mami', 'qb']:
return getQuote(self.parms[1].lower())
else:
return '可以使用"/madoka [人物名字]"的方式调用此命令\n可选人物有: Madoka, Homura, Sayaka, Kyoko, Mami, QB'
else:
return '不知道怎么使用的话,就看看"/madoka help"'
except Exception:
return '出现了奇怪的错误呢'