更新3.0

extend/util/utils.py

@@ -0,0 +1,161 @@
+"""
+通用的工具函数，主要是一些静态函数。
+
+:author: assassing
+:contact: https://github.com/hxz393
+:copyright: Copyright 2024, hxz393. 保留所有权利。
+"""
+import atexit
+import base64
+import os
+import random
+import re
+import string
+import tempfile
+import threading
+import zlib
+from typing import Any, Union, Tuple, Callable, List, Optional
+
+from src.constants import CONFIG_PATH, ICON_BASE64
+
+# 预编译正则表达式
+SHARE_ID_REGEX = re.compile(r'"shareid":(\d+?),"')
+USER_ID_REGEX = re.compile(r'"share_uk":"(\d+?)","')
+FS_ID_REGEX = re.compile(r'"fs_id":(\d+?),"')
+SERVER_FILENAME_REGEX = re.compile(r'"server_filename":"(.+?)","')
+ISDIR_REGEX = re.compile(r'"isdir":(\d+?),"')
+
+
+def thread_it(func: Callable, *args: Tuple[Any, ...]) -> None:
+    """
+    多线程防止转存时主界面卡死。
+
+    :param func: 要调用的函数
+    :param args: 函数参数
+    :return: 无返回值
+    """
+    t = threading.Thread(target=func, args=args)
+    t.start()
+
+
+def write_config(config: str) -> None:
+    """
+    写入配置文件，点击批量转存或分享按钮时才运行。
+
+    :param config: 配置文件内容，以换行转义字符 '\n' 拼接多个配置到一行字符串
+    :return: 无返回值
+    """
+    with open(CONFIG_PATH, 'w') as f:
+        f.write(config)
+
+
+def read_config() -> Optional[List[str]]:
+    """
+    读取配置文件，在 UI 初始化完毕后执行。
+
+    :return: 读取成功时返回配置列表，一个元素一个配置。配置文件不存在时返回 None，什么也不会发生
+    """
+    if os.path.exists(CONFIG_PATH):
+        with open(CONFIG_PATH) as f:
+            config = f.read().splitlines()
+            return config
+
+
+def create_icon() -> str:
+    """
+    从 base64 编码中生成临时图标，在程序结束时自动删除。
+    好处是打包时不用导入资源文件。
+
+    :return: 返回图标文件路径
+    """
+    with tempfile.NamedTemporaryFile(delete=False, suffix='.ico') as temp_file:
+        temp_file.write(zlib.decompress(base64.b64decode(ICON_BASE64)))
+
+    ico_path = temp_file.name
+    # 显式声明程序退出时，删除临时图标文件，避免在个别系统平台自动删除失败
+    atexit.register(os.remove, ico_path)
+    return ico_path
+
+
+def normalize_link(url_code: str) -> str:
+    """
+    预处理链接至标准格式。
+
+    :param url_code: 需要处理的的原始链接格式
+    :return: 返回标准格式：链接+空格+提取码
+    """
+    # 升级旧链接格式
+    normalized = url_code.replace("share/init?surl=", "s/1")
+    # 替换掉 ?pwd= 或 &pwd= 为空格
+    normalized = re.sub(r'[?&]pwd=', ' ', normalized)
+    # 替换掉提取码字样为空格
+    normalized = re.sub(r'提取码*[：:]', ' ', normalized)
+    # 替换 http 为 https，顺便处理掉开头没用的文字
+    normalized = re.sub(r'^.*?(https?://)', 'https://', normalized)
+    # 替换连续的空格
+    normalized = re.sub(r'\s+', ' ', normalized)
+    return normalized
+
+
+def parse_url_and_code(url_code: str) -> Tuple[str, str]:
+    """
+    以空格分割出 URL 和提取码。
+
+    :param url_code: 输入的标准链接格式
+    :return: 链接和提取码
+    """
+    # 不会分割失败
+    url, code = map(str.strip, url_code.split(' ', 1))
+    # 暴力切片，如果输入链接不是以提取码结尾，会得到错误提取码
+    return url[:47], code[-4:]
+
+
+def parse_response(response: str) -> Union[List[str], int]:
+    """
+    验证提取码通过后，再次访问网盘地址，此函数解析返回的页面源码并提取所需要参数。
+    shareid_list 和 user_id_list 只有一个值，fs_id_list 需要完整返回
+
+    :param response: 响应内容
+    :return: 没有获取到足够参数时，返回错误代码 -1；否则返回三个参数的列表
+    """
+    shareid_list = SHARE_ID_REGEX.findall(response)
+    user_id_list = USER_ID_REGEX.findall(response)
+    fs_id_list = FS_ID_REGEX.findall(response)
+    server_filename_list = SERVER_FILENAME_REGEX.findall(response)
+    isdir_list = ISDIR_REGEX.findall(response)
+    if not all([shareid_list, user_id_list, fs_id_list, server_filename_list, isdir_list]):
+        return -1
+
+    return [shareid_list[0], user_id_list[0], fs_id_list, list(dict.fromkeys(server_filename_list)), isdir_list]
+
+
+def update_cookie(bdclnd: str, cookie: str) -> str:
+    """
+    更新 cookie 字符串，以包含新的 BDCLND 值。
+
+    :param bdclnd: 新的 BDCLND 值
+    :param cookie: 当前的 cookie 字符串
+    :return: 返回新 cookie 字符串
+    """
+    # 拆分 cookie 字符串到字典。先用 ; 分割成列表，再用 = 分割出键和值
+    cookies_dict = dict(map(lambda item: item.split('=', 1), filter(None, cookie.split(';'))))
+    # 在 cookie 字典中，更新或添加 BDCLND 的值
+    cookies_dict['BDCLND'] = bdclnd
+    # 从更新后的字典重新构建 cookie 字符串
+    updated_cookie = ';'.join([f'{key}={value}' for key, value in cookies_dict.items()])
+
+    return updated_cookie
+
+
+def generate_code() -> str:
+    """
+    生成一个四位的随机提取码，包含大小写字母和数字。
+
+    :return: 随机提取码
+    """
+    # 包含大小写字母和数字
+    characters = string.ascii_letters + string.digits
+    # 随机选择四个字符
+    code = ''.join(random.choice(characters) for _ in range(4))
+
+    return code