utils.py

import os
import re
import shutil
import requests

EN_US = os.getenv("LANG") != "zh_CN.UTF-8"
API_QR = os.getenv("api_qr")
API_URL = os.getenv("api_url")
API_SMTP = os.getenv("api_smtp")
API_TRANS = os.getenv("api_caiyun")
KEY_TRANS = os.getenv("apikey_caiyun")
if not (API_SMTP and API_TRANS and KEY_TRANS):
    print("请检查环境变量")
    exit()


TMP_DIR = "./__pycache__"
HEADER = {
    "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/132.0.0.0 Safari/537.36 Edg/132.0.0.0",
}


def clean_dir(dir_path: str):
    if os.path.exists(dir_path):
        shutil.rmtree(dir_path)

    os.makedirs(dir_path)


def is_valid_url(url):
    # 定义 URL 的正则表达式
    pattern = re.compile(
        r"^(https?://)?"  # 协议（http 或 https，可选）
        r"([a-zA-Z0-9-]+\.)+[a-zA-Z]{2,}"  # 域名
        r"(:\d+)?"  # 端口号（可选）
        r"(/[^ ]*)?$"  # 路径（可选）
    )
    # 使用正则表达式匹配 URL
    return bool(pattern.match(url))


def download_file(url, local_filename):
    clean_dir(os.path.dirname(local_filename))
    response = requests.get(url, stream=True)
    response.raise_for_status()
    with open(local_filename, "wb") as f:
        for chunk in response.iter_content(chunk_size=8192):
            f.write(chunk)

    return local_filename