GithubBackup/BiliNote
1
0
Fork 0
mirror of https://github.com/JefferyHcool/BiliNote.git synced 2026-06-30 20:21:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #89 from JefferyHcool/feature/kuaishou

Browse Source 
refactor(utils): 更新模型目录获取逻辑以支持打包运行
This commit is contained in:
Jianwu Huang
2025-05-09 10:28:27 +08:00
committed by GitHub
parent 892ccd9ee4 0a5196a475
commit 274d7b9677
1 changed files with 11 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
backend/app/utils/path_helper.py
View File

@@ -1,4 +1,6 @@
import os import os
import sys
from pathlib import Path
PROJECT_ROOT = os.path.abspath(os.path.join(os.path.dirname(__file__), "../../")) PROJECT_ROOT = os.path.abspath(os.path.join(os.path.dirname(__file__), "../../"))
@@ -10,11 +12,15 @@ def get_data_dir():
def get_model_dir(subdir: str = "whisper") -> str: def get_model_dir(subdir: str = "whisper") -> str:
base = os.path.abspath(os.path.join(os.path.dirname(__file__), "../../models")) # 判断是否为打包状态（PyInstaller）
path = os.path.join(base, subdir) if getattr(sys, 'frozen', False):
# exe 执行，放在 APPDATA 或 ~/.cache 下
base_dir = os.path.join(os.getenv("APPDATA") or str(Path.home()), "BiliNote", "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) os.makedirs(path, exist_ok=True)
return path return path
if __name__ == '__main__':
print(get_data_dir())