43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
# 音频预缓存实用程序, 独立于主程序之外, 但依赖 particles 组件
|
|
import particles as pt
|
|
import edge_tts as tts
|
|
from pathlib import Path
|
|
import shutil
|
|
|
|
def precache(text):
|
|
cache_dir = Path(f"./cache/voice/")
|
|
cache_dir.mkdir(parents = True, exist_ok = True)
|
|
cache = cache_dir / f"{text}.wav"
|
|
if not cache.exists():
|
|
communicate = tts.Communicate(text, "zh-CN-YunjianNeural")
|
|
communicate.save_sync(f"./cache/voice/{text}.wav")
|
|
|
|
def proc_file(path):
|
|
nu = pt.NucleonUnion(path)
|
|
c = 0
|
|
for i in nu.nucleons:
|
|
c += 1
|
|
print(f"预缓存 [{nu.name}] ({c}/{len(nu)}): {i["content"]}")
|
|
precache(f"{i["content"]}")
|
|
|
|
def walk(path_str):
|
|
path = Path(path_str)
|
|
|
|
print(f"正在遍历目录: {path}")
|
|
for item in path.iterdir():
|
|
if item.is_file():
|
|
if item.suffix == ".toml":
|
|
print(f"正预缓存文件: {item.name}")
|
|
proc_file(item)
|
|
elif item.is_dir():
|
|
print(f"进入目录: {item.name}")
|
|
|
|
print("音频预缓存实用程序")
|
|
print("需要?")
|
|
print("全部缓存: A")
|
|
print("清空缓存: C")
|
|
choice = input("输入选项 $ ")
|
|
if choice == "a" or choice == "A":
|
|
walk("./nucleon")
|
|
if choice == "c" or choice == "C":
|
|
shutil.rmtree("./cache/voice") |