浏览代码

fix: The presence of ".pdf" multiple times in the pdf_path results in model_path not matching the expected.

赵小蒙 1 年之前
父节点
当前提交
8e149f381d
共有 1 个文件被更改,包括 5 次插入1 次删除
  1. 5 1
      magic_pdf/cli/magicpdf.py

+ 5 - 1
magic_pdf/cli/magicpdf.py

@@ -290,7 +290,11 @@ def pdf_command(pdf, model, method, inside_model):
     def get_model_json(model_path):
         # 这里处理pdf和模型相关的逻辑
         if model_path is None:
-            model_path = pdf.replace(".pdf", ".json")
+            file_name_without_extension, extension = os.path.splitext(pdf)
+            if extension == ".pdf":
+                model_path = file_name_without_extension + ".json"
+            else:
+                raise Exception("pdf_path input error")
             if not os.path.exists(model_path):
                 logger.warning(
                     f"not found json {model_path} existed"