Bladeren bron

Merge pull request #21 from papayalove/master

更新了 io
drunkpig 1 jaar geleden
bovenliggende
commit
5fca0a0a49
1 gewijzigde bestanden met toevoegingen van 5 en 2 verwijderingen
  1. 5 2
      magic_pdf/io/DiskReaderWriter.py

+ 5 - 2
magic_pdf/io/DiskReaderWriter.py

@@ -35,6 +35,9 @@ class DiskReaderWriter(AbsReaderWriter):
             abspath = path
         else:
             abspath = os.path.join(self.path, path)
+        directory_path = os.path.dirname(abspath)
+        if not os.path.exists(directory_path):
+            os.makedirs(directory_path)
         if mode == MODE_TXT:
             with open(abspath, "w", encoding=self.encoding) as f:
                 f.write(content)
@@ -53,11 +56,11 @@ class DiskReaderWriter(AbsReaderWriter):
 
 # 使用示例
 if __name__ == "__main__":
-    file_path = "io/example.txt"
+    file_path = "io/test/example.txt"
     drw = DiskReaderWriter("D:\projects\papayfork\Magic-PDF\magic_pdf")
 
     # 写入内容到文件
-    drw.write(b"Hello, World!", path="io/example.txt", mode="binary")
+    drw.write(b"Hello, World!", path="io/test/example.txt", mode="binary")
 
     # 从文件读取内容
     content = drw.read(path=file_path)