Procházet zdrojové kódy

fix s3_file_path get logic

赵小蒙 před 1 rokem
rodič
revize
29d2be623b
1 změnil soubory, kde provedl 3 přidání a 1 odebrání
  1. 3 1
      magic_pdf/cli/magicpdf.py

+ 3 - 1
magic_pdf/cli/magicpdf.py

@@ -135,7 +135,9 @@ def json_command(json, method):
         )
 
     jso = json_parse.loads(read_s3_path(json).decode("utf-8"))
-    s3_file_path = jso["file_location"]
+    s3_file_path = jso.get("file_location")
+    if s3_file_path is None:
+        s3_file_path = jso.get("path")
     pdf_file_name = Path(s3_file_path).stem
     pdf_data = read_s3_path(s3_file_path)
     local_image_dir, local_md_dir = prepare_env(pdf_file_name, method)