blob: 52cd93e3fdc44f702d090640f8086108b7a7c0a0 (
plain)
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
|
from random import choice
from nonebot.adapters.onebot.v11 import MessageEvent, Message, MessageSegment
from nonebot.adapters.onebot.v11.helpers import extract_image_urls, Cooldown
from ATRI import conf
from ATRI.log import log
from ATRI.service import Service, ServiceTools
from .data_source import SauceNao
plugin = Service("以图搜图").document("以图搜图，仅限二刺螈")
sau = SauceNao()
_search_flmt_notice = choice(["慢...慢一..点❤", "冷静1下", "歇会歇会~~"])
saucenao = plugin.on_command("以图搜图", "透过一张图搜索可能的来源")
@saucenao.got("saucenao_img", "图呢？", [Cooldown(5, prompt=_search_flmt_notice)])
async def _deal_search(event: MessageEvent):
# cache fix
if not conf.SauceNAO.key:
ServiceTools("以图搜图").service_controller(False)
log.warning("插件 SauceNao 所需的 key 未配置，将被全局禁用，后续填写请手动启用")
user_id = event.get_user_id()
img = extract_image_urls(event.message)
if not img:
await saucenao.reject("请发送图片而不是其他东西！！")
try:
a = SauceNao(conf.SauceNAO.key)
except Exception:
await saucenao.finish("失败了...")
result = f"> {MessageSegment.at(user_id)}" + await a.search(img[0]) # type: ignore
await saucenao.finish(Message(result))
|
