mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-05-23 23:16:46 +00:00
62 lines
2.4 KiB
Python
62 lines
2.4 KiB
Python
import regex as re
|
|
|
|
from app.db.systemconfig_oper import SystemConfigOper
|
|
from app.schemas.types import SystemConfigKey
|
|
from app.utils.singleton import Singleton
|
|
|
|
|
|
class CustomizationMatcher(metaclass=Singleton):
|
|
"""
|
|
识别自定义占位符
|
|
"""
|
|
|
|
def __init__(self):
|
|
self.systemconfig = SystemConfigOper()
|
|
self.customization = None
|
|
self.custom_separator = None
|
|
self._customization_re_cache = {}
|
|
|
|
@staticmethod
|
|
def _normalize_customization(customization):
|
|
"""
|
|
规范化自定义占位符配置，兼容历史字符串与列表两种保存格式。
|
|
"""
|
|
if isinstance(customization, str):
|
|
customization = customization.replace("\n", ";").replace("|", ";").strip(";").split(";")
|
|
if not customization:
|
|
return []
|
|
return list(filter(None, customization))
|
|
|
|
def match(self, title=None):
|
|
"""
|
|
:param title: 资源标题或文件名
|
|
:return: 匹配结果
|
|
"""
|
|
if not title:
|
|
return ""
|
|
# 自定义占位符需要跟随系统配置实时生效，避免单例缓存导致保存后仍沿用旧规则。
|
|
customization = self._normalize_customization(
|
|
self.systemconfig.get(SystemConfigKey.Customization)
|
|
)
|
|
if not customization:
|
|
self.customization = None
|
|
return ""
|
|
self.customization = "|".join([f"({item})" for item in customization])
|
|
|
|
customization_re = self._customization_re_cache.get(self.customization)
|
|
if not customization_re:
|
|
# 配置每次读取、编译结果按规则缓存，兼顾实时生效和高频识别性能。
|
|
customization_re = re.compile(r"%s" % self.customization)
|
|
self._customization_re_cache[self.customization] = customization_re
|
|
# 处理重复多次的情况，保留先后顺序（按添加自定义占位符的顺序）
|
|
unique_customization = {}
|
|
for item in customization_re.findall(title):
|
|
if not isinstance(item, tuple):
|
|
item = (item,)
|
|
for i in range(len(item)):
|
|
if item[i] and unique_customization.get(item[i]) is None:
|
|
unique_customization[item[i]] = i
|
|
unique_customization = list(dict(sorted(unique_customization.items(), key=lambda x: x[1])).keys())
|
|
separator = self.custom_separator or "@"
|
|
return separator.join(unique_customization)
|