feat(video-analysis): 多文件媒体改走合并转发

抖音 note/小红书等多图、图+视频混合、多视频(动图)作品,下载产物
超过一个文件时直接拼成一条消息发送会失败(视频段不能与其他段混合),
统一改为合并转发: 一个图/视频一个节点,本地文件优先、S3 预签名链接
兜底; 单文件作品保持原有直发链路,多级发送链(本地→S3→回退)不变。

Co-Authored-By: Claude Code <noreply@anthropic.com>
This commit is contained in:
2026-09-08 14:21:57 +08:00
co-authored by Claude Code
parent 8c1a106230
commit 30899688a7
2 changed files with 140 additions and 10 deletions
@@ -4,7 +4,6 @@ from __future__ import annotations
import html
import re
from typing import Optional
from nonebot import on_message, logger
from nonebot.adapters import Event
@@ -151,7 +150,7 @@ async def match_message(event: Event, plan: str | None = None):
return
if isinstance(message, PendingMedia):
ok, pub = await send_pending_media(message)
ok, pub = await send_pending_media(message, event)
if pub:
await UniMessage.text(f"{pub}").send()
if not ok:
@@ -166,7 +165,7 @@ async def dispatch_url(
url: str,
is_private: bool,
plan: str | None = None,
) -> tuple[Optional[UniMessage], Optional[str]]:
) -> tuple[UniMessage | None, str | None]:
"""按平台分派解析(文本链接与小程序卡片共用)。"""
url = url.rstrip(",。!?、;:)】》\"')")
@@ -246,7 +245,7 @@ async def dispatch_url(
return None, None
async def _extract_xcx_url(msg: str) -> Optional[str]:
async def _extract_xcx_url(msg: str) -> str | None:
"""从 CQ 卡片消息中提取跳转 URL(保留 query 参数)。"""
match = re.search(r'"qqdocurl":"(.*?)"', msg)
if not match:
@@ -5,14 +5,19 @@
2. 失败 → 上传本地 S3,用预签名链接发送
3. 再失败 → 回退 temp 本地文件再发一次
多媒体(2026-09-08 用户需求): 多图(图集)、图+视频混合、多视频(动图)
等一个作品落地成多个文件时,直接 UniMessage 拼成一个消息发送会报错
(视频段不能与其他段混合,一条消息也放不下多段视频/图集体验差)
→ 统一走合并转发,一个图/视频一个节点。
temp 下的文件发送成功后也不清理(用户手动处理 data/temp)。
"""
from dataclasses import dataclass
from pathlib import Path
from typing import Optional, Union
from nonebot import logger
from nonebot import get_bot, get_driver, logger
from nonebot.adapters import Event
from nonebot_plugin_alconna import UniMessage
from ..services.storage.s3 import upload_with_plan
@@ -25,16 +30,111 @@ class PendingMedia:
files: list[Path]
image_post: bool = False
is_private: bool = False
plan: Optional[str] = None
plan: str | None = None
title: str = ""
def _as_paths(file_paths: Union[Path, list[Path]]) -> list[Path]:
def _as_paths(file_paths: Path | list[Path]) -> list[Path]:
if isinstance(file_paths, list):
return [Path(p) for p in file_paths]
return [Path(file_paths)]
# ─────────────────────── 合并转发(多媒体专用) ───────────────────────
def _is_video(fp: Path) -> bool:
return fp.suffix.lower() == ".mp4"
def _needs_forward(files: list[Path]) -> bool:
"""是否需要走合并转发:多图/图+视频混合/多视频(动图)等多媒体作品
落地成多个文件(>1 个)统一一个媒体一个节点转发;
单图/单视频保持原有直发链路。
"""
return len(files) > 1
def _forward_sender_name() -> str:
"""合并转发节点的发送者昵称(用机器人配置昵称)"""
nickname = get_driver().config.nickname
return next(iter(nickname), "Bot") if nickname else "Bot"
def _media_segment(fp: Path, url: str | None = None) -> dict:
"""onebot v11 消息段字典:image / video,file 为本地路径或链接"""
media_type = "video" if _is_video(fp) else "image"
return {"type": media_type, "data": {"file": url or str(fp)}}
def _build_forward_nodes(
items: list[tuple[Path, str | None]],
sender_name: str,
self_id: int,
) -> list[dict]:
"""合并转发节点列表:一个媒体一个节点,content 为该媒体的消息段数组"""
return [
{
"type": "node",
"data": {
"name": sender_name,
"uin": str(self_id),
"content": [_media_segment(fp, url)],
},
}
for fp, url in items
]
async def _send_forward_media(
items: list[tuple[Path, str | None]],
sender_name: str,
*,
group_id: int | None = None,
user_id: int | None = None,
) -> bool:
"""发送合并转发消息(群聊 / 私聊),失败返回 False"""
try:
bot = get_bot()
nodes = _build_forward_nodes(items, sender_name, bot.self_id)
if group_id is not None:
await bot.call_api(
"send_group_forward_msg", group_id=group_id, messages=nodes
)
else:
await bot.call_api("send_forward_msg", user_id=user_id, messages=nodes)
return True
except Exception as e:
logger.warning(f"合并转发发送失败: {e}")
return False
async def _build_s3_forward_items(
media: PendingMedia,
) -> tuple[list[tuple[Path, str | None]], str | None]:
"""上传本地 S3,返回 ([(文件, 预签名链接)...], public_url)"""
items: list[tuple[Path, str | None]] = []
public_url = None
for fp in media.files:
local_url, pub = upload_with_plan(
fp,
plan=media.plan,
is_private=media.is_private,
title=media.title,
image_post=media.image_post,
)
if not local_url:
raise RuntimeError(f"上传本地 S3 失败: {fp}")
if pub:
public_url = pub
items.append((fp, local_url))
return items, public_url
# ───────────────────────────── 多级发送 ─────────────────────────────
def _build_local_msg(files: list[Path], image_post: bool) -> UniMessage:
"""本地文件版消息(mp4 → 视频,其余 → 图片)"""
msg = UniMessage()
@@ -48,7 +148,7 @@ def _build_local_msg(files: list[Path], image_post: bool) -> UniMessage:
def _build_s3_msg(
media: PendingMedia,
) -> tuple[UniMessage, Optional[str]]:
) -> tuple[UniMessage, str | None]:
"""上传本地 S3 并构建链接版消息,返回 (message, public_url)"""
msg = UniMessage()
public_url = None
@@ -71,15 +171,46 @@ def _build_s3_msg(
return msg, public_url
async def send_pending_media(media: PendingMedia) -> tuple[bool, Optional[str]]:
async def send_pending_media(
media: PendingMedia, event: Event | None = None
) -> tuple[bool, str | None]:
"""多级发送,返回 (是否成功, public_url)
public_url 仅在走 S3 链接发送成功时返回(调用方决定是否发文字)。
temp 文件发送成功后保留(用户手动清理 data/temp)。
多媒体作品(多图 / 图+视频 / 多视频)先走合并转发(本地文件 → S3 链接),
失败后仍落回下方普通消息链路兜底。
"""
if not media.files:
return False, None
# ── 0. 多媒体 → 合并转发(一个图/视频一个节点)─────────────────
if event is not None and _needs_forward(media.files):
group_id = getattr(event, "group_id", None)
user_id = None if group_id is not None else getattr(event, "user_id", None)
if group_id is not None or user_id is not None:
sender_name = _forward_sender_name()
if await _send_forward_media(
[(fp, None) for fp in media.files],
sender_name,
group_id=group_id,
user_id=user_id,
):
logger.info("媒体发送成功(合并转发,本地文件)")
return True, None
try:
s3_items, public_url = await _build_s3_forward_items(media)
if await _send_forward_media(
s3_items, sender_name, group_id=group_id, user_id=user_id
):
logger.info("媒体发送成功(合并转发,S3 链接)")
return True, public_url
except Exception as e:
logger.warning(f"S3 链接合并转发失败,走普通消息链路: {e}")
else:
logger.warning("无法确定群号/用户号,跳过合并转发,走普通消息链路")
# ── 1. temp 本地文件直接发送 ──────────────────────────────
try:
await _build_local_msg(media.files, media.image_post).send()