chickliu/quark-auto-save
1
0
Fork 0
mirror of https://github.com/Cp0204/quark-auto-save.git synced 2026-01-17 18:30:43 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

✨ 正则替换表达式支持 $TASKNAME 变量

Browse Source
This commit is contained in:
Cp0204 2024-11-02 16:29:49 +08:00
parent 86f8d3bbf6
commit 20ee540081
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
quark_auto_save.py
View File

@ -39,12 +39,14 @@ MAGIC_REGEX = {
# 魔法正则匹配 # 魔法正则匹配
def magic_regex_func(pattern, replace): def magic_regex_func(pattern, replace, taskname=""):
keyword = pattern keyword = pattern
if keyword in CONFIG_DATA["magic_regex"]: if keyword in CONFIG_DATA["magic_regex"]:
pattern = CONFIG_DATA["magic_regex"][keyword]["pattern"] pattern = CONFIG_DATA["magic_regex"][keyword]["pattern"]
if replace == "": if replace == "":
replace = CONFIG_DATA["magic_regex"][keyword]["replace"] replace = CONFIG_DATA["magic_regex"][keyword]["replace"]
if taskname:
replace = replace.replace("$TASKNAME", taskname)
return pattern, replace return pattern, replace
@ -531,7 +533,9 @@ class Quark:
if share_file["dir"] and task.get("update_subdir", False): if share_file["dir"] and task.get("update_subdir", False):
pattern, replace = task["update_subdir"], "" pattern, replace = task["update_subdir"], ""
else: else:
pattern, replace = magic_regex_func(task["pattern"], task["replace"]) pattern, replace = magic_regex_func(
task["pattern"], task["replace"], task["taskname"]
)
# 正则文件名匹配 # 正则文件名匹配
if re.search(pattern, share_file["file_name"]): if re.search(pattern, share_file["file_name"]):
# 替换后的文件名 # 替换后的文件名
@ -648,7 +652,9 @@ class Quark:
return response return response
def do_rename_task(self, task, subdir_path=""): def do_rename_task(self, task, subdir_path=""):
pattern, replace = magic_regex_func(task["pattern"], task["replace"]) pattern, replace = magic_regex_func(
task["pattern"], task["replace"], task["taskname"]
)
if not pattern or not replace: if not pattern or not replace:
return 0 return 0
savepath = re.sub(r"/{2,}", "/", f"/{task['savepath']}{subdir_path}") savepath = re.sub(r"/{2,}", "/", f"/{task['savepath']}{subdir_path}")