From 2883ccbe87f2784dd33f7843d2fa51fb38321383 Mon Sep 17 00:00:00 2001
From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com>
Date: Mon, 26 Jan 2026 04:32:11 +0000
Subject: [PATCH] Move category methods to ChainBase and use consistent naming

Co-authored-by: jxxghp <51039935+jxxghp@users.noreply.github.com>
---
 app/api/endpoints/category.py |  2 +-
 app/chain/__init__.py         | 13 +++++++++++++
 app/chain/tmdb.py             | 13 -------------
 3 files changed, 14 insertions(+), 14 deletions(-)

diff --git a/app/api/endpoints/category.py b/app/api/endpoints/category.py
index 771522c1..5f6d6977 100644
--- a/app/api/endpoints/category.py
+++ b/app/api/endpoints/category.py
@@ -13,7 +13,7 @@ def get_category_config(_: User = Depends(get_current_active_user)):
     """
     获取分类策略配置
     """
-    config = TmdbChain().load_category_config()
+    config = TmdbChain().category_config()
     return schemas.Response(success=True, data=config.model_dump())
 
 
diff --git a/app/chain/__init__.py b/app/chain/__init__.py
index 098fc951..f8a2aebd 100644
--- a/app/chain/__init__.py
+++ b/app/chain/__init__.py
@@ -26,6 +26,7 @@ from app.helper.service import ServiceConfigHelper
 from app.log import logger
 from app.schemas import TransferInfo, TransferTorrent, ExistMediaInfo, DownloadingTorrent, CommingMessage, Notification, \
     WebhookEventInfo, TmdbEpisode, MediaPerson, FileItem, TransferDirectoryConf
+from app.schemas.category import CategoryConfig
 from app.schemas.types import TorrentStatus, MediaType, MediaImageType, EventType, MessageChannel
 from app.utils.object import ObjectUtils
 
@@ -1062,6 +1063,18 @@ class ChainBase(metaclass=ABCMeta):
         """
         return self.run_module("media_category")
 
+    def category_config(self) -> CategoryConfig:
+        """
+        获取分类策略配置
+        """
+        return self.run_module("load_category_config")
+
+    def save_category_config(self, config: CategoryConfig) -> bool:
+        """
+        保存分类策略配置
+        """
+        return self.run_module("save_category_config", config=config)
+
     def register_commands(self, commands: Dict[str, dict]) -> None:
         """
         注册菜单命令
diff --git a/app/chain/tmdb.py b/app/chain/tmdb.py
index c3c201f2..0ed91550 100644
--- a/app/chain/tmdb.py
+++ b/app/chain/tmdb.py
@@ -5,7 +5,6 @@ from app import schemas
 from app.chain import ChainBase
 from app.core.context import MediaInfo
 from app.schemas import MediaType
-from app.schemas.category import CategoryConfig
 
 
 class TmdbChain(ChainBase):
@@ -321,15 +320,3 @@ class TmdbChain(ChainBase):
         if infos:
             return [info.backdrop_path for info in infos if info and info.backdrop_path][:num]
         return []
-
-    def load_category_config(self) -> CategoryConfig:
-        """
-        加载分类策略配置
-        """
-        return self.run_module("load_category_config")
-
-    def save_category_config(self, config: CategoryConfig) -> bool:
-        """
-        保存分类策略配置
-        """
-        return self.run_module("save_category_config", config=config)