mirror of
https://github.com/ClovertaTheTrilobita/SanYeCao-Nonebot.git
synced 2026-04-01 22:04:51 +00:00
5ae3d623a8
feat(animetrace): 新增animetrace_search_by_url功能用于搜番 feat(image): 添加搜番功能并集成animetrace_search_by_url chore(check): 更新菜单命令列表，添加搜番功能
67 lines
2.4 KiB
Python
67 lines
2.4 KiB
Python
import random
|
|
from pathlib import Path
|
|
from nonebot.rule import to_me
|
|
from nonebot.plugin import on_command
|
|
from nonebot.adapters.qq import MessageSegment,MessageEvent
|
|
from src.clover_image.get_image import get_image_names,get_anosu_image
|
|
from src.clover_image.download_image import download_image
|
|
from src.clover_image.animetrace import animetrace_search_by_url
|
|
from src.clover_image.delete_file import delete_file
|
|
from src.configs.path_config import temp_path
|
|
|
|
|
|
image = on_command("图", rule=to_me(), priority=10,block=False)
|
|
@image.handle()
|
|
async def handle_function():
|
|
|
|
local_image_path = await get_image_names()
|
|
await image.finish(MessageSegment.file_image(Path(local_image_path)))
|
|
|
|
|
|
|
|
random_keyword_image = on_command("随机图", rule=to_me(), priority=10, block=False)
|
|
@random_keyword_image.handle()
|
|
async def handle_function(message: MessageEvent):
|
|
|
|
values = message.get_plaintext().replace("/随机图", "").split(" ")
|
|
keyword,is_r18,num = "",0,1
|
|
|
|
for value in values:
|
|
if value.isdigit():
|
|
num = int(value)
|
|
elif value.lower() == "r18":
|
|
is_r18 = 1
|
|
else:
|
|
keyword = value
|
|
urls = await get_anosu_image(keyword=keyword,is_r18=is_r18,num=num)
|
|
file_paths = []
|
|
for url in urls:
|
|
filename = f"{message.get_user_id()}{random.randint(0, 10000)}.jpg"
|
|
image_path = temp_path + filename
|
|
file_paths.append(image_path)
|
|
await download_image(url,image_path)
|
|
|
|
try:
|
|
for file_path in file_paths:
|
|
try:
|
|
await random_keyword_image.send(MessageSegment.file_image(Path(file_path)))
|
|
except Exception as e:
|
|
print(f"发送文件 {file_path} 时出错: {e}")
|
|
await random_keyword_image.send("某个图被外星人抢走啦，请重试")
|
|
finally:
|
|
# 删除所有临时文件
|
|
for file_path in file_paths:
|
|
await delete_file(file_path)
|
|
|
|
|
|
|
|
search_image = on_command("搜番", rule=to_me(), priority=10, block=False)
|
|
@search_image.handle()
|
|
async def handle_function(message: MessageEvent):
|
|
if not message.attachments:
|
|
await search_image.finish("没有图片捏，你想让我搜什么捏？")
|
|
result = await animetrace_search_by_url(message.attachments[0].url)
|
|
if result is None:
|
|
await search_image.finish("未找到结果")
|
|
await search_image.finish(MessageSegment.text('\n 搜索结果\n'+result))
|
|
|