Browse Source

将imagedir修改为相对路径,以便在markdown中渲染

赵小蒙 1 năm trước cách đây
mục cha
commit
2edf2f8aa5
1 tập tin đã thay đổi với 4 bổ sung4 xóa
  1. 4 4
      magic_pdf/cli/magicpdf.py

+ 4 - 4
magic_pdf/cli/magicpdf.py

@@ -46,11 +46,11 @@ parse_pdf_methods = click.Choice(["ocr", "txt", "auto"])
 
 def prepare_env(pdf_file_name):
     local_parent_dir = os.path.join(
-        get_local_dir(), "magic-pdf",pdf_file_name
+        get_local_dir(), "magic-pdf", pdf_file_name
     )
 
     local_image_dir = os.path.join(local_parent_dir, "images")
-    local_md_dir = os.path.join(local_parent_dir, "md")
+    local_md_dir = local_parent_dir
     os.makedirs(local_image_dir, exist_ok=True)
     os.makedirs(local_md_dir, exist_ok=True)
     return local_image_dir, local_md_dir
@@ -143,7 +143,7 @@ def json_command(json, method):
         method,
         local_image_rw,
         local_md_rw,
-        local_image_dir,
+        os.path.basename(local_image_dir),
     )
 
 
@@ -184,7 +184,7 @@ def pdf_command(pdf, model, method):
         method,
         local_image_rw,
         local_md_rw,
-        local_image_dir,
+        os.path.basename(local_image_dir),
     )