Update caj2txt.py

caj2txt.py

@@ -3,18 +3,24 @@
 import os,re
 import glob
 
-#定义你下载的一堆caj文件目录
-caj_path = "/home/ubuntu/caj2txt/caj"
-pdf_path = "/home/ubuntu/caj2txt/pdf"
-#定义你的txt输出目录
-output_path = "/home/ubuntu/caj2txt/txt"
+#定义你下载的一堆caj文件目录(最后剩下的是无法转化的caj)
+caj_path = "/root/caj2txt/caj"
+#可以被解析成pdf的caj文件目录
+pdf_path = "/root/caj2txt/pdf"
+#可以从pdf输出为txt的目录
+output_path = "/root/caj2txt/txt"
+#不能被转化txt的pdf目录(原因最可能是caj是扫描件，pdf也是图片形式)
+error_path = "/root/caj2txt/error"
 
 def batch_convert_to_text(caj_path):
  # 获取指定文件夹下所有的 .caj 文件路径
  caj_files = glob.glob(os.path.join(caj_path, "*.caj"))
  for caj_file in caj_files:
  # 转换为 PDF
- caj2pdf.convert_to_pdf(caj_file)
+ try:
+ caj2pdf.convert_to_pdf(caj_file)
+ except:
+ pass
 def pdf_to_text(pdf_path, output_path):
  # 读取生成pdf位置
  pdf_files = glob.glob(os.path.join(pdf_path, "*.pdf"))
@@ -25,15 +31,19 @@ def pdf_to_text(pdf_path, output_path):
  try:
  text = extract_text(pdf)
  except:
- print("file wrong:",txt_output_path)
+ error_path_full = os.path.join(error_path, os.path.basename(pdf))
+ os.rename(pdf, error_path_full)
+ print(f'Error processing {pdf}')
+ print(f'Moved {pdf} to {error_path_full}')
+
  with open(txt_output_path, 'w', encoding='utf-8') as f:
  f.write(text)
 
 
 
 
 #开关caj转pdf
-batch_convert_to_text(caj_path)
+#batch_convert_to_text(caj_path)
 
 
 #原作者把输出封装了当前目录
@@ -47,13 +57,16 @@ def pdf_to_text(pdf_path, output_path):
 # 如果文件是以.pdf结尾的文件，则移动到目标文件夹
  if filename.endswith('.pdf'):
  source_path = os.path.join(source_folder, filename)
+ source_base = os.path.splitext(filename)[0]
  destination_path = os.path.join(destination_folder, filename)
  os.rename(source_path, destination_path)
  print(f'Moved {filename} to {destination_folder}')
+ caj_filename = f"{source_base}.caj"
+ caj_path = os.path.join(source_folder, caj_filename)
+ # 如果同名的.caj文件存在，则删除
+ if os.path.exists(caj_path):
+ os.remove(caj_path)
+ print(f'Deleted {caj_filename}')
 
 # 开关pdf转txt
 pdf_to_text(pdf_path, output_path)
-
-
-
-