52 lines
1.4 KiB
Python
52 lines
1.4 KiB
Python
from zzcore import StdAns, mysakuya
|
||
import requests
|
||
|
||
from config import LOLIKEY
|
||
|
||
|
||
class Ans(StdAns):
|
||
AllowGroup = [959613860, 973510746, 805197917, 343700338, 125733077, 1084566280,
|
||
920863253, 798595664, 655057127, 196268763, 247022495, 474907856, 940333876,
|
||
204097403,138966354]
|
||
|
||
def GETMSG(self):
|
||
if self.parms[-1] == 'p':
|
||
flash = ''
|
||
self.parms = self.parms[0:-1]
|
||
else:
|
||
flash = ',type=flash'
|
||
|
||
if len(self.parms) == 1:
|
||
code, picurl, pid = getsetu()
|
||
else:
|
||
keyword = self.parms[1]
|
||
if mysakuya(self, keyword) == False:
|
||
return "不许你们看咲夜的涩图！！"
|
||
code, picurl, pid = getsetu(keyword)
|
||
if code == 0:
|
||
self.sendmsg(f'[CQ:reply,id={self.mid}]Pixiv ID:{pid}')
|
||
return f'[CQ:image,file={picurl}{flash}]'
|
||
else:
|
||
return f'[CQ:reply,id={self.mid}] 什么东西出错了，code:{code}'
|
||
|
||
|
||
def getsetu(keyword=''):
|
||
url = 'https://api.lolicon.app/setu/'
|
||
params = {
|
||
'apikey': LOLIKEY,
|
||
'keyword': keyword,
|
||
}
|
||
|
||
try:
|
||
resp = requests.get(url=url, params=params, timeout=5).json()
|
||
except:
|
||
return 500, '',0
|
||
|
||
|
||
picurl = ''
|
||
pid = 0
|
||
if resp['code'] == 0:
|
||
picurl = "https://r.zjuyk.site/" + resp['data'][0]['url']
|
||
pid = resp['data'][0]['pid']
|
||
|
||
return resp['code'], picurl, pid
|