45 lines
1.5 KiB
Python
45 lines
1.5 KiB
Python
from zzcore import StdAns
|
|
import sqlite3
|
|
from random import randint, choice
|
|
|
|
n = {'madoka': '鹿目圆', 'homura': '晓美焰', 'sayaka': '美树沙耶香',
|
|
'kyoko': '佐仓杏子', 'mami': '巴麻美', 'qb': '丘比'}
|
|
|
|
|
|
def getQuote(name='random'):
|
|
conn = sqlite3.connect('./data/madoka/quote.db')
|
|
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 '出现了奇怪的错误呢'
|