init commit

Author: Zang Zhiya

.gitignore

@@ -0,0 +1,5 @@
+**/__pycache__
+*.pyc
+*.egg-info
+
+.DS_Store

config.cfg

@@ -0,0 +1,4 @@
+[default]
+pdf_folder=/Users/Zzy/Documents/Code/pdf2word/pdf
+word_folder=/Users/Zzy/Documents/Code/pdf2word/word
+max_worker=5

main.py

@@ -0,0 +1,76 @@
+import os
+from configparser import ConfigParser
+from io import StringIO
+from io import open
+from concurrent.futures import ProcessPoolExecutor
+
+from pdfminer.pdfinterp import PDFResourceManager
+from pdfminer.pdfinterp import process_pdf
+from pdfminer.converter import TextConverter
+from pdfminer.layout import LAParams
+from docx import Document
+
+
+def read_from_pdf(file_path):
+    with open(file_path, 'rb') as file:
+        resource_manager = PDFResourceManager()
+        return_str = StringIO()
+        lap_params = LAParams()
+
+        device = TextConverter(
+            resource_manager, return_str, laparams=lap_params)
+        process_pdf(resource_manager, device, file)
+        device.close()
+
+        content = return_str.getvalue()
+        return_str.close()
+        return content
+
+
+def save_text_to_word(content, file_path):
+    doc = Document()
+    for line in content.split('\n'):
+        paragraph = doc.add_paragraph()
+        paragraph.add_run(remove_control_characters(line))
+    doc.save(file_path)
+
+
+def remove_control_characters(content):
+    mpa = dict.fromkeys(range(32))
+    return content.translate(mpa)
+
+
+def pdf_to_word(pdf_file_path, word_file_path):
+    content = read_from_pdf(pdf_file_path)
+    save_text_to_word(content, word_file_path)
+
+
+def main():
+    config_parser = ConfigParser()
+    config_parser.read('config.cfg')
+    config = config_parser['default']
+
+    tasks = []
+    with ProcessPoolExecutor(max_workers=int(config['max_worker'])) as executor:
+        for file in os.listdir(config['pdf_folder']):
+            extension_name = os.path.splitext(file)[1]
+            if extension_name != '.pdf':
+                continue
+            file_name = os.path.splitext(file)[0]
+            pdf_file = config['pdf_folder'] + '/' + file
+            word_file = config['word_folder'] + '/' + file_name + '.docx'
+            print('正在处理: ', file)
+            result = executor.submit(pdf_to_word, pdf_file, word_file)
+            tasks.append(result)
+    while True:
+        exit_flag = True
+        for task in tasks:
+            if not task.done():
+                exit_flag = False
+        if exit_flag:
+            print('完成')
+            exit(0)
+
+
+if __name__ == '__main__':
+    main()

requirements.txt

@@ -0,0 +1,9 @@
+attrs==17.4.0
+lxml==4.1.1
+pdfminer3k==1.3.1
+pluggy==0.6.0
+ply==3.11
+py==1.5.2
+pytest==3.4.1
+python-docx==0.8.6
+six==1.11.0