Diuse/zhenxun_bot
1
0
Fork 0
mirror of https://github.com/zhenxun-org/zhenxun_bot.git synced 2026-10-10 06:10:01 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

update v0.0.5.0

Browse Source
This commit is contained in:
HibiKier
2021-10-03 14:24:07 +08:00
parent 87ed445a58
commit 00658cf165
182 changed files with 6439 additions and 3845 deletions
Download Patch File Download Diff File Expand all files Collapse all files
plugins/send_setu/__init__.py
+51 -28
View File
@@ -32,7 +32,7 @@ from .data_source import (
)
from nonebot.adapters.cqhttp.exception import ActionFailed
from configs.config import ONLY_USE_LOCAL_SETU, WITHDRAW_SETU_TIME, NICKNAME
from utils.static_data import withdraw_message_id_manager
from utils.manager import withdraw_message_manager
import re
try:
@@ -40,26 +40,38 @@ try:
except ModuleNotFoundError:
import json
__plugin_name__ = "色图"
__plugin_usage__ = f"""示例：
1. 色图 （随机本地色图）
2. 色图r （随机在线十张r18涩图）
3. 色图 666 （本地色图id）
4. 色图 xx xx （在线搜索xx xx色图）
5. 色图r xx （搜索十张xx的r18涩图，注意空格）（仅私聊，每日限制5次）
6. 来n张涩图 （本地涩图连发）（1<=n<=9）
7. 来n张xx的涩图 （在线搜索xx涩图）（较慢，看网速）
注：
xx 为 tag，多余20取前20 示例：原神 黑丝
本地涩图没有r18！
联网搜索会较慢！
如果图片数量与数字不符，
原因1：网络不好，网线被拔QAQ
原因2：搜索到的总数小于数字
原因3：图太色或者小错误了】
示例：
色图 萝莉|少女 白丝|黑丝
色图 萝莉 猫娘"""
__zx_plugin_name__ = "色图"
__plugin_usage__ = f"""
usage：
搜索 lolicon 图库，每日色图time...
指令：
色图: 随机本地色图
色图r: 随机在线十张r18涩图
色图 [id]: 本地指定id色图
色图 *[tags]: 在线搜索指定tag色图
色图r *[tags]: 同上
[1-9]张涩图: 本地随机色图连发
[1-9]张[tags]的涩图: 指定tag色图连发
示例：色图 萝莉|少女 白丝|黑丝
示例：色图 萝莉 猫娘
注：
tag至多取前20项，| 为或，萝莉|少女=萝莉或者少女
""".strip()
__plugin_des__ = "不要小看涩图啊混蛋！"
__plugin_cmd__ = ["色图 ?[id]", "色图 ?[tags]", "色图r ?[tags]", "[1-9]张?[tags]色图"]
__plugin_type__ = ("来点好康的",)
__plugin_version__ = 0.1
__plugin_author__ = "HibiKier"
__plugin_settings__ = {
"level": 9,
"default_status": True,
"limit_superuser": False,
"cmd": ["色图", "涩图", "瑟图"],
}
__plugin_block_limit__ = {}
__plugin_cd_limit__ = {
'rst': '您冲的太快了，请稍后再冲.',
}
setu_data_list = []
@@ -181,7 +193,7 @@ async def _(bot: Bot, event: MessageEvent, state: T_State):
num = 1
else:
return
await send_setu_handle(setu_reg, event, '色图', tags, num, 0)
await send_setu_handle(setu_reg, event, "色图", tags, num, 0)
@find_setu.args_parser
@@ -210,7 +222,12 @@ async def _(bot: Bot, event: MessageEvent, state: T_State):
async def send_setu_handle(
matcher: Type[Matcher], event: MessageEvent, command: str, msg: str, num: int, r18: int
matcher: Type[Matcher],
event: MessageEvent,
command: str,
msg: str,
num: int,
r18: int,
):
global setu_data_list
# 非 id，在线搜索
@@ -223,7 +240,9 @@ async def send_setu_handle(
msg_id = None
if not ONLY_USE_LOCAL_SETU and tags:
# 先尝试获取在线图片
urls, text_list, add_databases_list, code = await get_setu_urls(tags, num, r18, command)
urls, text_list, add_databases_list, code = await get_setu_urls(
tags, num, r18, command
)
for x in add_databases_list:
setu_data_list.append(x)
# 未找到符合的色图，想来本地应该也没有
@@ -240,7 +259,9 @@ async def send_setu_handle(
f"{event.group_id if isinstance(event, GroupMessageEvent) else 'private'})"
f" 发送色图 {index}.png"
)
msg_id = await matcher.send(Message(f"{text_list[i]}\n{setu_img}"))
msg_id = await matcher.send(
Message(f"{text_list[i]}\n{setu_img}")
)
else:
if setu_list is None:
setu_list, _ = await get_setu_list(tags=tags, r18=r18)
@@ -250,7 +271,9 @@ async def send_setu_handle(
msg_id = await matcher.send(
Message(
gen_message(setu_image)
+ (await check_local_exists_or_download(setu_image))[0]
+ (
await check_local_exists_or_download(setu_image)
)[0]
)
)
logger.info(
@@ -267,7 +290,7 @@ async def send_setu_handle(
return
# 本地无图 或 超过上下限
if code != 200:
await matcher.finish('网络连接失败...', at_sender=True)
await matcher.finish("网络连接失败...", at_sender=True)
if setu_list is None:
setu_list, code = await get_setu_list(tags=tags, r18=r18)
if code != 200:
@@ -303,4 +326,4 @@ def withdraw_message(event: MessageEvent, id_: int):
or (WITHDRAW_SETU_TIME[1] == 1 and isinstance(event, GroupMessageEvent))
or WITHDRAW_SETU_TIME[1] == 2
):
withdraw_message_id_manager['message_id'].append((id_, WITHDRAW_SETU_TIME[0]))
withdraw_message_manager.append((id_, WITHDRAW_SETU_TIME[0]))
plugins/send_setu/data_source.py
+19 -9
View File
@@ -7,7 +7,7 @@ from asyncpg.exceptions import UniqueViolationError
from utils.utils import get_local_proxy
from asyncio.exceptions import TimeoutError
from typing import List, Optional
from configs.config import INITIAL_SETU_PROBABILITY, NICKNAME
from configs.config import INITIAL_SETU_PROBABILITY, NICKNAME, DOWNLOAD_SETU
from models.setu import Setu
import aiohttp
import aiofiles
@@ -80,8 +80,14 @@ headers = {
async def search_online_setu(
url_: str, id_: int = None, path_: str = None
url_: str, id_: Optional[int] = None, path_: Optional[str] = None
) -> "MessageSegment, int":
"""
下载色图
:param url_: 色图url
:param id_: 本地id
:param path_: 存储路径
"""
if "i.pixiv.cat" in url_:
url_ = url_.replace("i.pixiv.cat", "i.pximg.net")
async with aiohttp.ClientSession(headers=headers) as session:
@@ -122,13 +128,17 @@ async def search_online_setu(
# 检测本地是否有id涩图，无的话则下载
async def check_local_exists_or_download(setu_image: Setu) -> "MessageSegment, int":
if setu_image.is_r18:
path_ = "_r18"
else:
path_ = path
if os.path.exists(f"{IMAGE_PATH}/{path_}/{setu_image.local_id}.jpg"):
return image(f"{setu_image.local_id}.jpg", path_), 200
return await search_online_setu(setu_image.img_url, setu_image.local_id, path_)
path_ = None
id_ = None
if DOWNLOAD_SETU:
id_ = setu_image.local_id
if setu_image.is_r18:
path_ = "_r18"
else:
path_ = path
if os.path.exists(f"{IMAGE_PATH}/{path_}/{setu_image.local_id}.jpg"):
return image(f"{setu_image.local_id}.jpg", path_), 200
return await search_online_setu(setu_image.img_url, id_, path_)
# 添加涩图数据到数据库