File size: 3,211 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 81 82 83 84 | import os
import sys
from pathlib import Path
PROJECT_ROOT = os.path.abspath(os.path.join(os.path.dirname(__file__), "../../"))
def get_data_dir():
if getattr(sys, 'frozen', False):
base_dir = os.path.dirname(sys.executable)
else:
base_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), "../../data"))
data_path = os.path.join(base_dir, "data")
os.makedirs(data_path, exist_ok=True)
return data_path
def get_model_dir(subdir: str = "whisper") -> str:
# 判断是否为打包状态（PyInstaller）
if getattr(sys, 'frozen', False):
# exe 执行，放在 APPDATA 或 ~/.cache 下
base_dir = os.path.join(os.getenv("APPDATA") or str(Path.home()), "VideoMemo", "models")
else:
# 开发时，相对项目根目录
base_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), "../../models"))
path = os.path.join(base_dir, subdir)
os.makedirs(path, exist_ok=True)
return path
def get_app_dir(subdir: str = "") -> str:
"""
返回一个稳定的可写目录：
- 开发时：使用项目 data 目录
- 打包后：使用 exe 所在目录
"""
if getattr(sys, 'frozen', False):
# 打包后运行：使用 main.exe 所在目录
base_dir = os.path.dirname(sys.executable)
else:
# 开发模式：使用项目的 /data 目录
base_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), "../../data"))
full_path = os.path.join(base_dir, subdir)
os.makedirs(full_path, exist_ok=True)
return full_path
def get_plugin_packages_dir() -> str:
"""用户自装 Python 包的「插件目录」（桌面端可选依赖如 mlx_whisper 装这里）。
打包后的应用是冻结的 Python 3.11，读不到系统 site-packages。
main.py 启动时会把该目录加进 sys.path（PyInstaller 内置包优先，
插件目录只补缺失的包），用户用 Python 3.11 安装后重启应用即可生效：
python3.11 -m pip install --target "<本目录>" mlx_whisper
与 get_model_dir 同一基准目录（macOS: ~/VideoMemo，Windows: %APPDATA%/VideoMemo）。
"""
base_dir = os.path.join(os.getenv("APPDATA") or str(Path.home()), "VideoMemo")
path = os.path.join(base_dir, "python-packages")
os.makedirs(path, exist_ok=True)
return path
def get_runtime_dir(name: str) -> str:
"""对外提供静态资源（static/uploads）的根目录，保证「写入」与「服务」同源。
- 打包(PyInstaller frozen)：使用 exe 所在目录的绝对路径。Tauri 以 sidecar 方式拉起
后端，其工作目录(cwd)不是项目目录（macOS 上常是只读位置），绝不能用 cwd 相对路径，
否则截图/封面的写入目录与 /static 挂载目录不一致，桌面端图片全部 404。
- 开发 / Docker：保持原有的 cwd 相对目录（"./static" 等），不改变既有部署与卷挂载行为。
"""
if getattr(sys, 'frozen', False):
base_dir = os.path.dirname(sys.executable)
else:
base_dir = "."
full_path = os.path.join(base_dir, name)
os.makedirs(full_path, exist_ok=True)
return full_path |