File size: 3,250 Bytes
6cfe55f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 | import os
import subprocess
import sys
from dotenv import load_dotenv
from app.utils.logger import get_logger
logger = get_logger(__name__)
def _load_dotenv_from_multiple_paths():
"""尝试多个位置加载 .env，适配源码运行和 PyInstaller 打包场景。
PyInstaller 打包后当前工作目录是 EXE 所在目录，而源码运行时 .env
通常在项目根目录或 backend/ 同级。遍历常见候选路径确保能命中。
"""
candidates = []
# 1. 当前工作目录（EXE 所在目录）
candidates.append(os.path.join(os.getcwd(), '.env'))
# 2. 本脚本所在目录（backend/）
script_dir = os.path.dirname(os.path.abspath(__file__))
candidates.append(os.path.join(script_dir, '.env'))
# 3. 项目根目录（backend/../.env）
candidates.append(os.path.join(script_dir, '..', '.env'))
# 4. PyInstaller 打包后的 _internal/ 子目录
if getattr(sys, 'frozen', False):
exe_dir = os.path.dirname(sys.executable)
candidates.append(os.path.join(exe_dir, '_internal', '.env'))
for path in candidates:
normalized = os.path.normpath(path)
if os.path.isfile(normalized):
load_dotenv(normalized)
return
# 都没找到，fallback 到默认行为（从 CWD 找）
load_dotenv()
_load_dotenv_from_multiple_paths()
def check_ffmpeg_exists() -> bool:
"""
检查 ffmpeg 是否可用。优先使用 FFMPEG_BIN_PATH 环境变量指定的路径。
"""
ffmpeg_bin_path = os.getenv("FFMPEG_BIN_PATH")
logger.info(f"FFMPEG_BIN_PATH: {ffmpeg_bin_path}")
if ffmpeg_bin_path and os.path.isdir(ffmpeg_bin_path):
os.environ["PATH"] = ffmpeg_bin_path + os.pathsep + os.environ.get("PATH", "")
logger.info(f"使用FFMPEG_BIN_PATH: {ffmpeg_bin_path}")
else:
# 遍历系统PATH寻找ffmpeg.exe
system_path = os.environ.get("PATH", "")
path_dirs = system_path.split(os.pathsep)
for path_dir in path_dirs:
ffmpeg_exe_path = os.path.join(path_dir, "ffmpeg.exe")
if os.path.isfile(ffmpeg_exe_path):
os.environ["PATH"] = path_dir + os.pathsep + system_path
logger.info(f"在系统PATH中找到ffmpeg: {path_dir}")
break
try:
subprocess.run(["ffmpeg", "-version"], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL, check=True)
logger.info("ffmpeg 已安装")
return True
except (FileNotFoundError, OSError, subprocess.CalledProcessError):
logger.info("ffmpeg 未安装")
return False
def ensure_ffmpeg_or_raise():
"""
校验 ffmpeg 是否可用，否则抛出异常并提示安装方式。
"""
if not check_ffmpeg_exists():
logger.error("未检测到 ffmpeg，请先安装后再使用本功能。")
raise EnvironmentError(
" 未检测到 ffmpeg，请先安装后再使用本功能。\n"
"👉 下载地址：https://ffmpeg.org/download.html\n"
"🪟 Windows 推荐：https://www.gyan.dev/ffmpeg/builds/\n"
"💡 如果你已安装，请将其路径写入 `.env` 文件，例如：\n"
"FFMPEG_BIN_PATH=/your/custom/ffmpeg/bin"
)
|