chawin.chen commited on
Commit
6128c41
·
1 Parent(s): 4e455e5
Files changed (2) hide show
  1. cleanup_scheduler.py +5 -2
  2. config.py +3 -0
cleanup_scheduler.py CHANGED
@@ -9,7 +9,7 @@ from datetime import datetime
9
 
10
  from apscheduler.schedulers.background import BackgroundScheduler
11
 
12
- from config import logger, IMAGES_DIR, CLEANUP_INTERVAL_HOURS, CLEANUP_AGE_HOURS
13
 
14
  from utils import delete_file_from_bos
15
 
@@ -69,7 +69,10 @@ class ImageCleanupScheduler:
69
 
70
  # 删除文件
71
  os.remove(file_path)
72
- delete_file_from_bos(file_path)
 
 
 
73
  deleted_files.append(os.path.basename(file_path))
74
  total_size_deleted += file_size
75
 
 
9
 
10
  from apscheduler.schedulers.background import BackgroundScheduler
11
 
12
+ from config import logger, IMAGES_DIR, CLEANUP_INTERVAL_HOURS, CLEANUP_AGE_HOURS, BOS_CLEANUP_ENABLED
13
 
14
  from utils import delete_file_from_bos
15
 
 
69
 
70
  # 删除文件
71
  os.remove(file_path)
72
+ # 仅在 BOS_CLEANUP_ENABLED 为 True 时删除 BOS 上的文件
73
+ if BOS_CLEANUP_ENABLED:
74
+ delete_file_from_bos(file_path)
75
+
76
  deleted_files.append(os.path.basename(file_path))
77
  total_size_deleted += file_size
78
 
config.py CHANGED
@@ -346,6 +346,9 @@ AUTO_INIT_RVM = os.environ.get("AUTO_INIT_RVM", "false").lower() in ("1", "true"
346
  CLEANUP_INTERVAL_HOURS = float(os.environ.get("CLEANUP_INTERVAL_HOURS", 1.0)) # 清理任务执行间隔(小时),默认1小时
347
  CLEANUP_AGE_HOURS = float(os.environ.get("CLEANUP_AGE_HOURS", 1.0)) # 清理文件的年龄阈值(小时),默认1小时
348
 
 
 
 
349
  # BOS 自动同步清单:定义 BOS 路径和本地目录的映射,启动时可迭代该结构完成批量下载
350
  BOS_DOWNLOAD_TARGETS = [
351
  # {
 
346
  CLEANUP_INTERVAL_HOURS = float(os.environ.get("CLEANUP_INTERVAL_HOURS", 1.0)) # 清理任务执行间隔(小时),默认1小时
347
  CLEANUP_AGE_HOURS = float(os.environ.get("CLEANUP_AGE_HOURS", 1.0)) # 清理文件的年龄阈值(小时),默认1小时
348
 
349
+ # BOS 清理开关(默认关闭)
350
+ BOS_CLEANUP_ENABLED = os.environ.get("BOS_CLEANUP_ENABLED", "false").lower() in ("1", "true", "on")
351
+
352
  # BOS 自动同步清单:定义 BOS 路径和本地目录的映射,启动时可迭代该结构完成批量下载
353
  BOS_DOWNLOAD_TARGETS = [
354
  # {