From 166e18fcfab47db6056ac3f4653c5649975771e5 Mon Sep 17 00:00:00 2001 From: Cp0204 Date: Wed, 3 Apr 2024 00:53:05 +0800 Subject: [PATCH] =?UTF-8?q?=E9=9D=A2=E5=90=91=E5=AF=B9=E8=B1=A1?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- quark_auto_save.py | 1108 ++++++++++++++++++++++++-------------------- 1 file changed, 617 insertions(+), 491 deletions(-) diff --git a/quark_auto_save.py b/quark_auto_save.py index 931e872..1ce3aee 100644 --- a/quark_auto_save.py +++ b/quark_auto_save.py @@ -1,6 +1,6 @@ # !/usr/bin/env python3 # -*- coding: utf-8 -*- -# Modify: 2024-02-29 +# Modify: 2024-04-03 # Repo: https://github.com/Cp0204/quark_auto_save # ConfigFile: quark_config.json """ @@ -16,8 +16,9 @@ import requests from datetime import datetime config_data = {} +check_data = {} notifys = [] -first_account = {} + magic_regex = { "$TV": { @@ -61,445 +62,7 @@ def add_notify(text): return text -def common_headers(): - return { - "cookie": first_account["cookie"], - "content-type": "application/json", - } - - -def get_growth_info(cookie): - url = "https://drive-m.quark.cn/1/clouddrive/capacity/growth/info" - querystring = {"pr": "ucpro", "fr": "pc", "uc_param_str": ""} - headers = { - "cookie": cookie, - "content-type": "application/json", - } - response = requests.request("GET", url, headers=headers, params=querystring).json() - if response.get("data"): - return response["data"] - else: - return False - - -def get_growth_sign(cookie): - url = "https://drive-m.quark.cn/1/clouddrive/capacity/growth/sign" - querystring = {"pr": "ucpro", "fr": "pc", "uc_param_str": ""} - payload = { - "sign_cyclic": True, - } - headers = { - "cookie": cookie, - "content-type": "application/json", - } - response = requests.request( - "POST", url, json=payload, headers=headers, params=querystring - ).json() - if response.get("data"): - return True, response["data"]["sign_daily_reward"] - else: - return False, response["message"] - - -def get_id_from_url(url): - pattern = r"/s/(\w+)(#/list/share.*/(\w+))?" - match = re.search(pattern, url) - if match: - pwd_id = match.group(1) - if match.group(2): - pdir_fid = match.group(3) - else: - pdir_fid = 0 - return pwd_id, pdir_fid - else: - return None - - -def get_account_info(cookie): - url = "https://pan.quark.cn/account/info" - querystring = {"fr": "pc", "platform": "pc"} - headers = { - "cookie": cookie, - "content-type": "application/json", - } - response = requests.request("GET", url, headers=headers, params=querystring).json() - if response.get("data"): - return response["data"] - else: - return False - - -# 可验证资源是否失效 -def get_stoken(pwd_id): - url = "https://pan.quark.cn/1/clouddrive/share/sharepage/token" - querystring = {"pr": "ucpro", "fr": "h5"} - payload = {"pwd_id": pwd_id, "passcode": ""} - headers = common_headers() - response = requests.request( - "POST", url, json=payload, headers=headers, params=querystring - ).json() - if response.get("data"): - return True, response["data"]["stoken"] - else: - return False, response["message"] - - -def get_detail(pwd_id, stoken, pdir_fid): - file_list = [] - page = 1 - while True: - url = "https://pan.quark.cn/1/clouddrive/share/sharepage/detail" - querystring = { - "pr": "ucpro", - "fr": "pc", - "pwd_id": pwd_id, - "stoken": stoken, - "pdir_fid": pdir_fid, - "force": "0", - "_page": page, - "_size": "50", - "_fetch_banner": "0", - "_fetch_share": "0", - "_fetch_total": "1", - "_sort": "file_type:asc,updated_at:desc", - } - headers = common_headers() - response = requests.request( - "GET", url, headers=headers, params=querystring - ).json() - if response["data"]["list"]: - file_list += response["data"]["list"] - page += 1 - else: - break - if len(file_list) >= response["metadata"]["_total"]: - break - # 仅有一个文件夹 - if len(file_list) == 1 and file_list[0]["file_type"] == 0: - print("🧠 该分享是一个文件夹，读取文件夹内列表") - file_list = get_detail(pwd_id, stoken, file_list[0]["fid"]) - return file_list - - -def get_fids(file_paths): - fids = [] - while True: - url = "https://drive.quark.cn/1/clouddrive/file/info/path_list" - querystring = {"pr": "ucpro", "fr": "pc"} - payload = {"file_path": file_paths[:50], "namespace": "0"} - headers = common_headers() - response = requests.request( - "POST", url, json=payload, headers=headers, params=querystring - ).json() - if response["code"] == 0: - fids += response["data"] - file_paths = file_paths[50:] - else: - print(f"获取目录ID：失败, {response['message']}") - break - if len(file_paths) == 0: - break - return fids - - -def ls_dir(pdir_fid): - file_list = [] - page = 1 - while True: - url = "https://drive.quark.cn/1/clouddrive/file/sort" - querystring = { - "pr": "ucpro", - "fr": "pc", - "uc_param_str": "", - "pdir_fid": pdir_fid, - "_page": page, - "_size": "50", - "_fetch_total": "1", - "_fetch_sub_dirs": "0", - "_sort": "file_type:asc,updated_at:desc", - } - headers = common_headers() - response = requests.request( - "GET", url, headers=headers, params=querystring - ).json() - if response["data"]["list"]: - file_list += response["data"]["list"] - page += 1 - else: - break - if len(file_list) >= response["metadata"]["_total"]: - break - return file_list - - -def save_file(fid_list, fid_token_list, to_pdir_fid, pwd_id, stoken): - url = "https://drive.quark.cn/1/clouddrive/share/sharepage/save" - querystring = { - "pr": "ucpro", - "fr": "pc", - "uc_param_str": "", - "__dt": int(random.uniform(1, 5) * 60 * 1000), - "__t": datetime.now().timestamp(), - } - payload = { - "fid_list": fid_list, - "fid_token_list": fid_token_list, - "to_pdir_fid": to_pdir_fid, - "pwd_id": pwd_id, - "stoken": stoken, - "pdir_fid": "0", - "scene": "link", - } - headers = common_headers() - response = requests.request( - "POST", url, json=payload, headers=headers, params=querystring - ).json() - return response - - -def mkdir(dir_path): - url = "https://drive-pc.quark.cn/1/clouddrive/file" - querystring = {"pr": "ucpro", "fr": "pc", "uc_param_str": ""} - payload = { - "pdir_fid": "0", - "file_name": "", - "dir_path": dir_path, - "dir_init_lock": False, - } - headers = common_headers() - response = requests.request( - "POST", url, json=payload, headers=headers, params=querystring - ).json() - return response - - -def rename(fid, file_name): - url = "https://drive-pc.quark.cn/1/clouddrive/file/rename" - querystring = {"pr": "ucpro", "fr": "pc", "uc_param_str": ""} - payload = {"fid": fid, "file_name": file_name} - headers = common_headers() - response = requests.request( - "POST", url, json=payload, headers=headers, params=querystring - ).json() - return response - - -def update_savepath_fid(tasklist): - dir_paths = [ - item["savepath"] - for item in tasklist - if not item.get("enddate") - or ( - datetime.now().date() - <= datetime.strptime(item["enddate"], "%Y-%m-%d").date() - ) - ] - if not dir_paths: - return False - dir_paths_exist_arr = get_fids(dir_paths) - dir_paths_exist = [item["file_path"] for item in dir_paths_exist_arr] - # 比较创建不存在的 - dir_paths_unexist = list(set(dir_paths) - set(dir_paths_exist)) - for dir_path in dir_paths_unexist: - mkdir_return = mkdir(dir_path) - if mkdir_return["code"] == 0: - new_dir = mkdir_return["data"] - dir_paths_exist_arr.append({"file_path": dir_path, "fid": new_dir["fid"]}) - print(f"创建文件夹：{dir_path}") - else: - print(f"创建文件夹：{dir_path} 失败, {mkdir_return['message']}") - # 更新到配置 - for task in tasklist: - 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) - - -def do_save_task(task): - # 判断资源失效记录 - if task.get("shareurl_ban"): - print(f"《{task['taskname']}》：{task['shareurl_ban']}") - return - - # 链接转换所需参数 - pwd_id, pdir_fid = get_id_from_url(task["shareurl"]) - # print("match: ", pwd_id, pdir_fid) - - # 获取stoken，同时可验证资源是否失效 - is_sharing, stoken = get_stoken(pwd_id) - if not is_sharing: - add_notify(f"《{task['taskname']}》：{stoken}") - task["shareurl_ban"] = stoken - return - # print("stoken: ", stoken) - - # 获取分享文件列表 - share_file_list = get_detail(pwd_id, stoken, pdir_fid) - if not share_file_list: - add_notify(f"《{task['taskname']}》：分享目录为空") - return - # print("share_file_list: ", share_file_list) - - # 获取目标目录文件列表 - task["savepath_fid"] = ( - task.get("savepath_fid") - if task.get("savepath_fid") - else get_fids([task["savepath"]])[0]["fid"] - ) - to_pdir_fid = task["savepath_fid"] - dir_file_list = ls_dir(to_pdir_fid) - # print("dir_file_list: ", dir_file_list) - - # 需保存的文件清单 - need_save_list = [] - # 添加符合的 - for share_file in share_file_list: - # 正则文件名匹配 - pattern, replace = magic_regex_func(task["pattern"], task["replace"]) - if re.search(pattern, share_file["file_name"]): - # 替换后的文件名 - save_name = ( - re.sub(pattern, replace, share_file["file_name"]) - if replace != "" - else share_file["file_name"] - ) - # 判断目标目录文件是否存在，可选忽略后缀 - if task.get("ignore_extension"): - 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 = any( - compare_func(dir_file["file_name"], share_file["file_name"], save_name) - for dir_file in dir_file_list - ) - if not file_exists: - share_file["save_name"] = save_name - need_save_list.append(share_file) - - fid_list = [item["fid"] for item in need_save_list] - fid_token_list = [item["share_fid_token"] for item in need_save_list] - save_name_list = [item["save_name"] for item in need_save_list] - if fid_list: - save_file_return = save_file( - fid_list, fid_token_list, to_pdir_fid, pwd_id, stoken - ) - if save_file_return["code"] == 0: - task_id = save_file_return["data"]["task_id"] - query_task_return = query_task(task_id) - if query_task_return["code"] == 0: - save_name_list.sort() - add_notify( - f"《{task['taskname']}》添加追更：{', '.join(save_name_list)}" - ) - return True - else: - err_msg = query_task_return["message"] - else: - err_msg = save_file_return["message"] - add_notify(f"《{task['taskname']}》转存失败：{err_msg}") - return False - else: - print("任务结束：没有新的转存任务") - return False - - -def query_task(task_id): - url = "https://drive-pc.quark.cn/1/clouddrive/task" - querystring = { - "pr": "ucpro", - "fr": "pc", - "uc_param_str": "", - "task_id": task_id, - "retry_index": "1", - "__dt": int(random.uniform(1, 5) * 60 * 1000), - "__t": datetime.now().timestamp(), - } - headers = common_headers() - response = requests.request("GET", url, headers=headers, params=querystring).json() - if response["code"] == 32003: - response["message"] = "容量限制" - return response - - -def do_rename_task(task): - dir_file_list = ls_dir(task["savepath_fid"]) - is_rename = False - for dir_file in dir_file_list: - pattern, replace = magic_regex_func(task["pattern"], task["replace"]) - if re.search(pattern, dir_file["file_name"]): - save_name = ( - re.sub(pattern, replace, dir_file["file_name"]) - if replace != "" - else dir_file["file_name"] - ) - if save_name != dir_file["file_name"]: - rename_return = rename(dir_file["fid"], save_name) - if rename_return["code"] == 0: - print(f"重命名：{dir_file['file_name']} → {save_name}") - is_rename = True - else: - print( - f"重命名：{dir_file['file_name']} → {save_name} 失败，{rename_return['message']}" - ) - return is_rename - - -def emby_refresh(emby_id): - emby_url = config_data.get("emby").get("url") - emby_apikey = config_data.get("emby").get("apikey") - if emby_url and emby_apikey and emby_id: - url = f"{emby_url}/emby/Items/{emby_id}/Refresh" - headers = {"X-Emby-Token": emby_apikey} - querystring = { - "Recursive": "true", - "MetadataRefreshMode": "FullRefresh", - "ImageRefreshMode": "FullRefresh", - "ReplaceAllMetadata": "false", - "ReplaceAllImages": "false", - } - response = requests.request("POST", url, headers=headers, params=querystring) - if response.text == "": - print(f"🎞 刷新Emby媒体库：成功✅") - return True - else: - print(f"🎞 刷新Emby媒体库：{response.text}❌") - return False - - -def emby_search(media_name): - emby_url = config_data.get("emby").get("url") - emby_apikey = config_data.get("emby").get("apikey") - if emby_url and emby_apikey and media_name: - url = f"{emby_url}/emby/Items" - headers = {"X-Emby-Token": emby_apikey} - querystring = { - "IncludeItemTypes": "Series", - "StartIndex": 0, - "SortBy": "SortName", - "SortOrder": "Ascending", - "ImageTypeLimit": 0, - "Recursive": "true", - "SearchTerm": media_name, - "Limit": 10, - "IncludeSearchTypes": "false", - } - response = requests.request("GET", url, headers=headers, params=querystring) - if "application/json" in response.headers["Content-Type"]: - response = response.json() - if response.get("Items"): - for item in response["Items"]: - if item["IsFolder"]: - print(f"🎞 《{item['Name']}》匹配到Emby媒体库ID：{item['Id']}") - return item["Id"] - else: - print(f"🎞 搜索Emby媒体库：{response.text}❌") - return False - - +# 下载配置 def download_file(url, save_path): response = requests.get(url) if response.status_code == 200: @@ -510,6 +73,7 @@ def download_file(url, save_path): return False +# 读取CK def get_cookies(cookie_val): if isinstance(cookie_val, list): return cookie_val @@ -522,56 +86,611 @@ def get_cookies(cookie_val): return False -def do_sign(cookies): - first_account = {} - print(f"===============签到任务===============") - for index, cookie in enumerate(cookies): - # 验证账号 - account_info = get_account_info(cookie) - print(f"▶️ 验证第{index+1}个账号") - if not account_info: - add_notify(f"👤 第{index+1}个账号登录失败，cookie无效❌") +class Quark: + def __init__(self, cookie, index=None): + self.cookie = cookie + self.index = index + 1 + self.is_active = False + self.nickname = "" + + def common_headers(self): + return { + "cookie": self.cookie, + "content-type": "application/json", + } + + def init(self): + account_info = self.get_account_info() + if account_info: + self.is_active = True + self.nickname = account_info["nickname"] + return account_info else: - if index == 0: - first_account = account_info - first_account["cookie"] = cookie - print(f"👤 账号昵称: {account_info['nickname']}✅") - # 每日领空间 - growth_info = get_growth_info(cookie) - if growth_info: - if growth_info["cap_sign"]["sign_daily"]: - print( - f"📅 执行签到: 今日已签到+{int(growth_info['cap_sign']['sign_daily_reward']/1024/1024)}MB，连签进度({growth_info['cap_sign']['sign_progress']}/{growth_info['cap_sign']['sign_target']})✅" + return False + + def get_account_info(self): + url = "https://pan.quark.cn/account/info" + querystring = {"fr": "pc", "platform": "pc"} + headers = { + "cookie": self.cookie, + "content-type": "application/json", + } + response = requests.request( + "GET", url, headers=headers, params=querystring + ).json() + if response.get("data"): + return response["data"] + else: + return False + + def get_growth_info(self): + url = "https://drive-m.quark.cn/1/clouddrive/capacity/growth/info" + querystring = {"pr": "ucpro", "fr": "pc", "uc_param_str": ""} + headers = { + "cookie": self.cookie, + "content-type": "application/json", + } + response = requests.request( + "GET", url, headers=headers, params=querystring + ).json() + if response.get("data"): + return response["data"] + else: + return False + + def get_growth_sign(self): + url = "https://drive-m.quark.cn/1/clouddrive/capacity/growth/sign" + querystring = {"pr": "ucpro", "fr": "pc", "uc_param_str": ""} + payload = { + "sign_cyclic": True, + } + headers = { + "cookie": self.cookie, + "content-type": "application/json", + } + response = requests.request( + "POST", url, json=payload, headers=headers, params=querystring + ).json() + if response.get("data"): + return True, response["data"]["sign_daily_reward"] + else: + return False, response["message"] + + def get_id_from_url(self, url): + url = url.replace("https://pan.quark.cn/s/", "") + pattern = r"(\w+)(#/list/share.*/(\w+))?" + match = re.search(pattern, url) + if match: + pwd_id = match.group(1) + if match.group(2): + pdir_fid = match.group(3) + else: + pdir_fid = 0 + return pwd_id, pdir_fid + else: + return None + + # 可验证资源是否失效 + def get_stoken(self, pwd_id): + url = "https://pan.quark.cn/1/clouddrive/share/sharepage/token" + querystring = {"pr": "ucpro", "fr": "h5"} + payload = {"pwd_id": pwd_id, "passcode": ""} + headers = self.common_headers() + response = requests.request( + "POST", url, json=payload, headers=headers, params=querystring + ).json() + if response.get("data"): + return True, response["data"]["stoken"] + else: + return False, response["message"] + + def get_detail(self, pwd_id, stoken, pdir_fid): + file_list = [] + page = 1 + while True: + url = "https://pan.quark.cn/1/clouddrive/share/sharepage/detail" + querystring = { + "pr": "ucpro", + "fr": "pc", + "pwd_id": pwd_id, + "stoken": stoken, + "pdir_fid": pdir_fid, + "force": "0", + "_page": page, + "_size": "50", + "_fetch_banner": "0", + "_fetch_share": "0", + "_fetch_total": "1", + "_sort": "file_type:asc,updated_at:desc", + } + headers = self.common_headers() + response = requests.request( + "GET", url, headers=headers, params=querystring + ).json() + if response["data"]["list"]: + file_list += response["data"]["list"] + page += 1 + else: + break + if len(file_list) >= response["metadata"]["_total"]: + break + return file_list + + def get_fids(self, file_paths): + fids = [] + while True: + url = "https://drive.quark.cn/1/clouddrive/file/info/path_list" + querystring = {"pr": "ucpro", "fr": "pc"} + payload = {"file_path": file_paths[:50], "namespace": "0"} + headers = self.common_headers() + response = requests.request( + "POST", url, json=payload, headers=headers, params=querystring + ).json() + if response["code"] == 0: + fids += response["data"] + file_paths = file_paths[50:] + else: + print(f"获取目录ID：失败, {response['message']}") + break + if len(file_paths) == 0: + break + return fids + + def ls_dir(self, pdir_fid): + file_list = [] + page = 1 + while True: + url = "https://drive.quark.cn/1/clouddrive/file/sort" + querystring = { + "pr": "ucpro", + "fr": "pc", + "uc_param_str": "", + "pdir_fid": pdir_fid, + "_page": page, + "_size": "50", + "_fetch_total": "1", + "_fetch_sub_dirs": "0", + "_sort": "file_type:asc,updated_at:desc", + } + headers = self.common_headers() + response = requests.request( + "GET", url, headers=headers, params=querystring + ).json() + if response["data"]["list"]: + file_list += response["data"]["list"] + page += 1 + else: + break + if len(file_list) >= response["metadata"]["_total"]: + break + return file_list + + def save_file(self, fid_list, fid_token_list, to_pdir_fid, pwd_id, stoken): + url = "https://drive.quark.cn/1/clouddrive/share/sharepage/save" + querystring = { + "pr": "ucpro", + "fr": "pc", + "uc_param_str": "", + "__dt": int(random.uniform(1, 5) * 60 * 1000), + "__t": datetime.now().timestamp(), + } + payload = { + "fid_list": fid_list, + "fid_token_list": fid_token_list, + "to_pdir_fid": to_pdir_fid, + "pwd_id": pwd_id, + "stoken": stoken, + "pdir_fid": "0", + "scene": "link", + } + headers = self.common_headers() + response = requests.request( + "POST", url, json=payload, headers=headers, params=querystring + ).json() + return response + + def mkdir(self, dir_path): + url = "https://drive-pc.quark.cn/1/clouddrive/file" + querystring = {"pr": "ucpro", "fr": "pc", "uc_param_str": ""} + payload = { + "pdir_fid": "0", + "file_name": "", + "dir_path": dir_path, + "dir_init_lock": False, + } + headers = self.common_headers() + response = requests.request( + "POST", url, json=payload, headers=headers, params=querystring + ).json() + return response + + def rename(self, fid, file_name): + url = "https://drive-pc.quark.cn/1/clouddrive/file/rename" + querystring = {"pr": "ucpro", "fr": "pc", "uc_param_str": ""} + payload = {"fid": fid, "file_name": file_name} + headers = self.common_headers() + response = requests.request( + "POST", url, json=payload, headers=headers, params=querystring + ).json() + return response + + def delete(self, filelist): + url = "https://drive-pc.quark.cn/1/clouddrive/file/delete" + querystring = {"pr": "ucpro", "fr": "pc", "uc_param_str": ""} + payload = {"action_type": 2, "filelist": filelist, "exclude_fids": []} + headers = self.common_headers() + response = requests.request( + "POST", url, json=payload, headers=headers, params=querystring + ).json() + return response + + def recycle_list(self, page=1, size=30): + url = "https://drive-m.quark.cn/1/clouddrive/file/recycle/list" + querystring = { + "_page": page, + "_size": size, + "pr": "ucpro", + "fr": "pc", + "uc_param_str": "", + } + headers = self.common_headers() + response = requests.request( + "GET", url, headers=headers, params=querystring + ).json() + return response["data"]["list"] + + def recycle_remove(self, record_list): + url = "https://drive-m.quark.cn/1/clouddrive/file/recycle/remove" + querystring = {"uc_param_str": "", "fr": "pc", "pr": "ucpro"} + payload = { + "select_mode": 2, + "record_list": record_list, + } + headers = self.common_headers() + response = requests.request( + "POST", url, json=payload, headers=headers, params=querystring + ).json() + return response + + def update_savepath_fid(self, tasklist): + dir_paths = [ + item["savepath"] + for item in tasklist + if not item.get("enddate") + or ( + datetime.now().date() + <= datetime.strptime(item["enddate"], "%Y-%m-%d").date() + ) + ] + if not dir_paths: + return False + dir_paths_exist_arr = self.get_fids(dir_paths) + dir_paths_exist = [item["file_path"] for item in dir_paths_exist_arr] + # 比较创建不存在的 + dir_paths_unexist = list(set(dir_paths) - set(dir_paths_exist)) + for dir_path in dir_paths_unexist: + mkdir_return = self.mkdir(dir_path) + if mkdir_return["code"] == 0: + new_dir = mkdir_return["data"] + dir_paths_exist_arr.append( + {"file_path": dir_path, "fid": new_dir["fid"]} + ) + print(f"创建文件夹：{dir_path}") + else: + print(f"创建文件夹：{dir_path} 失败, {mkdir_return['message']}") + # 更新到配置 + for task in tasklist: + 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) + + def do_save_check(self, shareurl, savepath): + try: + pwd_id, pdir_fid = self.get_id_from_url(shareurl) + is_sharing, stoken = self.get_stoken(pwd_id) + share_file_list = self.get_detail(pwd_id, stoken, pdir_fid) + fid_list = [item["fid"] for item in share_file_list] + fid_token_list = [item["share_fid_token"] for item in share_file_list] + file_name_list = [item["file_name"] for item in share_file_list] + if not fid_list: + return + self.mkdir(savepath) + to_pdir_fid = ( + "0" if savepath == "/" else self.get_fids([savepath])[0]["fid"] + ) + save_file = self.save_file( + fid_list, fid_token_list, to_pdir_fid, pwd_id, stoken + ) + if save_file["code"] == 41017: + return + elif save_file["code"] == 0: + dir_file_list = self.ls_dir(to_pdir_fid) + del_list = [ + item["fid"] + for item in dir_file_list + if (item["file_name"] in file_name_list) + and ((datetime.now().timestamp() - item["created_at"]) < 60) + ] + if del_list: + self.delete(del_list) + recycle_list = self.recycle_list() + record_id_list = [ + item["record_id"] + for item in recycle_list + if item["fid"] in del_list + ] + self.recycle_remove(record_id_list) + return save_file + else: + return False + except Exception as e: + if os.environ.get("DEBUG") == True: + print(f"转存测试失败: {str(e)}") + + def do_save_task(self, task): + # 判断资源失效记录 + if task.get("shareurl_ban"): + print(f"《{task['taskname']}》：{task['shareurl_ban']}") + return + + # 链接转换所需参数 + pwd_id, pdir_fid = self.get_id_from_url(task["shareurl"]) + # print("match: ", pwd_id, pdir_fid) + + # 获取stoken，同时可验证资源是否失效 + is_sharing, stoken = self.get_stoken(pwd_id) + if not is_sharing: + add_notify(f"《{task['taskname']}》：{stoken}") + task["shareurl_ban"] = stoken + return + # print("stoken: ", stoken) + + # 获取分享文件列表 + share_file_list = self.get_detail(pwd_id, stoken, pdir_fid) + # 仅有一个文件夹 + if len(share_file_list) == 1 and share_file_list[0]["dir"]: + print("🧠 该分享是一个文件夹，读取文件夹内列表") + share_file_list = self.get_detail(pwd_id, stoken, share_file_list[0]["fid"]) + if not share_file_list: + add_notify(f"《{task['taskname']}》：分享目录为空") + return + # print("share_file_list: ", share_file_list) + + # 获取目标目录文件列表 + task["savepath_fid"] = ( + task.get("savepath_fid") + if task.get("savepath_fid") + else self.get_fids([task["savepath"]])[0]["fid"] + ) + to_pdir_fid = task["savepath_fid"] + dir_file_list = self.ls_dir(to_pdir_fid) + # print("dir_file_list: ", dir_file_list) + + # 需保存的文件清单 + need_save_list = [] + # 添加符合的 + for share_file in share_file_list: + # 正则文件名匹配 + pattern, replace = magic_regex_func(task["pattern"], task["replace"]) + if re.search(pattern, share_file["file_name"]): + # 替换后的文件名 + save_name = ( + re.sub(pattern, replace, share_file["file_name"]) + if replace != "" + else share_file["file_name"] + ) + # 判断目标目录文件是否存在，可选忽略后缀 + if task.get("ignore_extension"): + 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: - sign, sign_return = get_growth_sign(cookie) - if sign: - message = f"📅 执行签到: 今日签到+{int(sign_return/1024/1024)}MB，连签进度({growth_info['cap_sign']['sign_progress']+1}/{growth_info['cap_sign']['sign_target']})✅" - if ( - config_data.get("push_config").get("QUARK_SIGN_NOTIFY") - == False - or os.environ.get("QUARK_SIGN_NOTIFY") == False - ): - print(message) - else: - message = message.replace( - "今日", f"[{account_info['nickname']}]今日" - ) - add_notify(message) + compare_func = lambda a, b1, b2: (a == b1 or a == b2) + file_exists = any( + compare_func( + dir_file["file_name"], share_file["file_name"], save_name + ) + for dir_file in dir_file_list + ) + if not file_exists: + share_file["save_name"] = save_name + need_save_list.append(share_file) + + fid_list = [item["fid"] for item in need_save_list] + fid_token_list = [item["share_fid_token"] for item in need_save_list] + save_name_list = [item["save_name"] for item in need_save_list] + if fid_list: + save_file_return = self.save_file( + fid_list, fid_token_list, to_pdir_fid, pwd_id, stoken + ) + if save_file_return["code"] == 0: + task_id = save_file_return["data"]["task_id"] + query_task_return = self.query_task(task_id) + if query_task_return["code"] == 0: + save_name_list.sort() + add_notify( + f"《{task['taskname']}》添加追更：{', '.join(save_name_list)}" + ) + return True + else: + err_msg = query_task_return["message"] + else: + err_msg = save_file_return["message"] + add_notify(f"《{task['taskname']}》转存失败：{err_msg}") + return False + else: + print("任务结束：没有新的转存任务") + return False + + def query_task(self, task_id): + url = "https://drive-pc.quark.cn/1/clouddrive/task" + querystring = { + "pr": "ucpro", + "fr": "pc", + "uc_param_str": "", + "task_id": task_id, + "retry_index": "1", + "__dt": int(random.uniform(1, 5) * 60 * 1000), + "__t": datetime.now().timestamp(), + } + headers = self.common_headers() + response = requests.request( + "GET", url, headers=headers, params=querystring + ).json() + if response["code"] == 32003: + response["message"] = "容量超限" + return response + + def do_rename_task(self, task): + dir_file_list = self.ls_dir(task["savepath_fid"]) + is_rename = False + for dir_file in dir_file_list: + pattern, replace = magic_regex_func(task["pattern"], task["replace"]) + if re.search(pattern, dir_file["file_name"]): + save_name = ( + re.sub(pattern, replace, dir_file["file_name"]) + if replace != "" + else dir_file["file_name"] + ) + if save_name != dir_file["file_name"]: + rename_return = self.rename(dir_file["fid"], save_name) + if rename_return["code"] == 0: + print(f"重命名：{dir_file['file_name']} → {save_name}") + is_rename = True else: - print(f"📅 执行签到: {sign_return}") - print(f"") + print( + f"重命名：{dir_file['file_name']} → {save_name} 失败，{rename_return['message']}" + ) + return is_rename + + +class Emby: + def __init__(self, emby_url, emby_apikey): + if emby_url and emby_apikey: + self.emby_url = emby_url + self.emby_apikey = emby_apikey + else: + return False + + def emby_refresh(self, emby_id): + if emby_id: + url = f"{self.emby_url}/emby/Items/{emby_id}/Refresh" + headers = {"X-Emby-Token": self.emby_apikey} + querystring = { + "Recursive": "true", + "MetadataRefreshMode": "FullRefresh", + "ImageRefreshMode": "FullRefresh", + "ReplaceAllMetadata": "false", + "ReplaceAllImages": "false", + } + response = requests.request( + "POST", url, headers=headers, params=querystring + ) + if response.text == "": + print(f"🎞 刷新Emby媒体库：成功✅") + return True + else: + print(f"🎞 刷新Emby媒体库：{response.text}❌") + return False + + def emby_search(self, media_name): + if media_name: + url = f"{self.emby_url}/emby/Items" + headers = {"X-Emby-Token": self.emby_apikey} + querystring = { + "IncludeItemTypes": "Series", + "StartIndex": 0, + "SortBy": "SortName", + "SortOrder": "Ascending", + "ImageTypeLimit": 0, + "Recursive": "true", + "SearchTerm": media_name, + "Limit": 10, + "IncludeSearchTypes": "false", + } + response = requests.request("GET", url, headers=headers, params=querystring) + if "application/json" in response.headers["Content-Type"]: + response = response.json() + if response.get("Items"): + for item in response["Items"]: + if item["IsFolder"]: + print( + f"🎞 《{item['Name']}》匹配到Emby媒体库ID：{item['Id']}" + ) + return item["Id"] + else: + print(f"🎞 搜索Emby媒体库：{response.text}❌") + return False + + +def save_check(account): + global check_data + if ( + config_data.get("SAVE_CHECK") == False + or os.environ.get("SAVE_CHECK") == "false" + ): + return + if not check_data: + check_data = requests.get( + "https://mirror.ghproxy.com/https://gist.githubusercontent.com/Cp0204/4764fd0110d5f5bd875eb9a9ff77ccd0/raw/quark_save_check.json" + ).json() + if check_data.get("pwd_id"): + return account.do_save_check( + check_data["pwd_id"], check_data["savepath"] + ) + + +def verify_account(account): + # 验证账号 + account_info = account.init() + print(f"▶️ 验证第{account.index}个账号") + if not account_info: + add_notify(f"👤 第{account.index}个账号登录失败，cookie无效❌") + return False + else: + print(f"👤 账号昵称: {account_info['nickname']}✅") + return True + + +def do_sign(account): + if not verify_account(account): + return + # 每日领空间 + growth_info = account.get_growth_info() + if growth_info: + if growth_info["cap_sign"]["sign_daily"]: + print( + f"📅 执行签到: 今日已签到+{int(growth_info['cap_sign']['sign_daily_reward']/1024/1024)}MB，连签进度({growth_info['cap_sign']['sign_progress']}/{growth_info['cap_sign']['sign_target']})✅" + ) + else: + sign, sign_return = account.get_growth_sign() + save_check_flag = "💹" if save_check(account) else "✅" + if sign: + message = f"📅 执行签到: 今日签到+{int(sign_return/1024/1024)}MB，连签进度({growth_info['cap_sign']['sign_progress']+1}/{growth_info['cap_sign']['sign_target']}){save_check_flag}" + if ( + config_data.get("push_config").get("QUARK_SIGN_NOTIFY") == False + or os.environ.get("QUARK_SIGN_NOTIFY") == "false" + ): + print(message) + else: + message = message.replace("今日", f"[{account.nickname}]今日") + add_notify(message) + else: + print(f"📅 执行签到: {sign_return}") print(f"") - return first_account -def do_save(): - print(f"===============转存任务===============") - print(f"转存账号: {first_account['nickname']}") +def do_save(account): + print(f"转存账号: {account.nickname}") # 任务列表 tasklist = config_data.get("tasklist", []) # 获取全部保存目录fid - update_savepath_fid(tasklist) + account.update_savepath_fid(tasklist) + emby = Emby( + config_data.get("emby").get("url"), config_data.get("emby").get("apikey") + ) def check_date(task): return ( @@ -604,21 +723,22 @@ def do_save(): if task.get("ignore_extension"): print(f"忽略后缀: {task['ignore_extension']}") print() - is_new = do_save_task(task) - is_rename = do_rename_task(task) - if (is_new or is_rename) and task.get("emby_id") != "0": + is_new = account.do_save_task(task) + is_rename = account.do_rename_task(task) + # 刷新媒体库 + if emby and (is_new or is_rename) and task.get("emby_id") != "0": if task.get("emby_id"): - emby_refresh(task["emby_id"]) + emby.emby_refresh(task["emby_id"]) else: - match_emby_id = emby_search(task["taskname"]) + match_emby_id = emby.emby_search(task["taskname"]) if match_emby_id: task["emby_id"] = match_emby_id - emby_refresh(match_emby_id) + emby.emby_refresh(match_emby_id) print(f"") def main(): - global config_data, first_account + global config_data start_time = datetime.now() print(f"===============程序开始===============") print(f"⏰ 执行时间: {start_time.strftime('%Y-%m-%d %H:%M:%S')}") @@ -654,11 +774,17 @@ def main(): if not cookies: print("❌ cookie 未配置") return + accounts = [Quark(cookie, index) for index, cookie in enumerate(cookies)] # 签到 - first_account = do_sign(cookies) + print(f"===============签到任务===============") + for account in accounts: + do_sign(account) + print(f"") # 转存 - if first_account and cookie_form_file: - do_save() + if accounts[0].is_active and cookie_form_file: + print(f"===============转存任务===============") + do_save(accounts[0]) + print(f"") # 通知 if notifys: notify_body = "\n".join(notifys)