From 4f27897e0853a9047c1f77ed69d827dd1ad4cd17 Mon Sep 17 00:00:00 2001 From: InfinityPacer <160988576+InfinityPacer@users.noreply.github.com> Date: Sun, 6 Oct 2024 01:58:19 +0800 Subject: [PATCH] refactor(config): replace hard-coded strings with SystemConfigKey --- app/chain/subscribe.py | 4 ++-- app/schemas/types.py | 4 ++++ 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/app/chain/subscribe.py b/app/chain/subscribe.py index 94bf52f5..57c3e73b 100644 --- a/app/chain/subscribe.py +++ b/app/chain/subscribe.py @@ -1118,9 +1118,9 @@ class SubscribeChain(ChainBase): """ default_subscribe_key = None if mtype == MediaType.TV: - default_subscribe_key = "DefaultTvSubscribeConfig" + default_subscribe_key = SystemConfigKey.DefaultTvSubscribeConfig.value if mtype == MediaType.MOVIE: - default_subscribe_key = "DefaultMovieSubscribeConfig" + default_subscribe_key = SystemConfigKey.DefaultMovieSubscribeConfig.value # 默认订阅规则 if hasattr(settings, default_subscribe_key): diff --git a/app/schemas/types.py b/app/schemas/types.py index ebea3ad5..0a1a644e 100644 --- a/app/schemas/types.py +++ b/app/schemas/types.py @@ -113,6 +113,10 @@ class SystemConfigKey(Enum): UserInstalledPlugins = "UserInstalledPlugins" # 插件安装统计 PluginInstallReport = "PluginInstallReport" + # 默认电影订阅规则 + DefaultMovieSubscribeConfig = "DefaultMovieSubscribeConfig" + # 默认电视剧订阅规则 + DefaultTvSubscribeConfig = "DefaultTvSubscribeConfig" # 处理进度Key字典