瀏覽代碼

fix s3_file_path get logic

赵小蒙 1 年之前
父節點
當前提交
29d2be623b
共有 1 個文件被更改,包括 3 次插入1 次删除
  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)