mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 06:12:53 +08:00
242 lines
8.9 KiB
Python
Executable File
242 lines
8.9 KiB
Python
Executable File
from nonebot import on_command
|
||
from nonebot.typing import T_State
|
||
from nonebot.adapters.onebot.v11 import Bot, MessageEvent, GroupMessageEvent, Message
|
||
from .data_source import (
|
||
add_live_sub,
|
||
delete_sub,
|
||
add_up_sub,
|
||
add_season_sub,
|
||
get_media_id,
|
||
get_sub_status,
|
||
SubManager,
|
||
BilibiliSub,
|
||
)
|
||
from models.level_user import LevelUser
|
||
from configs.config import Config
|
||
from utils.utils import is_number, scheduler, get_bot
|
||
from typing import Optional
|
||
from services.log import logger
|
||
from nonebot import Driver
|
||
from nonebot.params import CommandArg, ArgStr
|
||
import nonebot
|
||
|
||
__zx_plugin_name__ = "B站订阅"
|
||
__plugin_usage__ = """
|
||
usage：
|
||
B站直播，番剧，UP动态开播等提醒
|
||
主播订阅相当于 直播间订阅 + UP订阅
|
||
指令：[示例Id乱打的，仅做示例]
|
||
添加订阅 ['主播'/'UP'/'番剧'] [id/链接/番名]
|
||
删除订阅 [id]
|
||
查看订阅
|
||
示例：添加订阅主播 2345344 <-(直播房间id)
|
||
示例：添加订阅UP 2355543 <-(个人主页id)
|
||
示例：添加订阅番剧 史莱姆 <-(支持模糊搜索)
|
||
示例：添加订阅番剧 125344 <-(番剧id)
|
||
示例：删除订阅 2324344 <-(任意id，通过查看订阅获取)
|
||
""".strip()
|
||
__plugin_des__ = "非常便利的B站订阅通知"
|
||
__plugin_cmd__ = ["添加订阅 [主播/UP/番剧] [id/链接/番名]", "删除订阅 [id]", "查看订阅"]
|
||
__plugin_version__ = 0.1
|
||
__plugin_author__ = "HibiKier"
|
||
__plugin_settings__ = {
|
||
"level": 5,
|
||
"default_status": True,
|
||
"limit_superuser": False,
|
||
"cmd": ["B站订阅", "b站订阅", "添加订阅", "删除订阅", "查看订阅"],
|
||
}
|
||
__plugin_configs__ = {
|
||
"GROUP_BILIBILI_SUB_LEVEL": {
|
||
"value": 5,
|
||
"help": "群内bilibili订阅需要管理的权限",
|
||
"default_value": 5,
|
||
}
|
||
}
|
||
|
||
add_sub = on_command("添加订阅", priority=5, block=True)
|
||
del_sub = on_command("删除订阅", priority=5, block=True)
|
||
show_sub_info = on_command("查看订阅", priority=5, block=True)
|
||
|
||
driver: Driver = nonebot.get_driver()
|
||
|
||
|
||
sub_manager: Optional[SubManager] = None
|
||
|
||
|
||
@driver.on_startup
|
||
async def _():
|
||
global sub_manager
|
||
sub_manager = SubManager()
|
||
|
||
|
||
@add_sub.handle()
|
||
async def _(event: MessageEvent, state: T_State, arg: Message = CommandArg()):
|
||
msg = arg.extract_plain_text().strip().split()
|
||
if len(msg) < 2:
|
||
await add_sub.finish("参数不完全，请查看订阅帮助...")
|
||
sub_type = msg[0]
|
||
id_ = ""
|
||
if isinstance(event, GroupMessageEvent):
|
||
if not await LevelUser.check_level(
|
||
event.user_id,
|
||
event.group_id,
|
||
Config.get_config("bilibili_sub", "GROUP_BILIBILI_SUB_LEVEL"),
|
||
):
|
||
await add_sub.finish(
|
||
f"您的权限不足，群内订阅的需要 {Config.get_config('bilibili_sub', 'GROUP_BILIBILI_SUB_LEVEL')} 级权限..",
|
||
at_sender=True,
|
||
)
|
||
sub_user = f"{event.user_id}:{event.group_id}"
|
||
else:
|
||
sub_user = f"{event.user_id}"
|
||
state["sub_type"] = sub_type
|
||
state["sub_user"] = sub_user
|
||
if len(msg) > 1:
|
||
if "http" in msg[1]:
|
||
msg[1] = msg[1].split("?")[0]
|
||
msg[1] = msg[1][:-1] if msg[1][-1] == "/" else msg[1]
|
||
msg[1] = msg[1].split("/")[-1]
|
||
id_ = msg[1][2:] if msg[1].startswith("md") else msg[1]
|
||
if not is_number(id_):
|
||
if sub_type in ["season", "动漫", "番剧"]:
|
||
rst = "*以为您找到以下番剧，请输入Id选择：*\n"
|
||
state["season_data"] = await get_media_id(id_)
|
||
if len(state["season_data"]) == 0:
|
||
await add_sub.finish(f"未找到番剧：{msg}")
|
||
for i, x in enumerate(state["season_data"]):
|
||
rst += f'{i + 1}.{state["season_data"][x]["title"]}\n----------\n'
|
||
await add_sub.send("\n".join(rst.split("\n")[:-1]))
|
||
else:
|
||
await add_sub.finish("Id 必须为全数字！")
|
||
else:
|
||
state["id"] = int(id_)
|
||
|
||
|
||
@add_sub.got("sub_type")
|
||
@add_sub.got("sub_user")
|
||
@add_sub.got("id")
|
||
async def _(
|
||
event: MessageEvent,
|
||
state: T_State,
|
||
id_: str = ArgStr("id"),
|
||
sub_type: str = ArgStr("sub_type"),
|
||
sub_user: str = ArgStr("sub_user"),
|
||
):
|
||
if sub_type in ["season", "动漫", "番剧"] and state.get("season_data"):
|
||
season_data = state["season_data"]
|
||
if not is_number(id_) or int(id_) < 1 or int(id_) > len(season_data):
|
||
await add_sub.reject_arg("id", "Id必须为数字且在范围内！请重新输入...")
|
||
id_ = season_data[int(id_) - 1]["media_id"]
|
||
id_ = int(id_)
|
||
if sub_type in ["主播", "直播"]:
|
||
await add_sub.send(await add_live_sub(id_, sub_user))
|
||
elif sub_type.lower() in ["up", "用户"]:
|
||
await add_sub.send(await add_up_sub(id_, sub_user))
|
||
elif sub_type in ["season", "动漫", "番剧"]:
|
||
await add_sub.send(await add_season_sub(id_, sub_user))
|
||
else:
|
||
await add_sub.finish("参数错误，第一参数必须为：主播/up/番剧！")
|
||
logger.info(
|
||
f"(USER {event.user_id}, GROUP "
|
||
f"{event.group_id if isinstance(event, GroupMessageEvent) else 'private'})"
|
||
f" 添加订阅：{sub_type} -> {sub_user} -> {id_}"
|
||
)
|
||
|
||
|
||
@del_sub.handle()
|
||
async def _(event: MessageEvent, arg: Message = CommandArg()):
|
||
msg = arg.extract_plain_text().strip()
|
||
if not is_number(msg):
|
||
await del_sub.finish("Id必须为数字！", at_sender=True)
|
||
id_ = (
|
||
f"{event.user_id}:{event.group_id}"
|
||
if isinstance(event, GroupMessageEvent)
|
||
else f"{event.user_id}"
|
||
)
|
||
if await BilibiliSub.delete_bilibili_sub(int(msg), id_):
|
||
await del_sub.send(f"删除订阅id：{msg} 成功...")
|
||
logger.info(
|
||
f"(USER {event.user_id}, GROUP "
|
||
f"{event.group_id if isinstance(event, GroupMessageEvent) else 'private'})"
|
||
f" 删除订阅 {id_}"
|
||
)
|
||
else:
|
||
await del_sub.send(f"删除订阅id：{msg} 失败...")
|
||
|
||
|
||
@show_sub_info.handle()
|
||
async def _(event: MessageEvent):
|
||
if isinstance(event, GroupMessageEvent):
|
||
id_ = f"{event.group_id}"
|
||
else:
|
||
id_ = f"{event.user_id}"
|
||
data = await BilibiliSub.get_sub_data(id_)
|
||
live_rst = ""
|
||
up_rst = ""
|
||
season_rst = ""
|
||
for x in data:
|
||
if x.sub_type == "live":
|
||
live_rst += (
|
||
f"\t直播间id：{x.sub_id}\n" f"\t名称：{x.uname}\n" f"------------------\n"
|
||
)
|
||
if x.sub_type == "up":
|
||
up_rst += f"\tUP：{x.uname}\n" f"\tuid：{x.uid}\n" f"------------------\n"
|
||
if x.sub_type == "season":
|
||
season_rst += (
|
||
f"\t番剧id：{x.sub_id}\n"
|
||
f"\t番名：{x.season_name}\n"
|
||
f"\t当前集数：{x.season_current_episode}\n"
|
||
f"------------------\n"
|
||
)
|
||
live_rst = "当前订阅的直播：\n" + live_rst if live_rst else live_rst
|
||
up_rst = "当前订阅的UP：\n" + up_rst if up_rst else up_rst
|
||
season_rst = "当前订阅的番剧：\n" + season_rst if season_rst else season_rst
|
||
if not live_rst and not up_rst and not season_rst:
|
||
live_rst = (
|
||
"该群目前没有任何订阅..." if isinstance(event, GroupMessageEvent) else "您目前没有任何订阅..."
|
||
)
|
||
await show_sub_info.send(live_rst + up_rst + season_rst)
|
||
|
||
|
||
# 推送
|
||
@scheduler.scheduled_job(
|
||
"interval",
|
||
seconds=30,
|
||
)
|
||
async def _():
|
||
bot = get_bot()
|
||
sub = None
|
||
if bot:
|
||
try:
|
||
await sub_manager.reload_sub_data()
|
||
sub = await sub_manager.random_sub_data()
|
||
if sub:
|
||
logger.debug(f"Bilibili订阅开始检测：{sub.sub_id}")
|
||
rst = await get_sub_status(sub.sub_id, sub.sub_type)
|
||
await send_sub_msg(rst, sub, bot)
|
||
if sub.sub_type == "live":
|
||
rst = await get_sub_status(sub.sub_id, "up")
|
||
await send_sub_msg(rst, sub, bot)
|
||
except Exception as e:
|
||
logger.error(f"B站订阅推送发生错误 sub_id：{sub.sub_id if sub else 0} {type(e)}：{e}")
|
||
|
||
|
||
async def send_sub_msg(rst: str, sub: BilibiliSub, bot: Bot):
|
||
"""
|
||
推送信息
|
||
:param rst: 回复
|
||
:param sub: BilibiliSub
|
||
:param bot: Bot
|
||
"""
|
||
if rst:
|
||
for x in sub.sub_users.split(",")[:-1]:
|
||
try:
|
||
if ":" in x:
|
||
await bot.send_group_msg(
|
||
group_id=int(x.split(":")[1]), message=Message(rst)
|
||
)
|
||
else:
|
||
await bot.send_private_msg(user_id=int(x), message=Message(rst))
|
||
except Exception as e:
|
||
logger.error(f"B站订阅推送发生错误 sub_id：{sub.sub_id} {type(e)}：{e}")
|