mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
64 lines
2.1 KiB
Python
64 lines
2.1 KiB
Python
|
import random
|
|||
|
|
|
|||
|
|
from nonebot_plugin_saa import Image, Text
|
|||
|
|
|
|||
|
|
from zhenxun.configs.config import Config
|
|||
|
|
from zhenxun.configs.path_config import TEMP_PATH
|
|||
|
|
from zhenxun.services import logger
|
|||
|
|
from zhenxun.utils.http_utils import AsyncHttpx
|
|||
|
|
|
|||
|
|
API_URL_SAUCENAO = "https://saucenao.com/search.php"
|
|||
|
|
API_URL_ASCII2D = "https://ascii2d.net/search/url/"
|
|||
|
|
API_URL_IQDB = "https://iqdb.org/"
|
|||
|
|
|
|||
|
|
|
|||
|
|
async def get_saucenao_image(url: str) -> str | list[Image | Text]:
|
|||
|
|
"""获取图片源
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
url: 图片url
|
|||
|
|
|
|||
|
|
返回:
|
|||
|
|
str | list[Image | Text]: 识图数据
|
|||
|
|
"""
|
|||
|
|
api_key = Config.get_config("search_image", "API_KEY")
|
|||
|
|
if not api_key:
|
|||
|
|
return "Saucenao 缺失API_KEY！"
|
|||
|
|
params = {
|
|||
|
|
"output_type": 2,
|
|||
|
|
"api_key": api_key,
|
|||
|
|
"testmode": 1,
|
|||
|
|
"numres": 6,
|
|||
|
|
"db": 999,
|
|||
|
|
"url": url,
|
|||
|
|
}
|
|||
|
|
data = (await AsyncHttpx.post(API_URL_SAUCENAO, params=params)).json()
|
|||
|
|
if data["header"]["status"] != 0:
|
|||
|
|
return f"Saucenao识图失败..status：{data['header']['status']}"
|
|||
|
|
data = data["results"]
|
|||
|
|
data = (
|
|||
|
|
data
|
|||
|
|
if len(data) < Config.get_config("search_image", "MAX_FIND_IMAGE_COUNT")
|
|||
|
|
else data[: Config.get_config("search_image", "MAX_FIND_IMAGE_COUNT")]
|
|||
|
|
)
|
|||
|
|
msg_list = []
|
|||
|
|
index = random.randint(0, 10000)
|
|||
|
|
if await AsyncHttpx.download_file(url, TEMP_PATH / f"saucenao_search_{index}.jpg"):
|
|||
|
|
msg_list.append(Image(TEMP_PATH / f"saucenao_search_{index}.jpg"))
|
|||
|
|
for info in data:
|
|||
|
|
try:
|
|||
|
|
similarity = info["header"]["similarity"]
|
|||
|
|
tmp = f"相似度：{similarity}%\n"
|
|||
|
|
for x in info["data"].keys():
|
|||
|
|
if x != "ext_urls":
|
|||
|
|
tmp += f"{x}：{info['data'][x]}\n"
|
|||
|
|
try:
|
|||
|
|
if "source" not in info["data"].keys():
|
|||
|
|
tmp += f'source：{info["data"]["ext_urls"][0]}\n'
|
|||
|
|
except KeyError:
|
|||
|
|
tmp += f'source：{info["header"]["thumbnail"]}\n'
|
|||
|
|
msg_list.append(Text(tmp[:-1]))
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.warning(f"识图获取图片信息发生错误", e=e)
|
|||
|
|
return msg_list
|