fix #4602

app/helper/directory.py

@@ -1,3 +1,4 @@
+import re
 from pathlib import Path
 from typing import List, Optional
 
@@ -8,6 +9,8 @@ from app.log import logger
 from app.schemas.types import SystemConfigKey
 from app.utils.system import SystemUtils
 
+JINJA2_VAR_PATTERN = re.compile(r"\{\{.*?\}\}", re.DOTALL)
+
 
 class DirectoryHelper:
     """
@@ -120,26 +123,32 @@ class DirectoryHelper:
         :param rename_path: 重命名后的路径
         :return: 媒体文件根路径
         """
+        if not rename_format:
+            logger.error("重命名格式不能为空")
+            return None
         # 计算重命名中的文件夹层数
         rename_list = rename_format.split("/")
         rename_format_level = len(rename_list) - 1
-        if rename_format_level <= 0:
-            # 无效重命名格式
-            logger.error(f"重命名格式 {rename_format} 不正确")
-            return None
+        # 查找标题参数所在层
         for level, name in enumerate(rename_list):
+            matchs = JINJA2_VAR_PATTERN.findall(name)
+            if not matchs:
+                continue
             # 处理特例，有的人重命名的第一层是年份、分辨率
-            if "{{title}}" in name:
+            if any("title" in m for m in matchs):
                 # 找出含标题的这一层作为媒体根目录
                 rename_format_level -= level
                 break
         else:
             # 假定第一层目录是媒体根目录
-            logger.warn(f"重命名格式 {rename_format} 缺少 {{{{title}}}}")
+            logger.warn(f"重命名格式 {rename_format} 缺少标题参数")
         if rename_format_level > len(rename_path.parents):
             # 通常因为路径以/结尾，被Path规范化删除了
             logger.error(f"路径 {rename_path} 不匹配重命名格式 {rename_format}")
             return None
+        if rename_format_level <= 0:
+            # 所有媒体文件都存在一个目录内的特殊需求
+            rename_format_level = 1
         # 媒体根路径
         media_root = rename_path.parents[rename_format_level - 1]
         return media_root