原贴地址：https://www.52pojie.cn/thread-1877570-1-1.html
如有不妥之处将及时修改！侵则删！
PixPin_2024-01-07_17-07-10.png (286.76 KB, 下载次数: 0)
下载附件
2024-1-7 17:32 上传
[Python] 纯文本查看 复制代码# **************************************************
# **************************************************
# 免责声明：
# 本代码仅供学习和研究使用，严禁用于任何商业用途。
# 作者不对使用本代码导致的任何后果负责。
# 在使用本代码时，请遵守相关法律法规和网站的使用规定。
# **************************************************
import os
import tkinter as tk
from tkinter import simpledialog
import requests
import re
import json
# ANSI转义码
class Colors:
RED = '\033[91m'
YELLOW = '\033[93m'
GREEN = '\033[92m'
ENDC = '\033[0m'
def clean_filename(filename):
# 移除非法字符
return re.sub(r'[\/:*?"|]', '', filename)
def download_novel(id, page_num, start_page):
current_page_file = f"pageid_{id}.txt"
base_url = "https://www.ysts.cc/tingshu"
session = requests.Session()
for current_page in range(start_page, page_num + 1):
url = "{}/{}/?p={}".format(base_url, id, current_page)
headers = {
'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.7',
'Accept-Language': 'zh-CN,zh;q=0.9,en;q=0.8,en-GB;q=0.7,en-US;q=0.6,ja;q=0.5,zh-TW;q=0.4',
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36 Edg/120.0.0.0',
'Content-Type': 'application/x-www-form-urlencoded; charset=UTF-8',
'Origin': 'http://www.ysts.cc',
'Referer': '{}/{}/41440.html'.format(base_url, id),
}
try:
response = download_file(session, url, headers)
a_rule = fr''
match_a = re.findall(a_rule, response.text, re.S)
if not match_a:
print(Colors.RED + "未获取到数据，请检查输入参数是否正确！" + Colors.ENDC)
return
# 创建整个音频的保存路径
folder_path = os.path.join("Novel audio", f"ID_{id}")
os.makedirs(folder_path, exist_ok=True)
for match in match_a:
mid = match[0]
name = match[1]
name_cleaned = clean_filename(name)
url = "{}/{}/{}.html".format(base_url, id, mid)
response = download_file(session, url, headers)
c_rule = r''
match_b = re.findall(c_rule, response.text, re.S)
if not match_b:
print(Colors.RED + "未获取SC" + Colors.ENDC)
return
sc = match_b[0]
play_url = "https://www.ysts.cc/api/act/play"
headers_play = headers.copy()
headers_play["sc"] = sc
print(f"{name}...")
params_play = {'nid': id, 'cid': mid}
response_play = session.post(play_url, headers=headers_play, data=params_play, timeout=60)
response_play.raise_for_status()
# Check if the response is JSON
if 'application/json' in response_play.headers['content-type']:
try:
result_play = response_play.json()
except json.JSONDecodeError:
print(Colors.RED + "未能正确解析JSON响应" + Colors.ENDC)
return
else:
print(Colors.RED + "未获取正确的播放数据" + Colors.ENDC)
return
novel_title_rule = r'(.*?)'
novel_title_match = re.search(novel_title_rule, response.text)
if novel_title_match:
novel_title = clean_filename(novel_title_match.group(1).strip())
else:
print(Colors.RED + "未获取小说标题" + Colors.ENDC)
return
# 检查文件是否已存在
audio_file_path = os.path.join(folder_path, f"{name_cleaned}.m4a")
if os.path.exists(audio_file_path):
print(Colors.YELLOW + f"此音频文件已存在，跳过下载！" + Colors.ENDC)
else:
# 保存音频文件
with open(audio_file_path, "wb") as file:
file.write(download_file(session, result_play['url'], headers_play, timeout=60).content)
print(Colors.GREEN + "此音频下载完成!\n" + Colors.ENDC)
# 将当前页数写入文件
with open(current_page_file, "w") as file:
file.write(str(current_page))
print(Colors.GREEN + "全部下载完成" + Colors.ENDC)
except KeyboardInterrupt:
print(Colors.YELLOW + "下载已被用户中断！" + Colors.ENDC)
break
except Exception as e:
print(Colors.RED + f'下载失败：{str(e)}' + Colors.ENDC)
break
def download_file(session, url, headers, params=None, method='GET', timeout=30):
response = session.request(method, url, headers=headers, params=params, timeout=timeout)
response.raise_for_status()
return response
def main():
root = tk.Tk()
root.withdraw() # 不显示 tkinter 窗口
try:
album_id = simpledialog.askinteger("输入", "请输入专辑ID:")
page_num = simpledialog.askinteger("输入", "请输入总下载页数:")
start_page = simpledialog.askinteger("输入", "请输入从第几页开始下载:")
download_novel(album_id, page_num, start_page)
except KeyboardInterrupt:
print(Colors.YELLOW + "下载已被用户中断！" + Colors.ENDC)
except Exception as e:
print(Colors.RED + f'下载失败：{str(e)}' + Colors.ENDC)
if __name__ == "__main__":
main()
使用方法
[ol]
[/ol]
主要特点
修复BUG
注意事项
希望这个脚本能够方便你获取喜欢的小说音频，祝你使用愉快！
