🎨 自动删除过期备份, 优化依赖安装

Author: KimigaiiWuyi

gsuid_core/buildin_plugins/core_command/core_backup/__init__.py

@@ -5,9 +5,12 @@
 from gsuid_core.logger import logger
 from gsuid_core.data_store import get_res_path
 from gsuid_core.utils.database.base_models import DB_PATH
-from gsuid_core.utils.plugins_config.gs_config import backup_config
-from gsuid_core.utils.backup.backup_core import copy_and_rebase_paths
 from gsuid_core.utils.backup.backup_files import clean_log, backup_file
+from gsuid_core.utils.plugins_config.gs_config import log_config, backup_config
+from gsuid_core.utils.backup.backup_core import (
+    remove_old_backups,
+    copy_and_rebase_paths,
+)
 from gsuid_core.utils.database.global_val_models import (
     CoreDataSummary,
     CoreDataAnalysis,
@@ -24,6 +27,7 @@
 
 DB_BACKUP = get_res_path(['GsCore', 'database_backup'])
 
+CLEAN_DAY: str = log_config.get_config('ScheduledCleanLogDay').data
 backup_time: str = backup_config.get_config('backup_time').data
 backup_time = backup_time.lstrip('0')
 backup_hour, backup_minute = backup_time.split(':')
@@ -40,9 +44,10 @@ async def backup_path_files():
     '''
     凌晨自动备份`备份管理`中的路径树
     '''
-
     copy_and_rebase_paths()
     logger.success('♻️ [早柚核心] 路径已备份!')
+    remove_old_backups(int(CLEAN_DAY))
+    logger.success(f'♻️ [早柚核心] 已删除超过 {CLEAN_DAY} 天的备份文件!')
 
 
 @sv_core_backup.on_fullmatch('强制执行文件备份')

gsuid_core/server.py

@@ -18,6 +18,7 @@
     from packaging.requirements import Requirement
 except ImportError:
     print("正在安装必要依赖 'packaging'...")
+    subprocess.check_call([sys.executable, "-m", "ensurepip"])
     subprocess.check_call(
         [sys.executable, "-m", "pip", "install", "packaging"]
     )
@@ -433,27 +434,53 @@ def install_packages(packages: List[str], upgrade: bool = False):
 
     logger.info(f'🚀 [安装/更新依赖] 开始安装以下包: {packages}')
 
-    # 使用当前 Python 解释器路径，避免环境混乱
-    cmd = [sys.executable, "-m", "pip", "install"]
+    # 定义镜像源列表 (名称, URL)
+    # 顺序: 字节 -> 阿里 -> 清华 -> 官方
+    mirrors = [
+        ("字节源 (Volces)", "https://mirrors.volces.com/pypi/simple/"),
+        ("阿里源 (Aliyun)", "https://mirrors.aliyun.com/pypi/simple/"),
+        ("清华源 (Tsinghua)", "https://pypi.tuna.tsinghua.edu.cn/simple"),
+        ("官方源 (PyPI)", "https://pypi.org/simple"),
+    ]
+
+    # 构建基础命令
+    base_cmd = [sys.executable, "-m", "pip", "install"]
     if upgrade:
-        cmd.append("-U")
+        base_cmd.append("-U")
 
-    # 将包名作为参数追加
-    cmd.extend(packages)
+    # 追加包名
+    base_cmd.extend(packages)
 
-    # 使用国内源可选项 (建议在配置中做，这里保留原逻辑的简化版)
-    cmd.extend(["-i", "https://pypi.org/simple"])
+    install_success = False
 
-    retcode = execute_cmd(cmd)
+    # 轮询尝试
+    for mirror_name, mirror_url in mirrors:
+        logger.info(f'⏳ [安装/更新依赖] 正在尝试使用 [{mirror_name}] ...')
 
-    if retcode != 0:
-        logger.warning('[安装/更新依赖] 安装失败，尝试使用清华源重试...')
-        cmd_retry = cmd[:-2] + [
-            "-i",
-            "https://pypi.tuna.tsinghua.edu.cn/simple",
-        ]
-        execute_cmd(cmd_retry)
+        # 组装完整命令，加入 -i 参数
+        cmd = base_cmd + ["-i", mirror_url]
+
+        # 有些环境可能需要信任 host，防止 SSL 报错，可选添加:
+        # host = mirror_url.split("//")[-1].split("/")[0]
+        # cmd.extend(["--trusted-host", host])
+
+        retcode = execute_cmd(cmd)
+
+        if retcode == 0:
+            logger.info(f'✅ [安装/更新依赖] 使用 [{mirror_name}] 安装成功!')
+            install_success = True
+            break  # 安装成功，跳出循环
+        else:
+            logger.warning(
+                f'⚠️ [安装/更新依赖] 使用 [{mirror_name}] 安装失败，准备尝试下一个源...'
+            )
+
+    if not install_success:
+        logger.error(
+            '❌ [安装/更新依赖] 所有源均尝试失败，请检查网络或包名是否正确。'
+        )
 
+    # 刷新依赖状态
     refresh_installed_dependencies()
 
 

gsuid_core/utils/backup/backup_core.py

@@ -94,3 +94,70 @@ def copy_and_rebase_paths(
         return -10
 
     return 0
+
+
+def remove_old_backups(days: int = 30) -> int:
+    """
+    删除超过指定天数的备份文件或目录。
+
+    :param days: 保留的天数，默认为30天。
+    :return: 被删除的文件/目录数量。
+    """
+    if not backup_path.exists():
+        logger.warning(f"备份目录不存在: {backup_path}，跳过清理。")
+        return 0
+
+    current_time = datetime.now()
+    deleted_count = 0
+
+    logger.info(f"开始清理超过 {days} 天的备份文件...")
+
+    # 遍历备份目录下的所有项目
+    for item in backup_path.iterdir():
+        # 获取文件名（不含扩展名），例如 'mydata-2023-11-19'
+        # item.stem 会自动去掉 .zip 后缀
+        name_stem = item.stem
+
+        # 尝试提取日期部分
+        # 你的命名格式是: f'{file_id}-{date_str}'，date_str 是 "YYYY-MM-DD" (10个字符)
+        # 所以我们取 stem 的最后 10 位
+        if len(name_stem) < 10:
+            continue
+
+        date_str_part = name_stem[-10:]
+
+        try:
+            # 尝试将后缀解析为日期
+            backup_date = datetime.strptime(date_str_part, "%Y-%m-%d")
+        except ValueError:
+            # 如果解析失败（说明不是符合该日期格式的文件），则跳过
+            continue
+
+        # 计算时间差
+        time_delta = current_time - backup_date
+
+        if time_delta.days > days:
+            try:
+                if item.is_file():
+                    item.unlink()  # 删除文件 (通常是 .zip)
+                    logger.info(
+                        "🗑️ [早柚核心] 已删除过期备份文件:"
+                        f" {item.name} ({time_delta.days}天前)"
+                    )
+                elif item.is_dir():
+                    shutil.rmtree(item)  # 删除目录 (如果存在未压缩的残留目录)
+                    logger.info(
+                        "🗑️ [早柚核心] 已删除过期备份目录:"
+                        f" {item.name} ({time_delta.days}天前)"
+                    )
+
+                deleted_count += 1
+            except Exception as e:
+                logger.warning(f"❌ 删除 {item.name} 失败: {e}")
+
+    if deleted_count > 0:
+        logger.success(f"✅ 清理完成，共删除 {deleted_count} 个过期备份。")
+    else:
+        logger.info("✨ 没有发现需要删除的过期备份。")
+
+    return deleted_count