48 lines
1.4 KiB
Python
48 lines
1.4 KiB
Python
from worker import emmm
|
|
from zzcore import DM
|
|
from config import DAAPI
|
|
|
|
import requests
|
|
|
|
|
|
def ZZRouter(data):
|
|
|
|
if data['post_type'] != 'message' or data['message_type'] != 'group' or data['message'][0] != '/':
|
|
if data['user_id'] == 1318000868 :
|
|
return
|
|
if len(data['raw_message']) == 54 and data['raw_message'][:15] == '[CQ:image,file=': # use re
|
|
url = data['message'][58:][:-1]
|
|
r = requests.post(
|
|
"https://api.deepai.org/api/image-similarity",
|
|
data={
|
|
'image1': 'https://img.vim-cn.com/fa/4a7f3996e3601e98bd3c1b245fcb88e05f32ec.jpg',
|
|
'image2': url,
|
|
},
|
|
headers={'api-key': DAAPI}
|
|
)
|
|
if r.json()['output']['distance'] < 10:
|
|
DM(data['message_id'])
|
|
return
|
|
|
|
uid = data['user_id']
|
|
gid = data['group_id']
|
|
role = data['sender']['role']
|
|
parms = str.split(data['message'][1:])
|
|
print(uid, gid, role, parms)
|
|
if parms == []:
|
|
parms[0] = 'help'
|
|
worker = parms[0]
|
|
|
|
try:
|
|
package = __import__(name='worker.' + worker, fromlist=worker)
|
|
Ans_ = getattr(package, 'Ans')
|
|
except:
|
|
Ans_ = emmm.Ans
|
|
|
|
Ans = Ans_(parms, uid, gid, role, data)
|
|
Message = Ans.CheckPermission()
|
|
if Message == 0:
|
|
Message = Ans.GETMSG()
|
|
elif Message == -1:
|
|
return
|
|
Ans.sendmsg(Message)
|