Bladeren bron

pdf_converter_v2: zip 文件名改为时间+taskID(年月日_时分秒_taskID前8位)

何文松 3 weken geleden
bovenliggende
commit
b30b464fdb
1 gewijzigde bestanden met toevoegingen van 5 en 3 verwijderingen
  1. 5 3
      pdf_converter_v2/api/main.py

+ 5 - 3
pdf_converter_v2/api/main.py

@@ -10,6 +10,7 @@ import shutil
 import tempfile
 import uuid
 import base64
+from datetime import datetime
 import json
 import zipfile
 from pathlib import Path
@@ -626,12 +627,13 @@ async def process_pdf_to_markdown_task(
         task_status[task_id]["document_type"] = None
 
         if return_images:
-            zip_path = os.path.join(output_dir, "markdown_with_images.zip")
+            zip_basename = f"{datetime.now().strftime('%Y%m%d_%H%M%S')}_{task_id[:8]}.zip"
+            zip_path = os.path.join(output_dir, zip_basename)
             try:
                 with zipfile.ZipFile(zip_path, "w", zipfile.ZIP_DEFLATED) as zf:
                     for root, _, files in os.walk(output_dir):
                         for f in files:
-                            if f == "markdown_with_images.zip":
+                            if f == zip_basename:
                                 continue
                             abs_path = os.path.join(root, f)
                             arcname = os.path.relpath(abs_path, output_dir)
@@ -1097,7 +1099,7 @@ async def download_zip(task_id: str):
     return FileResponse(
         zip_file,
         media_type="application/zip",
-        filename="markdown_with_images.zip",
+        filename=os.path.basename(zip_file),
     )