judge if files directory and output_files directory are exist

c-3lab · Feb 17, 2024 · c90e14c · c90e14c
1 parent 3a5e605
commit c90e14c
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/main.py b/main.py
@@ -4,9 +4,12 @@
 import os
 import shutil
 
-# lattice=Trueでテーブルの軸線でセルを判定
-os.mkdir("./files")
-os.mkdir("./output_files")
+if not os.path.exists("./files"):
+    os.mkdir("./files")
+
+if not os.path.exists("./output_files"):
+    os.mkdir("./output_files")
+
 dfs = tabula.read_pdf("000876235.pdf", lattice=True, pages='all', pandas_options={'header': None})
 i = 0
 for df in dfs: