forked from Lu-mi-nous/Voice-from-prts-to-moegirl
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathvoice.py
121 lines (119 loc) · 7.18 KB
/
voice.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
from pypinyin import lazy_pinyin
from pydub import AudioSegment
import os
import re
import requests
from time import sleep
from tqdm import tqdm
import random
import urllib.parse
op=input("输入干员名称以获取该干员全部语音文件：")
html=requests.get('https://m.prts.wiki/api.php', params={'action': 'query',
'prop': 'revisions',
'rvprop': 'content',
'titles': f'{op}|{op}/语音记录|后勤技能一览',
'format': 'json',
'formatversion': 'latest'
}).json()
for _ in html['query']['pages']:
if _['title'] == op+'/语音记录':
res = _['revisions'][0]['content']
header=["Mozilla/5.0 (Windows NT 6.3; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/39.0.2171.95 Safari/537.36",
"Mozilla/5.0 (Macintosh; Intel Mac OS X 10_9_2) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/35.0.1916.153 Safari/537.36",
"Mozilla/5.0 (Windows NT 6.1; WOW64; rv:30.0) Gecko/20100101 Firefox/30.0",
"Mozilla/5.0 (Macintosh; Intel Mac OS X 10_9_2) AppleWebKit/537.75.14 (KHTML, like Gecko) Version/7.0.3 Safari/537.75.14",
"Mozilla/5.0 (compatible; MSIE 10.0; Windows NT 6.2; Win64; x64; Trident/6.0)",
'Mozilla/5.0 (Windows; U; Windows NT 5.1; it; rv:1.8.1.11) Gecko/20071127 Firefox/2.0.0.11',
'Opera/9.25 (Windows NT 5.1; U; en)',
'Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1; SV1; .NET CLR 1.1.4322; .NET CLR 2.0.50727)',
'Mozilla/5.0 (compatible; Konqueror/3.5; Linux) KHTML/3.5.5 (like Gecko) (Kubuntu)',
'Mozilla/5.0 (X11; U; Linux i686; en-US; rv:1.8.0.12) Gecko/20070731 Ubuntu/dapper-security Firefox/1.5.0.12',
'Lynx/2.8.5rel.1 libwww-FM/2.14 SSL-MM/1.4.1 GNUTLS/1.2.9',
"Mozilla/5.0 (X11; Linux i686) AppleWebKit/535.7 (KHTML, like Gecko) Ubuntu/11.04 Chromium/16.0.912.77 Chrome/16.0.912.77 Safari/535.7",
"Mozilla/5.0 (X11; Ubuntu; Linux i686; rv:10.0) Gecko/20100101 Firefox/10.0 "]
res=re.search("语音key=(.*?)\\n",str(res)).group(1)
res1="N" # 跳过日文语音？(Y/N[default])
res3="N" # 是否方言？(Y/N[default])
res4="Y" # 是否获取韩文语音？(Y[default]/N)
res5="Y" # 是否获取英文语音？(Y[default]/N)
print("即将获取 "+op+" 的中文语音")
if res1 == "N" or res1 == "n":
print("即将获取 "+op+" 的日文语音")
if res3 == "Y" or res3 == "y":
print("即将获取 "+op+" 的方言语音")
if res4 == "Y" or res4 == "y":
print("即将获取 "+op+" 的韩文语音")
if res5 == "Y" or res5 == "y":
print("即将获取 "+op+" 的英文语音")
if res3!="Y" and res3!="y":
res2=input("特殊语音key：")
if res2 != "":
print("即将获取 "+op+" 的 "+res2+" 语音")
else:
res2=res+"_cn_topolect"
id={'任命助理':'001','交谈1':'002','交谈2':'003','交谈3':'004','晋升后交谈1':'005','晋升后交谈2':'006','信赖提升后交谈1':'007','信赖提升后交谈2':'008','信赖提升后交谈3':'009','闲置':'010','干员报到':'011','观看作战记录':'012','精英化晋升1':'013','精英化晋升2':'014','编入队伍':'017','任命队长':'018','行动出发':'019','行动开始':'020','选中干员1':'021','选中干员2':'022','部署1':'023','部署2':'024','作战中1':'025','作战中2':'026','作战中3':'027','作战中4':'028','完成高难行动':'029','3星结束行动':'030','非3星结束行动':'031','行动失败':'032','进驻设施':'033','戳一下':'034','信赖触摸':'036','标题':'037','问候':'042'}
workdir=os.getcwd()
operator=re.search('char_(\d+)_(\w+)',res).group(2) if re.search('char_(\d+)_(\w+)',res) else ""
num=re.search('char_(\d+)_(\w+)',res).group(1)
#zh=urllib.parse.unquote(re.search("(%.*)_+",url).group(1)) if re.search("([\u4e00-\u9fa5]+)",urllib.parse.unquote(re.search("(%.*)_+",url).group(1))).group(1)==urllib.parse.unquote(re.search("(%.*)_+",url).group(1)) else None
os.mkdir(workdir+"\\"+op+"\\") if not os.path.exists(workdir+"\\"+op+"\\") else print(op+"语音目录已存在！")
os.mkdir(workdir+"\\temp\\") if not os.path.exists(workdir+"\\temp\\") else print("缓存目录已存在")
py=lazy_pinyin(op)
pinyin=[]
for a in py:
pinyin.append(a.title())
pinyin="".join(pinyin)
pinyin_=input("确认干员名称拼音（与预期不一致时输入拼音）：" + pinyin + " ")
if pinyin_:
pinyin=pinyin_
langdict={"中文":"_cn", "英文":"_en", "韩文":"_kr", "日文":""}
def download(lang:str,kind="",res=res):
zh="_zh" if lang=="中文" else langdict[lang]
lan=langdict[lang]
custom="_custom" if lang=="方言" else ""
if kind and lang!="方言":
zh="_skin1"+zh
elif kind and lang=="方言":
zh="_fy"+zh
os.mkdir(workdir+"\\"+op+"\\") if not os.path.exists(workdir+"\\"+op+"\\") else print(op+"语音目录已存在！")
pbar=tqdm(id,unit="file")
for key in pbar:
filename=pinyin+zh+"_CN_"+id[key]+".wav"
if os.path.exists(workdir + "\\" + op + "\\" + filename):
if os.stat(workdir + "\\" + op + "\\" + filename).st_size <= 1024:
# os.remove(workdir+"\\"+op+"\\"+filename)
print("\n"+filename+"可能为损坏文件，如该干员无本条语音请忽略")
# exit(1)
if not os.path.exists(workdir+"\\"+op+"\\"+filename):
pbar.set_description("正在获取"+filename+"...")
response=requests.get("https://static.prts.wiki/voice"+lan+custom+"/"+res+"/"+"CN_"+id[key]+".wav",headers={'User-Agent': random.choice(header), 'Referer':"https://prts.wiki/w/"+urllib.parse.quote(op)+r"/%E8%AF%AD%E9%9F%B3%E8%AE%B0%E5%BD%95"})
if response.content.__sizeof__() <= 1024:
# os.remove(workdir+"\\"+op+"\\"+filename)
print("\n获取"+filename+"时可能遇到了网络错误，如该干员无本条语音请忽略")
else:
with open(workdir + "\\" + op + "\\" + filename, "wb") as f:
f.write(response.content)
# try:
# mp3=AudioSegment.from_wav(workdir+"\\temp\\"+filename) # 如果报错请检查44行是否包含该干员没有的语言
# mp3.export(workdir+"\\"+op+"\\"+filename.rstrip("wav")+"mp3",format="mp3",bitrate="320k")
# except(FileNotFoundError):
# print("未找到文件" + res+"/"+"CN_"+id[key]+".wav")
# os.remove(workdir+"\\temp\\"+filename)
#pbar.set_description(filename.rstrip("wav")+"wav已获取完成！")
sleep(0.2) # 去除这个sleep可能会导致经常性网络问题
else:
pbar.set_description(filename+"已存在!")
sleep(0.05)
if res1!="Y" and res1!="y":
download("日文")
download("中文")
if res2 and res3!="Y" and res3!="y":
download("日文",res=res2)
download("中文",res=res2)
if res3=="Y" or res3=="y":
download("方言",kind="特殊",res=res2)
if res4=="Y" or res4=="y":
download("韩文")
if res5=="Y" or res5=="y":
download("英文")
input("\n\n按下回车关闭窗口")