XZZ/worker/lsp.py

42 lines
1.1 KiB
Python
Raw Normal View History

2020-08-03 15:28:59 +08:00
from zzcore import StdAns, mysakuya
import requests
2021-05-06 20:35:46 +08:00
from config import LOLIKEY
2020-08-03 15:28:59 +08:00
2020-10-30 10:13:44 +08:00
2020-08-03 15:28:59 +08:00
class Ans(StdAns):
2021-05-06 20:35:46 +08:00
AllowGroup = [959613860, 973510746, 805197917, 343700338, 125733077, 1084566280,
2020-10-30 10:13:44 +08:00
920863253, 798595664, 655057127, 196268763, 247022495, 474907856, 940333876]
2020-08-03 15:28:59 +08:00
def GETMSG(self):
2021-05-06 20:35:46 +08:00
if len(self.parms) == 1:
code, picurl = getsetu()
2020-08-03 15:28:59 +08:00
else:
keyword = self.parms[1]
if mysakuya(self, keyword) == False:
return "不许你们看咲夜的涩图!!"
2021-05-06 20:35:46 +08:00
code, picurl = getsetu()
if code == 0:
return f'[CQ:reply,id={self.mid}][CQ:image,file={picurl},type=flash]'
else:
return f'[CQ:reply,id={self.mid}] 什么东西出错了code:{code}'
2020-10-30 10:13:44 +08:00
2021-05-06 20:35:46 +08:00
def getsetu(keyword=''):
url = 'https://api.lolicon.app/setu/'
params = {
2021-05-06 20:35:46 +08:00
'apikey': LOLIKEY,
'keyword': keyword,
}
2020-10-30 10:13:44 +08:00
try:
2021-05-06 20:35:46 +08:00
resp = requests.get(url=url, params=params, timeout=5).json()
except:
2021-05-06 20:35:46 +08:00
return 500, ''
2020-10-30 10:13:44 +08:00
2021-05-06 20:35:46 +08:00
picurl = ''
if resp['code'] == 0:
2021-05-06 20:35:46 +08:00
picurl = "https://r.zjuyk.site/" + resp['data'][0]['url']
return resp['code'], picurl