1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
|
import re
import json
from random import choice
from nonebot.adapters.cqhttp import Bot, MessageEvent
from nonebot.adapters.cqhttp.message import Message, MessageSegment
from nonebot.typing import T_State
from ATRI.config import Config
from ATRI.service import Service as sv
from ATRI.rule import is_in_service
from ATRI.exceptions import RequestTimeOut
from .data_source import SauceNao
__doc__ = """
以图搜图
权限组：所有人
用法：
以图搜图 (pic)
"""
saucenao = sv.on_command(
cmd='以图搜图',
docs=__doc__,
rule=is_in_service('以图搜图')
)
@saucenao.args_parser # type: ignore
async def _load_saucenao(bot: Bot, event: MessageEvent,
state: T_State) -> None:
msg = str(event.message)
quit_list = ['算了', '罢了', '不搜了']
if msg in quit_list:
await saucenao.finish('好吧...')
if not msg:
await saucenao.reject('图呢？')
else:
state['pic_sau'] = msg
@saucenao.handle()
async def _sauce_nao(bot: Bot, event: MessageEvent, state: T_State) -> None:
msg = str(event.message).strip()
if msg:
state['pic_sau'] = msg
@saucenao.got('pic_sau', prompt='图呢？')
async def _deal_saucenao(bot: Bot, event: MessageEvent, state: T_State) -> None:
msg = state['pic_sau']
img = re.findall(r'url=(.*?)]', msg)
if not img:
await saucenao.finish('请发送图片而不是其他东西！！')
try:
task = SauceNao(api_key=Config.SauceNAO.key)
data = json.loads(await task.search(img[0]))
except RequestTimeOut:
raise RequestTimeOut('Request failed!')
res = data['results']
result = list()
for i in range(0, 3):
data = res[i]
_result = dict()
_result['similarity'] = data['header']['similarity']
_result['index_name'] = data['header']['index_name']
_result['url'] = choice(data['data'].get('ext_urls', ['None']))
result.append(_result)
msg0 = f"> {MessageSegment.at(event.user_id)}"
for i in result:
msg0 = msg0 + (
"\n——————————\n"
f"Similarity: {i['similarity']}\n"
f"Name: {i['index_name']}\n"
f"URL: {i['url'].replace('https://', '')}"
)
await saucenao.finish(Message(msg0))
|
