mirror of
https://github.com/JefferyHcool/BiliNote.git
synced 2026-05-06 20:42:52 +08:00
c65de4654f
- 实现本地视频上传和处理功能 - 新增 LocalDownloader 类处理本地视频 - 更新前端界面支持本地视频选择 - 添加视频封面提取和保存功能 - 优化后端路由支持本地视频上传
138 lines
4.9 KiB
Python
138 lines
4.9 KiB
Python
import os
|
||
import subprocess
|
||
from abc import ABC
|
||
from typing import Optional
|
||
|
||
from app.downloaders.base import Downloader
|
||
from app.enmus.note_enums import DownloadQuality
|
||
from app.models.audio_model import AudioDownloadResult
|
||
import os
|
||
import subprocess
|
||
|
||
from app.utils.video_helper import save_cover_to_static
|
||
|
||
|
||
class LocalDownloader(Downloader, ABC):
|
||
def __init__(self):
|
||
|
||
super().__init__()
|
||
|
||
|
||
def extract_cover(self, input_path: str, output_dir: Optional[str] = None) -> str:
|
||
"""
|
||
从本地视频文件中提取一张封面图（默认取第一帧）
|
||
:param input_path: 输入视频路径
|
||
:param output_dir: 输出目录，默认和视频同目录
|
||
:return: 提取出的封面图片路径
|
||
"""
|
||
if not os.path.exists(input_path):
|
||
raise FileNotFoundError(f"输入文件不存在: {input_path}")
|
||
|
||
if output_dir is None:
|
||
output_dir = os.path.dirname(input_path)
|
||
|
||
base_name = os.path.splitext(os.path.basename(input_path))[0]
|
||
output_path = os.path.join(output_dir, f"{base_name}_cover.jpg")
|
||
|
||
try:
|
||
command = [
|
||
'ffmpeg',
|
||
'-i', input_path,
|
||
'-ss', '00:00:01', # 跳到视频第1秒，防止黑屏
|
||
'-vframes', '1', # 只截取一帧
|
||
'-q:v', '2', # 输出质量高一点（qscale，2是很高）
|
||
'-y', # 覆盖
|
||
output_path
|
||
]
|
||
subprocess.run(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, check=True)
|
||
|
||
if not os.path.exists(output_path):
|
||
raise RuntimeError(f"封面图片生成失败: {output_path}")
|
||
|
||
return output_path
|
||
except subprocess.CalledProcessError as e:
|
||
raise RuntimeError(f"提取封面失败: {output_path}") from e
|
||
|
||
def convert_to_mp3(self,input_path: str, output_path: str = None) -> str:
|
||
"""
|
||
将本地视频文件转为 MP3 音频文件
|
||
:param input_path: 输入文件路径（如 .mp4）
|
||
:param output_path: 输出文件路径（可选，默认同目录同名 .mp3）
|
||
:return: 生成的 mp3 文件路径
|
||
"""
|
||
if not os.path.exists(input_path):
|
||
raise FileNotFoundError(f"输入文件不存在: {input_path}")
|
||
|
||
if output_path is None:
|
||
base, _ = os.path.splitext(input_path)
|
||
output_path = base + ".mp3"
|
||
try:
|
||
# 调用 ffmpeg 转换
|
||
command = [
|
||
'ffmpeg',
|
||
'-i', input_path,
|
||
'-vn', # 不要视频流
|
||
'-acodec', 'libmp3lame', # 使用mp3编码
|
||
'-y', # 覆盖输出文件
|
||
output_path
|
||
]
|
||
|
||
subprocess.run(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, check=True)
|
||
|
||
if not os.path.exists(output_path):
|
||
raise RuntimeError(f"mp3 文件生成失败: {output_path}")
|
||
|
||
return output_path
|
||
except subprocess.CalledProcessError as e:
|
||
raise RuntimeError(f"mp3 文件生成失败: {output_path}") from e
|
||
def download_video(self, video_url: str, output_dir: str = None) -> str:
|
||
"""
|
||
处理本地文件路径，返回视频文件路径
|
||
"""
|
||
if video_url.startswith('/uploads'):
|
||
project_root = os.getcwd()
|
||
video_url = os.path.join(project_root, video_url.lstrip('/'))
|
||
video_url = os.path.normpath(video_url)
|
||
|
||
if not os.path.exists(video_url):
|
||
raise FileNotFoundError()
|
||
return video_url
|
||
def download(
|
||
self,
|
||
video_url: str,
|
||
output_dir: str = None,
|
||
quality: DownloadQuality = "fast",
|
||
need_video: Optional[bool] = False
|
||
) -> AudioDownloadResult:
|
||
"""
|
||
处理本地文件路径，返回音频元信息
|
||
"""
|
||
if video_url.startswith('/uploads'):
|
||
project_root = os.getcwd()
|
||
video_url = os.path.join(project_root, video_url.lstrip('/'))
|
||
video_url = os.path.normpath(video_url)
|
||
|
||
if not os.path.exists(video_url):
|
||
raise FileNotFoundError(f"本地文件不存在: {video_url}")
|
||
|
||
file_name = os.path.basename(video_url)
|
||
title, _ = os.path.splitext(file_name)
|
||
print(title, file_name,video_url)
|
||
file_path=self.convert_to_mp3(video_url)
|
||
cover_path = self.extract_cover(video_url)
|
||
cover_url = save_cover_to_static(cover_path)
|
||
|
||
print('file——path',file_path)
|
||
return AudioDownloadResult(
|
||
file_path=file_path,
|
||
title=title,
|
||
duration=0, # 可选：后续加上读取时长
|
||
cover_url=cover_url, # 暂无封面
|
||
platform="local",
|
||
video_id=title,
|
||
raw_info={
|
||
'path': file_path
|
||
},
|
||
video_path=None
|
||
)
|