Initial commit

2026-01-19 09:51:07 +08:00
commit 9ef82393be
174 changed files with 22285 additions and 0 deletions
@@ -0,0 +1,67 @@
+
+import asyncio
+import sys
+import glob
+from pathlib import Path
+from loguru import logger
+
+# Add project root to path
+sys.path.insert(0, str(Path(__file__).parent))
+
+# Import the logic from the previous test script to reuse it
+# (Assuming it's safe to import, or I'll copy the core logic if cleaner)
+from test_full_flow_random_sample import run_random_sample_test
+
+async def run_batch_test():
+    print("🚀 Starting Batch Test on all EPUBs")
+    print("=" * 60)
+    
+    # Find all EPUBs
+    epub_files = []
+    # Search in current directory
+    epub_files.extend(glob.glob("*.epub"))
+    # Search in 'input' directory if it exists
+    if Path("input").exists():
+        epub_files.extend(glob.glob("input/*.epub"))
+    # Search in subfolders (e.g. "未命名文件夹")
+    epub_files.extend(glob.glob("**/*.epub", recursive=True))
+    
+    # Deduplicate and filter out output files
+    unique_epubs = set()
+    for f in epub_files:
+        path = Path(f)
+        if "output" in path.parts or "_bilingual" in path.name or "test_output" in path.parts:
+            continue
+        unique_epubs.add(str(path))
+        
+    sorted_epubs = sorted(list(unique_epubs))
+    
+    if not sorted_epubs:
+        print("❌ No EPUB files found.")
+        return
+
+    print(f"📚 Found {len(sorted_epubs)} unique EPUBs to test:")
+    for f in sorted_epubs:
+        print(f"  - {f}")
+    print("-" * 60)
+    
+    results = {}
+    
+    for i, epub_file in enumerate(sorted_epubs, 1):
+        print(f"\n[{i}/{len(sorted_epubs)}] Testing: {epub_file}")
+        try:
+            await run_random_sample_test(epub_file)
+            results[epub_file] = "✅ Success"
+        except Exception as e:
+            print(f"❌ Failed: {e}")
+            logger.error(f"Test failed for {epub_file}", exc_info=True)
+            results[epub_file] = f"❌ Failed: {e}"
+            
+    print("\n" + "=" * 60)
+    print("📊 Batch Test Summary")
+    print("=" * 60)
+    for f, status in results.items():
+        print(f"{status} - {f}")
+
+if __name__ == "__main__":
+    asyncio.run(run_batch_test())