From 55c515db4ed4a489604b0a4c5421daf008192dad Mon Sep 17 00:00:00 2001 From: x1ao4 Date: Sat, 26 Apr 2025 22:48:48 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E5=89=A7=E9=9B=86=E7=BC=96?= =?UTF-8?q?=E5=8F=B7=E6=8F=90=E5=8F=96=E5=87=BD=E6=95=B0=E5=8F=AF=E8=83=BD?= =?UTF-8?q?=E4=BC=9A=E6=8A=8A=E6=97=A5=E6=9C=9F=E6=8F=90=E5=8F=96=E4=B8=BA?= =?UTF-8?q?=E5=89=A7=E9=9B=86=E7=BC=96=E5=8F=B7=E7=9A=84=E9=97=AE=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- quark_auto_save.py | 111 +++++++++++++++++++++++++++++++++++++++++---- 1 file changed, 101 insertions(+), 10 deletions(-) diff --git a/quark_auto_save.py b/quark_auto_save.py index 7678c6d..2beb709 100644 --- a/quark_auto_save.py +++ b/quark_auto_save.py @@ -193,14 +193,76 @@ def extract_episode_number(filename, episode_patterns=None, config_data=None): Returns: int: 提取到的剧集号，如果无法提取则返回None """ + # 预处理：排除文件名中可能是日期的部分，避免误识别 + date_patterns = [ + # YYYY-MM-DD 或 YYYY.MM.DD 或 YYYY/MM/DD 或 YYYY MM DD格式（四位年份） + r'((?:19|20)\d{2})[-./\s](\d{1,2})[-./\s](\d{1,2})', + # YY-MM-DD 或 YY.MM.DD 或 YY/MM/DD 或 YY MM DD格式（两位年份） + r'((?:19|20)?\d{2})[-./\s](\d{1,2})[-./\s](\d{1,2})', + # 完整的YYYYMMDD格式（无分隔符） + r'((?:19|20)\d{2})(\d{2})(\d{2})', + # YYMMDD格式（两位年份，无分隔符） + r'(?= 3: + if re.match(r'(?:19|20)\d{2}', match.group(1)): # 首个分组是年份 + month = int(match.group(2)) + day = int(match.group(3)) + elif re.match(r'(?:19|20)\d{2}', match.group(3)): # 末尾分组是年份 + month = int(match.group(1)) + day = int(match.group(2)) + else: + # 处理两位数年份的情况（如25.03.21） + try: + # 假设第一个是年份，第二个是月，第三个是日 + year = int(match.group(1)) + month = int(match.group(2)) + day = int(match.group(3)) + + # 如果月和日在有效范围内，则这可能是一个日期 + if 1 <= month <= 12 and 1 <= day <= 31: + pass # 保持month和day的值 + else: + # 尝试另一种解释：月.日.年 + month = int(match.group(1)) + day = int(match.group(2)) + # 检查月和日的有效性 + if not (1 <= month <= 12 and 1 <= day <= 31): + # 仍然无效，重置month和day + month = None + day = None + except ValueError: + # 转换失败，保持month和day为None + pass + + # 如果能确定月日且是有效的日期，则从文件名中删除该日期 + if month and day and 1 <= month <= 12 and 1 <= day <= 31: + filename_without_dates = filename_without_dates.replace(date_str, " ") + # 优先匹配SxxExx格式 - match_s_e = re.search(r'[Ss](\d+)[Ee](\d+)', filename) + match_s_e = re.search(r'[Ss](\d+)[Ee](\d+)', filename_without_dates) if match_s_e: # 直接返回E后面的集数 return int(match_s_e.group(2)) # 其次匹配E01格式 - match_e = re.search(r'[Ee][Pp]?(\d+)', filename) + match_e = re.search(r'[Ee][Pp]?(\d+)', filename_without_dates) if match_e: return int(match_e.group(1)) @@ -233,19 +295,37 @@ def extract_episode_number(filename, episode_patterns=None, config_data=None): else: patterns = default_patterns - # 尝试使用每个正则表达式匹配文件名 + # 尝试使用每个正则表达式匹配文件名（使用不含日期的文件名） for pattern_regex in patterns: try: - match = re.search(pattern_regex, filename) + match = re.search(pattern_regex, filename_without_dates) if match: - return int(match.group(1)) + episode_num = int(match.group(1)) + + # 检查提取的数字是否可能是日期的一部分 + # 如果是纯数字并且可能是日期格式，则跳过 + if str(episode_num).isdigit() and is_date_format(str(episode_num)): + continue + + return episode_num except: continue - # 尝试其他通用提取方法 - 提取任何数字 - num_match = re.search(r'(\d+)', filename) + # 如果从不含日期的文件名中没有找到剧集号，尝试从原始文件名中提取 + # 这是为了兼容某些特殊情况，但要检查提取的数字不是日期 + file_name_without_ext = os.path.splitext(filename)[0] + + # 如果文件名是纯数字，且不是日期格式，则可能是剧集号 + if file_name_without_ext.isdigit() and not is_date_format(file_name_without_ext): + return int(file_name_without_ext) + + # 最后尝试提取任何数字，但要排除日期可能性 + num_match = re.search(r'(\d+)', filename_without_dates) if num_match: - return int(num_match.group(1)) + episode_num = int(num_match.group(1)) + # 检查提取的数字是否可能是日期 + if not is_date_format(str(episode_num)): + return episode_num return None @@ -260,7 +340,7 @@ NOTIFYS = [] def is_date_format(number_str): """ 判断一个纯数字字符串是否可能是日期格式 - 支持的格式：YYYYMMDD, MMDD + 支持的格式：YYYYMMDD, MMDD, YYMMDD """ # 判断YYYYMMDD格式 (8位数字) if len(number_str) == 8 and number_str.startswith('20'): @@ -273,6 +353,17 @@ def is_date_format(number_str): # 可能是日期格式 return True + # 判断YYMMDD格式 (6位数字) + elif len(number_str) == 6: + year_str = number_str[:2] + month = int(number_str[2:4]) + day = int(number_str[4:6]) + + # 检查月份和日期是否有效 + if 1 <= month <= 12 and 1 <= day <= 31: + # 可能是日期格式 + return True + # 判断MMDD格式 (4位数字) elif len(number_str) == 4: month = int(number_str[:2]) @@ -283,7 +374,7 @@ def is_date_format(number_str): # 可能是日期格式 return True - # 其他长度的纯数字不视为日期格式 + # 其他格式不视为日期格式 return False # 兼容青龙