150 lines
5.1 KiB
Python
150 lines
5.1 KiB
Python
|
import requests
|
|||
|
|
from bs4 import BeautifulSoup
|
|||
|
|
import re
|
|||
|
|
import json
|
|||
|
|
import os
|
|||
|
|
import hmac
|
|||
|
|
from hashlib import sha1
|
|||
|
|
|
|||
|
|
|
|||
|
|
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"])]
|
|||
|
|
|
|||
|
|
# 按版本划分链接
|
|||
|
|
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)]
|
|||
|
|
|
|||
|
|
# 设置全局变量
|
|||
|
|
download_file = True
|
|||
|
|
file_updated = False
|
|||
|
|
# 创建版本文件夹并保存链接
|
|||
|
|
def save_links_to_folder(links, folder_name):
|
|||
|
|
folder_path = os.path.join(os.getcwd(), "file", 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)
|
|||
|
|
|
|||
|
|
# 检查文件是否需要更新
|
|||
|
|
global download_file
|
|||
|
|
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:
|
|||
|
|
download_file = False
|
|||
|
|
print(f"文件 {file_name} 已更新。无需下载。")
|
|||
|
|
else:
|
|||
|
|
download_file = True
|
|||
|
|
os.remove(file_path)
|
|||
|
|
|
|||
|
|
# 如果需要下载文件
|
|||
|
|
if download_file:
|
|||
|
|
with open(file_path, "wb") as file:
|
|||
|
|
response = requests.get(link)
|
|||
|
|
file.write(response.content)
|
|||
|
|
print(f"文件 {file_name} 已下载到 {folder_name}")
|
|||
|
|
|
|||
|
|
# 保存链接到对应版本文件夹中
|
|||
|
|
save_links_to_folder(ver3_links, "Ver3")
|
|||
|
|
save_links_to_folder(ver5_links, "Ver5")
|
|||
|
|
|
|||
|
|
# 将链接存储为JSON文件
|
|||
|
|
data = {"Ver3": ver3_links, "Ver5": ver5_links}
|
|||
|
|
|
|||
|
|
with open("file_links.json", "w") as json_file:
|
|||
|
|
json.dump(data, json_file, indent=4)
|
|||
|
|
|
|||
|
|
print("链接已保存至file_links.json")
|
|||
|
|
|
|||
|
|
def downloadFile():
|
|||
|
|
global file_updated
|
|||
|
|
if download_file:
|
|||
|
|
print("download_file的值为:", download_file)
|
|||
|
|
file_updated = True
|
|||
|
|
else:
|
|||
|
|
print("download_file的值为:", download_file)
|
|||
|
|
file_updated = False
|
|||
|
|
downloadFile()
|
|||
|
|
print("file_updated的最终值为:", file_updated)
|
|||
|
|
|
|||
|
|
# 生成本地文件下载链接
|
|||
|
|
base_download_url = "https://file-teamspeak-download.lolicon.team/file/"
|
|||
|
|
|
|||
|
|
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")
|
|||
|
|
|
|||
|
|
# 将本地文件下载链接存储为JSON文件
|
|||
|
|
local_download_links_data = {"Ver3": ver3_local_download_links, "Ver5": ver5_local_download_links}
|
|||
|
|
|
|||
|
|
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")
|
|||
|
|
|
|||
|
|
# 判断是否有文件更新
|
|||
|
|
if file_updated:
|
|||
|
|
print("文件已更新，开始预热CDN...")
|
|||
|
|
# 预热 CDN
|
|||
|
|
def dogecloud_api(api_path, data={}, json_mode=False):
|
|||
|
|
print('正在调用多吉云 API...')
|
|||
|
|
access_key = '' # 请将此处替换为你的多吉云 Access Key
|
|||
|
|
secret_key = '' # 请将此处替换为你的多吉云 Secret Key
|
|||
|
|
# 如果没有 access_key 和 secret_key，则跳过下面所有操作
|
|||
|
|
if not access_key or not secret_key:
|
|||
|
|
print('请填写多吉云 Access Key 和 Secret Key。')
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
body = ''
|
|||
|
|
mime = ''
|
|||
|
|
if json_mode:
|
|||
|
|
body = json.dumps(data)
|
|||
|
|
mime = 'application/json'
|
|||
|
|
else:
|
|||
|
|
body = urllib.parse.urlencode(data)
|
|||
|
|
mime = '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()
|
|||
|
|
|
|||
|
|
api_path = '/cdn/refresh/add.json'
|
|||
|
|
data = {
|
|||
|
|
'rtype': 'prefetch',
|
|||
|
|
'urls': json.dumps(all_local_links)
|
|||
|
|
}
|
|||
|
|
response = dogecloud_api(api_path, data, json_mode=True)
|
|||
|
|
print(response)
|
|||
|
|
data = {
|
|||
|
|
'rtype': 'path',
|
|||
|
|
'urls': 'https://file-teamspeak-download.lolicon.team/file/'
|
|||
|
|
}
|
|||
|
|
response = dogecloud_api(api_path, data, json_mode=True)
|
|||
|
|
print(response)
|
|||
|
|
else:
|
|||
|
|
print("文件未更新，无需预热CDN。")
|