156 lines
5.5 KiB
Python
156 lines
5.5 KiB
Python
|
import requests
|
|||
|
|
from bs4 import BeautifulSoup
|
|||
|
|
import re
|
|||
|
|
import json
|
|||
|
|
import os
|
|||
|
|
import hmac
|
|||
|
|
from hashlib import sha1
|
|||
|
|
|
|||
|
|
# 配置选项
|
|||
|
|
ENABLE_CDN_REFRESH = True # 设置为False禁用CDN预热功能
|
|||
|
|
|
|||
|
|
print("Start downloading file links...")
|
|||
|
|
|
|||
|
|
# 发送GET请求获取页面内容
|
|||
|
|
url = "https://teamspeak.com/zh-CN/downloads/"
|
|||
|
|
response = requests.get(url)
|
|||
|
|
soup = BeautifulSoup(response.content, "html.parser")
|
|||
|
|
|
|||
|
|
# 找到所有<a>标签并提取链接
|
|||
|
|
links = soup.find_all("a", href=True)
|
|||
|
|
|
|||
|
|
# 匹配包含"https://files.teamspeak-services.com"的链接
|
|||
|
|
file_links = [link["href"] for link in links if re.search(r"https://files.teamspeak-services.com", link["href"])]
|
|||
|
|
|
|||
|
|
# 按版本划分链接（新增Ver6支持）
|
|||
|
|
ver3_links = [link for link in file_links if re.search(r"\/3\.\d+\.\d+\/", link)]
|
|||
|
|
ver5_links = [link for link in file_links if re.search(r"\/(?:5\.\d+\.\d+|5\.\d+\.\d+\-\w+)/", link)]
|
|||
|
|
ver6_links = [link for link in file_links if re.search(r"\/6\.\d+\.\d+.*\/", link)] # 新增Ver6匹配模式
|
|||
|
|
|
|||
|
|
# 设置全局变量
|
|||
|
|
download_file = True
|
|||
|
|
file_updated = False
|
|||
|
|
|
|||
|
|
# 创建版本文件夹并保存链接
|
|||
|
|
def save_links_to_folder(links, folder_name):
|
|||
|
|
global download_file, file_updated
|
|||
|
|
folder_path = os.path.join(os.getcwd(), "tsfile", folder_name)
|
|||
|
|
if not os.path.exists(folder_path):
|
|||
|
|
os.makedirs(folder_path)
|
|||
|
|
|
|||
|
|
for link in links:
|
|||
|
|
file_name = link.split("/")[-1]
|
|||
|
|
file_path = os.path.join(folder_path, file_name)
|
|||
|
|
|
|||
|
|
# 检查文件是否需要更新
|
|||
|
|
if os.path.exists(file_path):
|
|||
|
|
response = requests.head(link)
|
|||
|
|
remote_file_size = int(response.headers.get('Content-Length', 0))
|
|||
|
|
local_file_size = os.path.getsize(file_path)
|
|||
|
|
if remote_file_size == local_file_size:
|
|||
|
|
print(f"文件 {file_name} 无需更新。")
|
|||
|
|
continue
|
|||
|
|
else:
|
|||
|
|
os.remove(file_path)
|
|||
|
|
print(f"发现新版本 {file_name}，开始下载...")
|
|||
|
|
file_updated = True
|
|||
|
|
|
|||
|
|
# 下载文件
|
|||
|
|
with open(file_path, "wb") as file:
|
|||
|
|
response = requests.get(link)
|
|||
|
|
file.write(response.content)
|
|||
|
|
print(f"文件 {file_name} 已下载到 {folder_name}")
|
|||
|
|
file_updated = True
|
|||
|
|
|
|||
|
|
# 保存链接到对应版本文件夹中（新增Ver6处理）
|
|||
|
|
save_links_to_folder(ver3_links, "Ver3")
|
|||
|
|
save_links_to_folder(ver5_links, "Ver5")
|
|||
|
|
save_links_to_folder(ver6_links, "Ver6")
|
|||
|
|
|
|||
|
|
# 将链接存储为JSON文件（新增Ver6数据）
|
|||
|
|
data = {
|
|||
|
|
"Ver3": ver3_links,
|
|||
|
|
"Ver5": ver5_links,
|
|||
|
|
"Ver6": ver6_links # 新增Ver6数据
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
with open("file_links.json", "w") as json_file:
|
|||
|
|
json.dump(data, json_file, indent=4)
|
|||
|
|
|
|||
|
|
print("链接已保存至file_links.json")
|
|||
|
|
|
|||
|
|
# 生成本地文件下载链接
|
|||
|
|
base_download_url = "https://file-us.ovofish.com/tsfile/"
|
|||
|
|
|
|||
|
|
def generate_local_download_links(links, folder_name):
|
|||
|
|
local_download_links = [base_download_url + folder_name + "/" + link.split("/")[-1] for link in links]
|
|||
|
|
return local_download_links
|
|||
|
|
|
|||
|
|
ver3_local_download_links = generate_local_download_links(ver3_links, "Ver3")
|
|||
|
|
ver5_local_download_links = generate_local_download_links(ver5_links, "Ver5")
|
|||
|
|
ver6_local_download_links = generate_local_download_links(ver6_links, "Ver6") # 新增Ver6链接生成
|
|||
|
|
|
|||
|
|
# 合并所有下载链接用于CDN预热
|
|||
|
|
all_local_links = ver3_local_download_links + ver5_local_download_links + ver6_local_download_links
|
|||
|
|
|
|||
|
|
# 将本地文件下载链接存储为JSON文件（新增Ver6数据）
|
|||
|
|
local_download_links_data = {
|
|||
|
|
"Ver3": ver3_local_download_links,
|
|||
|
|
"Ver5": ver5_local_download_links,
|
|||
|
|
"Ver6": ver6_local_download_links # 新增Ver6数据
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
with open("local_download_links.json", "w") as local_json_file:
|
|||
|
|
json.dump(local_download_links_data, local_json_file, indent=4)
|
|||
|
|
|
|||
|
|
print("本地下载链接已保存至local_download_links.json")
|
|||
|
|
|
|||
|
|
# CDN刷新功能（新增开关控制）
|
|||
|
|
def dogecloud_api(api_path, data={}, json_mode=False):
|
|||
|
|
"""多吉云API接口"""
|
|||
|
|
print('正在调用多吉云API...')
|
|||
|
|
access_key = '' # 请替换为你的Access Key
|
|||
|
|
secret_key = '' # 请替换为你的Secret Key
|
|||
|
|
|
|||
|
|
if not access_key or not secret_key:
|
|||
|
|
print('请填写多吉云Access Key和Secret Key。')
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
body = json.dumps(data) if json_mode else urllib.parse.urlencode(data)
|
|||
|
|
mime = 'application/json' if json_mode else 'application/x-www-form-urlencoded'
|
|||
|
|
|
|||
|
|
sign_str = api_path + "\n" + body
|
|||
|
|
signed_data = hmac.new(secret_key.encode('utf-8'), sign_str.encode('utf-8'), sha1)
|
|||
|
|
sign = signed_data.digest().hex()
|
|||
|
|
authorization = 'TOKEN ' + access_key + ':' + sign
|
|||
|
|
|
|||
|
|
response = requests.post(
|
|||
|
|
'https://api.dogecloud.com' + api_path,
|
|||
|
|
data=body,
|
|||
|
|
headers={'Authorization': authorization, 'Content-Type': mime}
|
|||
|
|
)
|
|||
|
|
return response.json()
|
|||
|
|
|
|||
|
|
# 根据配置决定是否执行CDN预热
|
|||
|
|
if file_updated and ENABLE_CDN_REFRESH:
|
|||
|
|
print("文件已更新，开始预热CDN...")
|
|||
|
|
|
|||
|
|
# 预热具体文件
|
|||
|
|
refresh_data = {
|
|||
|
|
'rtype': 'prefetch',
|
|||
|
|
'urls': json.dumps(all_local_links)
|
|||
|
|
}
|
|||
|
|
response = dogecloud_api('/cdn/refresh/add.json', refresh_data, json_mode=True)
|
|||
|
|
print("文件预热响应：", response)
|
|||
|
|
|
|||
|
|
# 刷新目录
|
|||
|
|
path_refresh_data = {
|
|||
|
|
'rtype': 'path',
|
|||
|
|
'urls': 'https://file-us.ovofish.com/file/'
|
|||
|
|
}
|
|||
|
|
response = dogecloud_api('/cdn/refresh/add.json', path_refresh_data, json_mode=True)
|
|||
|
|
print("目录刷新响应：", response)
|
|||
|
|
elif file_updated:
|
|||
|
|
print("文件已更新，但CDN刷新功能已禁用。")
|
|||
|
|
else:
|
|||
|
|
print("文件未更新，无需任何操作。")
|