mirror of
https://github.com/Cp0204/quark-auto-save.git
synced 2026-01-16 17:30:43 +08:00
将正则/剧集命名模式的查重逻辑改为基于文件名进行判断，为三种模式增加对忽略后缀功能的支持
1. 修改了正则命名模式的查重逻辑，只基于文件名进行判断，不再使用文件大小和修改时间。 2. 修改了剧集命名模式的查重逻辑，只基于文件名进行判断，不再使用文件大小和修改时间，也不再使用剧集号列表。 3. 为顺序命名模式、正则命名模式和剧集命名模式都添加了对忽略后缀功能的支持。
This commit is contained in:
x1ao4
parent
3903c69351
commit
38a3e0787d
@ -1369,10 +1369,18 @@ class Quark:
|
|||||||
# 生成新文件名
|
# 生成新文件名
|
||||||
save_name = sequence_pattern.replace("{}", f"{current_sequence:02d}") + file_ext
|
save_name = sequence_pattern.replace("{}", f"{current_sequence:02d}") + file_ext
|
||||||
|
|
||||||
# 检查目标目录是否已存在此文件
|
# 检查目标目录是否已存在此文件，支持忽略后缀选项
|
||||||
file_exists = any(
|
if task.get("ignore_extension", False):
|
||||||
dir_file["file_name"] == save_name for dir_file in dir_file_list
|
# 忽略后缀模式：只比较文件名部分，不比较扩展名
|
||||||
)
|
save_name_base = os.path.splitext(save_name)[0]
|
||||||
|
file_exists = any(
|
||||||
|
os.path.splitext(dir_file["file_name"])[0] == save_name_base for dir_file in dir_file_list
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
# 不忽略后缀模式：完整文件名必须匹配
|
||||||
|
file_exists = any(
|
||||||
|
dir_file["file_name"] == save_name for dir_file in dir_file_list
|
||||||
|
)
|
||||||
|
|
||||||
if not file_exists:
|
if not file_exists:
|
||||||
# 设置保存文件名（单独的{}不在这里重命名，而是在do_rename_task中处理）
|
# 设置保存文件名（单独的{}不在这里重命名，而是在do_rename_task中处理）
|
||||||
@ -1536,30 +1544,30 @@ class Quark:
|
|||||||
# 如果原文件名已包含任务名作为前缀，保持原样
|
# 如果原文件名已包含任务名作为前缀，保持原样
|
||||||
save_name = share_file["file_name"]
|
save_name = share_file["file_name"]
|
||||||
|
|
||||||
# 忽略后缀
|
# 为正则模式实现基于文件名的查重逻辑，支持忽略后缀选项
|
||||||
if task.get("ignore_extension") and not share_file["dir"]:
|
|
||||||
compare_func = lambda a, b1, b2: (
|
|
||||||
os.path.splitext(a)[0] == os.path.splitext(b1)[0]
|
|
||||||
or os.path.splitext(a)[0] == os.path.splitext(b2)[0]
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
compare_func = lambda a, b1, b2: (a == b1 or a == b2)
|
|
||||||
# 判断目标目录文件是否存在
|
# 判断目标目录文件是否存在
|
||||||
file_exists = False
|
file_exists = False
|
||||||
for dir_file in dir_file_list:
|
for dir_file in dir_file_list:
|
||||||
if compare_func(
|
if dir_file["dir"]:
|
||||||
dir_file["file_name"], share_file["file_name"], save_name
|
|
||||||
):
|
|
||||||
file_exists = True
|
|
||||||
# print(f"跳过已存在的文件: {dir_file['file_name']} - 与源文件或保存文件同名")
|
|
||||||
break
|
|
||||||
|
|
||||||
if not file_exists:
|
|
||||||
# 再次检查是否已经通过文件内容(大小+时间)被识别为重复
|
|
||||||
if is_duplicate and not share_file["dir"]: # 修改：只有非文件夹时才考虑重复过滤
|
|
||||||
# print(f"跳过已存在的文件: {share_file['file_name']} - 通过大小和时间匹配到相同文件")
|
|
||||||
continue
|
continue
|
||||||
|
|
||||||
|
if task.get("ignore_extension", False):
|
||||||
|
# 忽略后缀：只比较文件名部分，不管扩展名
|
||||||
|
original_name_base = os.path.splitext(share_file["file_name"])[0]
|
||||||
|
renamed_name_base = os.path.splitext(save_name)[0]
|
||||||
|
existing_name_base = os.path.splitext(dir_file["file_name"])[0]
|
||||||
|
|
||||||
|
# 如果原文件名或重命名后文件名与目标目录中文件名相同（忽略后缀），则视为已存在
|
||||||
|
if existing_name_base == original_name_base or existing_name_base == renamed_name_base:
|
||||||
|
file_exists = True
|
||||||
|
break
|
||||||
|
else:
|
||||||
|
# 不忽略后缀：文件名和扩展名都要一致才视为同一个文件
|
||||||
|
if dir_file["file_name"] == share_file["file_name"] or dir_file["file_name"] == save_name:
|
||||||
|
file_exists = True
|
||||||
|
break
|
||||||
|
|
||||||
|
if not file_exists:
|
||||||
# 不打印保存信息
|
# 不打印保存信息
|
||||||
share_file["save_name"] = save_name
|
share_file["save_name"] = save_name
|
||||||
share_file["original_name"] = share_file["file_name"] # 保存原文件名，用于排序
|
share_file["original_name"] = share_file["file_name"] # 保存原文件名，用于排序
|
||||||
@ -1972,50 +1980,51 @@ class Quark:
|
|||||||
filtered_share_files.append(share_file)
|
filtered_share_files.append(share_file)
|
||||||
continue
|
continue
|
||||||
|
|
||||||
# 检查文件是否已存在（基于大小和修改时间）
|
# 从共享文件中提取剧集号
|
||||||
file_size = share_file.get("size", 0)
|
episode_num = extract_episode_number_local(share_file["file_name"])
|
||||||
file_ext = os.path.splitext(share_file["file_name"])[1].lower()
|
|
||||||
share_update_time = share_file.get("last_update_at", 0) or share_file.get("updated_at", 0)
|
|
||||||
|
|
||||||
key = f"{file_size}_{file_ext}"
|
|
||||||
is_duplicate = False
|
is_duplicate = False
|
||||||
|
|
||||||
if key in dir_files_map:
|
# 通过文件名判断是否已存在（新的查重逻辑）
|
||||||
for existing_file in dir_files_map[key]:
|
if not is_duplicate:
|
||||||
existing_update_time = existing_file.get("updated_at", 0)
|
# 如果没有重命名，判断原文件名是否已存在
|
||||||
|
original_name = share_file["file_name"]
|
||||||
|
# 如果有剧集号，判断重命名后的文件名是否已存在
|
||||||
|
file_ext = os.path.splitext(original_name)[1]
|
||||||
|
|
||||||
# 防止除零错误
|
# 构建可能的新文件名
|
||||||
if existing_update_time == 0:
|
if episode_num is not None:
|
||||||
continue
|
if episode_pattern == "[]":
|
||||||
|
new_name = f"{episode_num:02d}{file_ext}"
|
||||||
# 如果修改时间相近（30天内）或者差距不大（10%以内），认为是同一个文件
|
else:
|
||||||
time_diff = abs(share_update_time - existing_update_time)
|
new_name = episode_pattern.replace("[]", f"{episode_num:02d}") + file_ext
|
||||||
time_ratio = abs(1 - (share_update_time / existing_update_time)) if existing_update_time else 1
|
|
||||||
|
|
||||||
if time_diff < 2592000 or time_ratio < 0.1:
|
|
||||||
# 文件已存在，跳过处理
|
|
||||||
is_duplicate = True
|
|
||||||
# print(f"跳过已存在的文件: {share_file['file_name']} (size={file_size}, time_diff={time_diff}s, ratio={time_ratio:.2f})")
|
|
||||||
break
|
|
||||||
|
|
||||||
# 检查剧集号是否已经存在
|
|
||||||
episode_num = extract_episode_number_local(share_file["file_name"])
|
|
||||||
if episode_num is not None and episode_num in existing_episode_numbers:
|
|
||||||
# print(f"跳过已存在的剧集号: {episode_num} ({share_file['file_name']})")
|
|
||||||
is_duplicate = True
|
|
||||||
|
|
||||||
# 生成预期的目标文件名并检查是否已存在
|
|
||||||
if episode_num is not None and not is_duplicate:
|
|
||||||
file_ext = os.path.splitext(share_file["file_name"])[1]
|
|
||||||
if episode_pattern == "[]":
|
|
||||||
# 对于单独的[]，直接使用数字序号作为文件名
|
|
||||||
expected_name = f"{episode_num:02d}{file_ext}"
|
|
||||||
else:
|
else:
|
||||||
expected_name = episode_pattern.replace("[]", f"{episode_num:02d}") + file_ext
|
new_name = None
|
||||||
# 检查目标文件名是否存在于目录中
|
|
||||||
if any(dir_file["file_name"] == expected_name for dir_file in dir_file_list):
|
# 根据是否忽略后缀进行检查
|
||||||
# print(f"跳过已存在的文件名: {expected_name}")
|
if task.get("ignore_extension", False):
|
||||||
is_duplicate = True
|
# 忽略后缀模式：只比较文件名，不比较扩展名
|
||||||
|
original_name_base = os.path.splitext(original_name)[0]
|
||||||
|
|
||||||
|
# 检查原文件名是否存在
|
||||||
|
exists_original = any(os.path.splitext(dir_file["file_name"])[0] == original_name_base for dir_file in dir_file_list)
|
||||||
|
|
||||||
|
# 如果有新文件名，检查新文件名是否存在
|
||||||
|
exists_new = False
|
||||||
|
if new_name:
|
||||||
|
new_name_base = os.path.splitext(new_name)[0]
|
||||||
|
exists_new = any(os.path.splitext(dir_file["file_name"])[0] == new_name_base for dir_file in dir_file_list)
|
||||||
|
|
||||||
|
is_duplicate = exists_original or exists_new
|
||||||
|
else:
|
||||||
|
# 不忽略后缀模式：文件名和扩展名都要一致
|
||||||
|
exists_original = any(dir_file["file_name"] == original_name for dir_file in dir_file_list)
|
||||||
|
|
||||||
|
# 如果有新文件名，检查新文件名是否存在
|
||||||
|
exists_new = False
|
||||||
|
if new_name:
|
||||||
|
exists_new = any(dir_file["file_name"] == new_name for dir_file in dir_file_list)
|
||||||
|
|
||||||
|
is_duplicate = exists_original or exists_new
|
||||||
|
|
||||||
# 只处理非重复文件
|
# 只处理非重复文件
|
||||||
if not is_duplicate:
|
if not is_duplicate:
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user