codfrm/MoviePilot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: 订阅搜索支持默认包含与排除规则

Browse Source
This commit is contained in:
WithdewHua
2023-09-17 18:04:52 +08:00
parent 951be74a21
commit 42311f0118
2 changed files with 22 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
app/chain/subscribe.py
View File

@ -285,15 +285,21 @@ class SubscribeChain(ChainBase):
torrent_meta = context.meta_info torrent_meta = context.meta_info
torrent_info = context.torrent_info torrent_info = context.torrent_info
torrent_mediainfo = context.media_info torrent_mediainfo = context.media_info
# 包含与排除规则
default_include_exclude = self.systemconfig.get(SystemConfigKey.DefaultIncludeExcludeFilter) or {}
include = subscribe.include or default_include_exclude.get("include")
exclude = subscribe.exclude or default_include_exclude.get("exclude")
# 包含 # 包含
if subscribe.include: if include:
if not re.search(r"%s" % subscribe.include, if not re.search(r"%s" % include,
f"{torrent_info.title} {torrent_info.description}", re.I): f"{torrent_info.title} {torrent_info.description}", re.I):
logger.info(f"{torrent_info.title} 不匹配包含规则 {include}")
continue continue
# 排除 # 排除
if subscribe.exclude: if exclude:
if re.search(r"%s" % subscribe.exclude, if re.search(r"%s" % exclude,
f"{torrent_info.title} {torrent_info.description}", re.I): f"{torrent_info.title} {torrent_info.description}", re.I):
logger.info(f"{torrent_info.title} 匹配排除规则 {exclude}")
continue continue
# 非洗版 # 非洗版
if not subscribe.best_version: if not subscribe.best_version:
@ -557,15 +563,21 @@ class SubscribeChain(ChainBase):
if torrent_meta.episode_list: if torrent_meta.episode_list:
logger.info(f'{subscribe.name} 正在洗版，{torrent_info.title} 不是整季') logger.info(f'{subscribe.name} 正在洗版，{torrent_info.title} 不是整季')
continue continue
# 包含与排除规则
default_include_exclude = self.systemconfig.get(SystemConfigKey.DefaultIncludeExcludeFilter) or {}
include = subscribe.include or default_include_exclude.get("include")
exclude = subscribe.exclude or default_include_exclude.get("exclude")
# 包含 # 包含
if subscribe.include: if include:
if not re.search(r"%s" % subscribe.include, if not re.search(r"%s" % include,
f"{torrent_info.title} {torrent_info.description}", re.I): f"{torrent_info.title} {torrent_info.description}", re.I):
logger.info(f"{torrent_info.title} 不匹配包含规则 {include}")
continue continue
# 排除 # 排除
if subscribe.exclude: if exclude:
if re.search(r"%s" % subscribe.exclude, if re.search(r"%s" % exclude,
f"{torrent_info.title} {torrent_info.description}", re.I): f"{torrent_info.title} {torrent_info.description}", re.I):
logger.info(f"{torrent_info.title} 匹配排除规则 {exclude}")
continue continue
# 匹配成功 # 匹配成功
logger.info(f'{mediainfo.title_year} 匹配成功：{torrent_info.title}') logger.info(f'{mediainfo.title_year} 匹配成功：{torrent_info.title}')

2
app/schemas/types.py
View File

@ -64,6 +64,8 @@ class SystemConfigKey(Enum):
FilterRules = "FilterRules" FilterRules = "FilterRules"
# 洗版规则 # 洗版规则
FilterRules2 = "FilterRules2" FilterRules2 = "FilterRules2"
# 默认包含与排除规则
DefaultIncludeExcludeFilter = "DefaultIncludeExcludeFilter"
# 转移屏蔽词 # 转移屏蔽词
TransferExcludeWords = "TransferExcludeWords" TransferExcludeWords = "TransferExcludeWords"