mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-10-03 19:00:00 +08:00
feat✨: 添加B站转发解析功能
This commit is contained in:
HibiKier
@@ -0,0 +1,57 @@
|
||||
from nonebot import on_message
|
||||
from nonebot.plugin import PluginMetadata
|
||||
from nonebot_plugin_alconna import UniMsg
|
||||
from nonebot_plugin_session import EventSession
|
||||
|
||||
from zhenxun.configs.utils import PluginExtraData, RegisterConfig, Task
|
||||
from zhenxun.models.group_console import GroupConsole
|
||||
from zhenxun.models.task_info import TaskInfo
|
||||
from zhenxun.services.log import logger
|
||||
|
||||
from .data_source import Parser
|
||||
|
||||
__plugin_meta__ = PluginMetadata(
|
||||
name="B站转发解析",
|
||||
description="B站转发解析",
|
||||
usage="""
|
||||
usage：
|
||||
B站转发解析，解析b站分享信息，支持bv，bilibili链接，b站手机端转发卡片，cv，b23.tv，且30秒内内不解析相同url
|
||||
""".strip(),
|
||||
extra=PluginExtraData(
|
||||
author="HibiKier",
|
||||
version="0.1",
|
||||
menu_type="其他",
|
||||
configs=[
|
||||
RegisterConfig(
|
||||
module="_task",
|
||||
key="DEFAULT_BILIBILI_PARSE",
|
||||
value=True,
|
||||
default_value=True,
|
||||
help="被动 B站转发解析 进群默认开关状态",
|
||||
type=bool,
|
||||
)
|
||||
],
|
||||
tasks=[Task(module="bilibili_parse", name="b站转发解析")],
|
||||
).dict(),
|
||||
)
|
||||
|
||||
|
||||
async def _rule(session: EventSession) -> bool:
|
||||
task = await TaskInfo.get_or_none(module="bilibili_parse")
|
||||
if not task or not task.status:
|
||||
logger.debug("b站转发解析被动全局关闭，已跳过...")
|
||||
return False
|
||||
if gid := session.id3 or session.id2:
|
||||
return not await GroupConsole.is_block_task(gid, "bilibili_parse")
|
||||
return False
|
||||
|
||||
|
||||
_matcher = on_message(priority=1, block=False, rule=_rule)
|
||||
|
||||
|
||||
@_matcher.handle()
|
||||
async def _(session: EventSession, message: UniMsg):
|
||||
data = message[0]
|
||||
if result := await Parser.parse(data, message.extract_plain_text().strip()):
|
||||
await result.send()
|
||||
logger.info(f"b站转发解析: {result}", "BILIBILI_PARSE", session=session)
|
||||
@@ -0,0 +1,186 @@
|
||||
import re
|
||||
import time
|
||||
import uuid
|
||||
from pathlib import Path
|
||||
from typing import Any
|
||||
|
||||
import aiohttp
|
||||
import ujson as json
|
||||
from bilireq import video
|
||||
from nonebot_plugin_alconna import Hyper
|
||||
from nonebot_plugin_saa import Image, MessageFactory, Text
|
||||
|
||||
from zhenxun.configs.path_config import TEMP_PATH
|
||||
from zhenxun.services.log import logger
|
||||
from zhenxun.utils.http_utils import AsyncPlaywright
|
||||
from zhenxun.utils.user_agent import get_user_agent
|
||||
|
||||
|
||||
class Parser:
|
||||
|
||||
time_watch: dict[str, float] = {}
|
||||
|
||||
@classmethod
|
||||
async def parse(cls, data: Any, raw: str | None = None) -> MessageFactory | None:
|
||||
"""解析
|
||||
|
||||
参数:
|
||||
data: data数据
|
||||
raw: 文本.
|
||||
|
||||
返回:
|
||||
MessageFactory | None: 返回信息
|
||||
"""
|
||||
if isinstance(data, Hyper) and data.raw:
|
||||
json_data = json.loads(data.raw)
|
||||
if video_info := await cls.__parse_video_share(json_data):
|
||||
return await cls.__handle_video_info(video_info)
|
||||
if path := await cls.__parse_news_share(json_data):
|
||||
return MessageFactory([Image(path)])
|
||||
if raw:
|
||||
return await cls.__search(raw)
|
||||
return None
|
||||
|
||||
@classmethod
|
||||
async def __search(cls, message: str) -> MessageFactory | None:
|
||||
"""根据bv，av，链接获取视频信息
|
||||
|
||||
参数:
|
||||
message: 文本内容
|
||||
|
||||
返回:
|
||||
MessageFactory | None: 返回信息
|
||||
"""
|
||||
if "BV" in message:
|
||||
index = message.find("BV")
|
||||
if len(message[index + 2 :]) >= 10:
|
||||
msg = message[index : index + 12]
|
||||
url = f"https://www.bilibili.com/video/{msg}"
|
||||
return await cls.__handle_video_info(
|
||||
await video.get_video_base_info(msg), url
|
||||
)
|
||||
elif "av" in message:
|
||||
index = message.find("av")
|
||||
if len(message[index + 2 :]) >= 1:
|
||||
if r := re.search(r"av(\d+)", message):
|
||||
url = f"https://www.bilibili.com/video/av{r.group(1)}"
|
||||
return await cls.__handle_video_info(
|
||||
await video.get_video_base_info(f"av{r.group(1)}"), url
|
||||
)
|
||||
elif "https://b23.tv" in message:
|
||||
url = (
|
||||
"https://"
|
||||
+ message[message.find("b23.tv") : message.find("b23.tv") + 14]
|
||||
)
|
||||
async with aiohttp.ClientSession(headers=get_user_agent()) as session:
|
||||
async with session.get(
|
||||
url,
|
||||
timeout=7,
|
||||
) as response:
|
||||
url = (str(response.url).split("?")[0]).strip("/")
|
||||
bvid = url.split("/")[-1]
|
||||
return await cls.__handle_video_info(
|
||||
await video.get_video_base_info(bvid), url
|
||||
)
|
||||
return None
|
||||
|
||||
@classmethod
|
||||
async def __handle_video_info(
|
||||
cls, vd_info: dict, url: str = ""
|
||||
) -> MessageFactory | None:
|
||||
"""处理视频信息
|
||||
|
||||
参数:
|
||||
vd_info: 视频数据
|
||||
url: 视频url.
|
||||
|
||||
返回:
|
||||
MessageFactory | None: 返回信息
|
||||
"""
|
||||
if url:
|
||||
if url in cls.time_watch.keys() and time.time() - cls.time_watch[url] < 30:
|
||||
logger.debug("b站 url 解析在30秒内重复， 跳过解析...")
|
||||
return None
|
||||
cls.time_watch[url] = time.time()
|
||||
aid = vd_info["aid"]
|
||||
title = vd_info["title"]
|
||||
author = vd_info["owner"]["name"]
|
||||
reply = vd_info["stat"]["reply"] # 回复
|
||||
favorite = vd_info["stat"]["favorite"] # 收藏
|
||||
coin = vd_info["stat"]["coin"] # 投币
|
||||
# like = vd_info['stat']['like'] # 点赞
|
||||
# danmu = vd_info['stat']['danmaku'] # 弹幕
|
||||
date = time.strftime("%Y-%m-%d", time.localtime(vd_info["ctime"]))
|
||||
return MessageFactory(
|
||||
[
|
||||
Image(vd_info["pic"]),
|
||||
Text(
|
||||
f"\nav{aid}\n标题：{title}\nUP：{author}\n上传日期：{date}\n回复：{reply}，收藏：{favorite}，投币：{coin}\n{url}"
|
||||
),
|
||||
]
|
||||
)
|
||||
|
||||
@classmethod
|
||||
async def __parse_video_share(cls, data: dict) -> dict | None:
|
||||
"""解析视频转发
|
||||
|
||||
参数:
|
||||
data: data数据
|
||||
|
||||
返回:
|
||||
dict | None: 视频信息
|
||||
"""
|
||||
try:
|
||||
if data["meta"]["detail_1"]["title"] == "哔哩哔哩":
|
||||
try:
|
||||
async with aiohttp.ClientSession(
|
||||
headers=get_user_agent()
|
||||
) as session:
|
||||
async with session.get(
|
||||
data["meta"]["detail_1"]["qqdocurl"],
|
||||
timeout=7,
|
||||
) as response:
|
||||
url = str(response.url).split("?")[0]
|
||||
if url[-1] == "/":
|
||||
url = url[:-1]
|
||||
bvid = url.split("/")[-1]
|
||||
return await video.get_video_base_info(bvid)
|
||||
except Exception as e:
|
||||
logger.warning("解析b站视频失败", e=e)
|
||||
except Exception as e:
|
||||
pass
|
||||
return None
|
||||
|
||||
@classmethod
|
||||
async def __parse_news_share(cls, data: dict) -> Path | None:
|
||||
"""解析b站专栏
|
||||
|
||||
参数:
|
||||
data: data数据
|
||||
|
||||
返回:
|
||||
Path | None: 截图路径
|
||||
"""
|
||||
try:
|
||||
if data["meta"]["news"]["desc"] == "哔哩哔哩专栏":
|
||||
try:
|
||||
url = data["meta"]["news"]["jumpUrl"]
|
||||
async with AsyncPlaywright.new_page() as page:
|
||||
await page.goto(url, wait_until="networkidle", timeout=10000)
|
||||
await page.set_viewport_size({"width": 2560, "height": 1080})
|
||||
try:
|
||||
await page.locator("div.bili-mini-close-icon").click()
|
||||
except Exception:
|
||||
pass
|
||||
if div := await page.query_selector("#app > div"):
|
||||
path = TEMP_PATH / f"bl_share_{uuid.uuid1()}.png"
|
||||
await div.screenshot(
|
||||
path=path,
|
||||
timeout=100000,
|
||||
)
|
||||
return path
|
||||
except Exception as e:
|
||||
logger.warning("解析b站专栏失败", e=e)
|
||||
except Exception as e:
|
||||
pass
|
||||
return None
|
||||
Reference in New Issue
Block a user