add update for docker

Author: shinonomeow

.github/workflows/build.yml

@@ -295,6 +295,10 @@ jobs:
           echo ${{ needs.version-info.outputs.version }}
           echo "VERSION='${{ needs.version-info.outputs.version }}'" >> module/__version__.py
 
+      - name: Copy uv files
+        run: |
+          cp backend/uv.lock backend/src/
+          cp backend/pyproject.toml backend/src/
 
       - name: Zip app
         run: |

backend/src/module/api/program.py

@@ -1,3 +1,4 @@
+import asyncio
 import logging
 from contextlib import asynccontextmanager
 
@@ -8,7 +9,8 @@
 from module.core import Program
 from module.models import APIResponse
 from module.models.response import ResponseModel
-from module.security.api import UNAUTHORIZED, get_current_user
+from module.security.api import get_current_user
+from module.update import ReleaseChecker
 
 from .response import u_response
 
@@ -128,4 +130,129 @@ async def shutdown_program():
     dependencies=[Depends(get_current_user)],
 )
 async def check_downloader_status():
-    return await program.program_status.check_downloader()
+    return program.program_status.check_downloader()
+
+
+# Check for updates
+@router.get(
+    "/check/update",
+    tags=["update"],
+    response_model=dict,
+    dependencies=[Depends(get_current_user)],
+)
+async def check_for_update(include_prerelease: bool = False):
+    """检查是否有可用的版本更新
+
+    Args:
+        include_prerelease: 是否包含预发布版本
+
+    Returns:
+        dict: 包含版本检查结果的字典
+    """
+    try:
+        checker = ReleaseChecker("shinonomeow", "Auto_Bangumi")
+        result = await checker.check_for_update(include_prerelease=True)
+        return result
+    except Exception as e:
+        logger.error(f"Failed to check for updates: {e}")
+        raise HTTPException(
+            status_code=500,
+            detail={"msg_en": "Failed to check for updates.", "msg_zh": "检查更新失败。", "error": str(e)},
+        )
+
+
+@router.post("/update", response_model=APIResponse, dependencies=[Depends(get_current_user)])
+async def update_program(download_url: str):
+    """Docker 环境更新接口
+
+    Args:
+        download_url: 更新包下载 URL（必须是 GitHub releases）
+
+    Returns:
+        APIResponse: 更新结果
+    """
+    try:
+        # 检测是否在 Docker 环境
+        from module.utils.environment import is_docker_environment
+
+        if not is_docker_environment():
+            raise HTTPException(
+                status_code=400,
+                detail={
+                    "msg_en": "Update only supported in Docker environment.",
+                    "msg_zh": "更新功能仅支持 Docker 环境。",
+                },
+            )
+
+        # 导入更新器
+        from module.update import docker_updater
+
+        # 创建异步任务执行更新
+        async def perform_update():
+            try:
+                # 等待一小段时间让 API 响应返回
+                await asyncio.sleep(1)
+
+                logger.info(f"[Program] Starting update from URL: {download_url}")
+
+                # 停止应用核心
+                await program.stop()
+
+                # 执行更新
+                await docker_updater.update(download_url)
+
+                # 强制重启容器
+                docker_updater.force_restart()
+
+            except Exception as e:
+                logger.error(f"[Program] Update failed: {e}")
+                # 尝试重启应用核心
+                try:
+                    await program.start()
+                except Exception as restart_error:
+                    logger.error(f"[Program] Failed to restart after update failure: {restart_error}")
+
+        # 启动异步更新任务
+        asyncio.create_task(perform_update())
+
+        # 立即返回响应
+        return u_response(
+            ResponseModel(
+                status=True,
+                status_code=200,
+                msg_en="Update started successfully. Container will restart shortly.",
+                msg_zh="更新已开始。容器将很快重启。",
+                data={"download_url": download_url},
+            )
+        )
+
+    except HTTPException:
+        raise
+    except Exception as e:
+        logger.error(f"[Program] Update initiation failed: {e}")
+        raise HTTPException(
+            status_code=500,
+            detail={"msg_en": "Failed to start update process.", "msg_zh": "启动更新进程失败。", "error": str(e)},
+        )
+
+
+# Update status check
+@router.get("/update/status", response_model=dict, dependencies=[Depends(get_current_user)])
+async def get_update_status():
+    """获取更新状态
+
+    Returns:
+        dict: 更新状态信息
+    """
+    import os
+
+    from module.utils.environment import get_environment_info, is_docker_environment
+
+    # 检查是否有更新锁文件
+    update_lock_exists = os.path.exists("/tmp/auto_bangumi_update.lock")
+
+    return {
+        "update_in_progress": update_lock_exists,
+        "environment": get_environment_info(),
+        "update_supported": is_docker_environment(),
+    }

backend/src/module/database/bangumi.py

@@ -53,9 +53,9 @@ def add(self, data: Bangumi):
         return True
 
     def update(self, data: Bangumi) -> bool:
-        self.session.merge(data)
+        merge_data = self.session.merge(data)
         self.session.commit()
-        self.session.refresh(data)
+        self.session.refresh(merge_data)
         # logger.debug(f"[Database] Update {data.official_title}")
         return True
 

backend/src/module/update/__init__.py

@@ -1,8 +1,14 @@
+from .docker_updater import docker_updater, DockerUpdater
+from .release_checker import ReleaseChecker, ReleaseInfo
 from .startup import first_run, start_up
 from .version_check import version_check
 
 __all__ = [
     "first_run",
     "start_up",
     "version_check",
+    "ReleaseChecker",
+    "ReleaseInfo",
+    "docker_updater",
+    "DockerUpdater",
 ]