使用Python实现将Word文档转换为PNG图片_Python

前言

在这篇博客中，我将介绍一个使用python编写的小工具，它能够将指定文件夹中的所有word文档（.doc和.docx格式）转换为png图片。这个工具基于wxpython库构建图形用户界面，并结合了win32com和pymupdf库实现文档格式的转换。接下来，我将详细说明这个工具的功能及其实现。

全部代码

import os
import wx
from win32com import client as wc
import pythoncom
import sys
import traceback
import tempfile
import fitz  # pymupdf

def convert_word_to_pdf(word_path, pdf_path):
    pythoncom.coinitialize()
    word = none
    doc = none
    try:
        word = wc.dispatch("word.application")
        word.visible = false
        doc = word.documents.open(word_path)
        doc.exportasfixedformat(pdf_path, 17)  # 17 is wdexportformatpdf
        print(f"successfully exported {word_path} to {pdf_path}")
    except exception as e:
        print(f"error in convert_word_to_pdf: {str(e)}")
        print("traceback:")
        traceback.print_exc()
        raise
    finally:
        if doc:
            doc.close(savechanges=false)
        if word:
            word.quit()
        pythoncom.couninitialize()

def convert_pdf_to_png(pdf_path, png_path):
    try:
        doc = fitz.open(pdf_path)
        page = doc.load_page(0)  # load the first page
        pix = page.get_pixmap()
        pix.save(png_path)
        doc.close()
        print(f"successfully converted {pdf_path} to {png_path}")
    except exception as e:
        print(f"error in convert_pdf_to_png: {str(e)}")
        print("traceback:")
        traceback.print_exc()
        raise

class myframe(wx.frame):
    def __init__(self):
        super().__init__(parent=none, title='word to png converter')
        panel = wx.panel(self)
        
        self.folder_path = wx.textctrl(panel, pos=(5, 5), size=(350, 25))
        browse_button = wx.button(panel, label='browse', pos=(360, 5), size=(70, 25))
        browse_button.bind(wx.evt_button, self.on_browse)
        
        convert_button = wx.button(panel, label='convert', pos=(5, 35), size=(425, 25))
        convert_button.bind(wx.evt_button, self.on_convert)
        
        self.setsize((450, 100))
        self.show()

    def on_browse(self, event):
        dlg = wx.dirdialog(self, "choose a directory:", style=wx.dd_default_style)
        if dlg.showmodal() == wx.id_ok:
            self.folder_path.setvalue(dlg.getpath())
        dlg.destroy()

    def on_convert(self, event):
        folder = self.folder_path.getvalue()
        if not folder:
            wx.messagebox('please select a folder first', 'error', wx.ok | wx.icon_error)
            return
        
        log = []
        for filename in os.listdir(folder):
            if filename.endswith('.doc') or filename.endswith('.docx'):
                word_path = os.path.join(folder, filename)
                png_path = os.path.splitext(word_path)[0] + '.png'
                try:
                    with tempfile.namedtemporaryfile(suffix='.pdf', delete=false) as tmp_pdf:
                        pdf_path = tmp_pdf.name
                    
                    convert_word_to_pdf(word_path, pdf_path)
                    convert_pdf_to_png(pdf_path, png_path)
                    
                    os.unlink(pdf_path)  # remove the temporary pdf file
                    log.append(f'converted {filename} to png')
                except exception as e:
                    log.append(f'error converting {filename}: {str(e)}')
        
        log_str = '\n'.join(log)
        with open('conversion_log.txt', 'w') as f:
            f.write(log_str)
        
        wx.messagebox('conversion completed. check conversion_log.txt for details.', 'info', wx.ok | wx.icon_information)

if __name__ == '__main__':
    app = wx.app()
    frame = myframe()
    app.mainloop()

功能概述

这个工具的主要功能包括：

选择文件夹：用户可以通过gui界面选择包含word文档的文件夹。
转换文档：将选择的文件夹中的所有word文档转换为png图片，并记录转换日志。
显示消息：在转换完成后，显示一个消息框提示用户检查转换日志。

代码实现

导入必要的库

首先，我们需要导入一些必要的python库：

import os
import wx
from win32com import client as wc
import pythoncom
import sys
import traceback
import tempfile
import fitz  # pymupdf

word到pdf的转换函数

使用win32com库中的word应用程序接口，我们可以将word文档转换为pdf格式：

def convert_word_to_pdf(word_path, pdf_path):
    pythoncom.coinitialize()
    word = none
    doc = none
    try:
        word = wc.dispatch("word.application")
        word.visible = false
        doc = word.documents.open(word_path)
        doc.exportasfixedformat(pdf_path, 17)  # 17 is wdexportformatpdf
        print(f"successfully exported {word_path} to {pdf_path}")
    except exception as e:
        print(f"error in convert_word_to_pdf: {str(e)}")
        print("traceback:")
        traceback.print_exc()
        raise
    finally:
        if doc:
            doc.close(savechanges=false)
        if word:
            word.quit()
        pythoncom.couninitialize()

pdf到png的转换函数

接着，我们使用pymupdf库将pdf文件转换为png图片：

def convert_pdf_to_png(pdf_path, png_path):
    try:
        doc = fitz.open(pdf_path)
        page = doc.load_page(0)  # load the first page
        pix = page.get_pixmap()
        pix.save(png_path)
        doc.close()
        print(f"successfully converted {pdf_path} to {png_path}")
    except exception as e:
        print(f"error in convert_pdf_to_png: {str(e)}")
        print("traceback:")
        traceback.print_exc()
        raise

图形用户界面（gui）

我们使用wxpython库创建一个简单的gui，允许用户选择文件夹并启动转换：

class myframe(wx.frame):
    def __init__(self):
        super().__init__(parent=none, title='word to png converter')
        panel = wx.panel(self)
        
        self.folder_path = wx.textctrl(panel, pos=(5, 5), size=(350, 25))
        browse_button = wx.button(panel, label='browse', pos=(360, 5), size=(70, 25))
        browse_button.bind(wx.evt_button, self.on_browse)
        
        convert_button = wx.button(panel, label='convert', pos=(5, 35), size=(425, 25))
        convert_button.bind(wx.evt_button, self.on_convert)
        
        self.setsize((450, 100))
        self.show()

    def on_browse(self, event):
        dlg = wx.dirdialog(self, "choose a directory:", style=wx.dd_default_style)
        if dlg.showmodal() == wx.id_ok:
            self.folder_path.setvalue(dlg.getpath())
        dlg.destroy()

    def on_convert(self, event):
        folder = self.folder_path.getvalue()
        if not folder:
            wx.messagebox('please select a folder first', 'error', wx.ok | wx.icon_error)
            return
        
        log = []
        for filename in os.listdir(folder):
            if filename.endswith('.doc') or filename.endswith('.docx'):
                word_path = os.path.join(folder, filename)
                png_path = os.path.splitext(word_path)[0] + '.png'
                try:
                    with tempfile.namedtemporaryfile(suffix='.pdf', delete=false) as tmp_pdf:
                        pdf_path = tmp_pdf.name
                    
                    convert_word_to_pdf(word_path, pdf_path)
                    convert_pdf_to_png(pdf_path, png_path)
                    
                    os.unlink(pdf_path)  # remove the temporary pdf file
                    log.append(f'converted {filename} to png')
                except exception as e:
                    log.append(f'error converting {filename}: {str(e)}')
        
        log_str = '\n'.join(log)
        with open('conversion_log.txt', 'w') as f:
            f.write(log_str)
        
        wx.messagebox('conversion completed. check conversion_log.txt for details.', 'info', wx.ok | wx.icon_information)

if __name__ == '__main__':
    app = wx.app()
    frame = myframe()
    app.mainloop()

代码解析

导入库：我们导入了os、wx、win32com、pythoncom、sys、traceback、tempfile和fitz库。这些库分别用于文件操作、创建gui、与word应用程序交互、处理异常、创建临时文件以及处理pdf文件。
convert_word_to_pdf函数：这个函数使用win32com库将word文档转换为pdf格式。它首先初始化com库，然后创建一个word应用程序实例，打开指定的word文档，并将其导出为pdf格式。最后，它关闭文档并退出word应用程序。
convert_pdf_to_png函数：这个函数使用pymupdf库将pdf文件的第一页转换为png图片。它打开指定的pdf文件，加载第一页，生成图像并保存为png格式。
myframe类：这是我们的gui类，继承自wx.frame。它包含一个文本框用于显示和输入文件夹路径，一个浏览按钮用于选择文件夹，以及一个转换按钮用于启动转换过程。on_browse方法用于处理浏览按钮点击事件，on_convert方法用于处理转换按钮点击事件。
on_convert方法：这个方法首先获取用户选择的文件夹路径，然后遍历该文件夹中的所有word文档，依次将其转换为pdf格式，再将pdf文件转换为png图片。转换过程中记录日志，并在转换完成后显示消息框。
主程序：在主程序中，我们创建一个wx.app实例，并创建myframe实例来显示gui。