规范代码

precaching.py

@@ -4,40 +4,49 @@ 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)
+
+def precache(text: str):
+    """预缓存单个文本的音频"""
+    cache_dir = Path("./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):
+
+def proc_file(path: 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"]}")
+        print(f"预缓存 [{nu.name}] ({c}/{len(nu)}): {i['content']}")
+        precache(i['content'])
 
-def walk(path_str):
+
+def walk(path_str: 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)
+        if item.is_file() and 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")
+
+if __name__ == "__main__":
+    print("音频预缓存实用程序")
+    print("A: 全部缓存")
+    print("C: 清空缓存")
+    
+    choice = input("输入选项 $ ").upper()
+    
+    if choice == "A":
+        walk("./nucleon")
+    elif choice == "C":
+        shutil.rmtree("./cache/voice", ignore_errors=True)
+        print("缓存已清空")