From 048a3b70df3a67aecd278bc14aed97490ffbe1db Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=BB=84=E5=BB=BA=E6=AD=A6?= <huangjianwu@ynyuc.com>
Date: Fri, 9 May 2025 12:40:36 +0800
Subject: [PATCH] =?UTF-8?q?refactor(app/utils):=20=E6=9B=B4=E6=96=B0=20Vid?=
 =?UTF-8?q?eoReader=20=E7=B1=BB=E7=9A=84=E7=9B=AE=E5=BD=95=E8=AE=BE?=
 =?UTF-8?q?=E7=BD=AE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 引入 get_app_dir 函数用于获取应用目录路径
- 修改 frame_dir 和 grid_dir 参数默认值为 None
- 在构造函数中使用 get_app_dir 设置默认目录路径
---
 backend/app/utils/video_reader.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/backend/app/utils/video_reader.py b/backend/app/utils/video_reader.py
index 6922bd6..cd291d4 100644
--- a/backend/app/utils/video_reader.py
+++ b/backend/app/utils/video_reader.py
@@ -6,6 +6,7 @@ import ffmpeg
 from PIL import Image, ImageDraw, ImageFont
 
 from app.utils.logger import get_logger
+from app.utils.path_helper import get_app_dir
 
 logger = get_logger(__name__)
 class VideoReader:
@@ -17,16 +18,17 @@ class VideoReader:
                  unit_height=540,
                  save_quality=90,
                  font_path="fonts/arial.ttf",
-                 frame_dir="data/output_frames",
-                 grid_dir="data/grid_output"):
+                 frame_dir=None,
+                 grid_dir=None):
         self.video_path = video_path
         self.grid_size = grid_size
         self.frame_interval = frame_interval
         self.unit_width = unit_width
         self.unit_height = unit_height
         self.save_quality = save_quality
+        self.frame_dir = frame_dir or get_app_dir("output_frames")
+        self.grid_dir = grid_dir or get_app_dir("grid_output")
         self.font_path = font_path
-        self.frame_dir = frame_dir
         self.grid_dir = grid_dir
 
     def format_time(self, seconds: float) -> str: