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

更改保存目录 fid 的存取方式

Browse Source
This commit is contained in:
Cp0204 2024-04-04 01:22:19 +08:00
parent a8a6675eee
commit 518c6112b8
1 changed files with 11 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
quark_auto_save.py
View File

@ -93,6 +93,7 @@ class Quark:
self.is_active = False self.is_active = False
self.nickname = "" self.nickname = ""
self.st = self.match_st_form_cookie(cookie) self.st = self.match_st_form_cookie(cookie)
self.savepath_fid = {"/": "0"}
def match_st_form_cookie(self, cookie): def match_st_form_cookie(self, cookie):
match = re.search(r"=(st[a-zA-Z0-9]+);", cookie) match = re.search(r"=(st[a-zA-Z0-9]+);", cookie)
@ -388,14 +389,9 @@ class Quark:
print(f"创建文件夹：{dir_path}") print(f"创建文件夹：{dir_path}")
else: else:
print(f"创建文件夹：{dir_path} 失败, {mkdir_return['message']}") print(f"创建文件夹：{dir_path} 失败, {mkdir_return['message']}")
# 更新到配置 # 储存目标目录的fid
for task in tasklist: for dir_path in dir_paths_exist_arr:
if task["savepath"] == "/": self.savepath_fid[dir_path["file_path"]] = dir_path["fid"]
task["savepath_fid"] = "0"
else:
for dir_path in dir_paths_exist_arr:
if task["savepath"] == dir_path["file_path"]:
task["savepath_fid"] = dir_path["fid"]
# print(dir_paths_exist_arr) # print(dir_paths_exist_arr)
def do_save_check(self, shareurl, savepath): def do_save_check(self, shareurl, savepath):
@ -408,9 +404,9 @@ class Quark:
file_name_list = [item["file_name"] for item in share_file_list] file_name_list = [item["file_name"] for item in share_file_list]
if not fid_list: if not fid_list:
return return
self.mkdir(savepath) get_fids = self.get_fids([savepath])
to_pdir_fid = ( to_pdir_fid = (
"0" if savepath == "/" else self.get_fids([savepath])[0]["fid"] get_fids[0]["fid"] if get_fids else self.mkdir(savepath)["data"]["fid"]
) )
save_file = self.save_file( save_file = self.save_file(
fid_list, fid_token_list, to_pdir_fid, pwd_id, stoken fid_list, fid_token_list, to_pdir_fid, pwd_id, stoken
@ -471,9 +467,10 @@ class Quark:
# print("share_file_list: ", share_file_list) # print("share_file_list: ", share_file_list)
# 获取目标目录文件列表 # 获取目标目录文件列表
if not task.get("savepath_fid"): savepath = task["savepath"]
task["savepath_fid"] = self.get_fids([task["savepath"]])[0]["fid"] if not self.savepath_fid.get(savepath):
to_pdir_fid = task["savepath_fid"] self.savepath_fid[savepath] = self.get_fids([savepath])[0]["fid"]
to_pdir_fid = self.savepath_fid[savepath]
dir_file_list = self.ls_dir(to_pdir_fid) dir_file_list = self.ls_dir(to_pdir_fid)
# print("dir_file_list: ", dir_file_list) # print("dir_file_list: ", dir_file_list)
@ -554,7 +551,7 @@ class Quark:
return response return response
def do_rename_task(self, task): def do_rename_task(self, task):
dir_file_list = self.ls_dir(task["savepath_fid"]) dir_file_list = self.ls_dir(self.savepath_fid[task["savepath"]])
is_rename = False is_rename = False
for dir_file in dir_file_list: for dir_file in dir_file_list:
pattern, replace = magic_regex_func(task["pattern"], task["replace"]) pattern, replace = magic_regex_func(task["pattern"], task["replace"])