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

在这篇博客中,我将介绍一个使用Python编写的小工具,它能够将指定文件夹中的所有Word文档(.doc和.docx格式)转换为PNG图片。这个工具基于wxPython库构建图形用户界面,并结合了win32com和PyMuPDF库实现文档格式的转换。接下来,我将详细说明这个工具的功能及其实现。
D:\spiderdocs\wordtoscreenshot.py

全部代码

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()

功能概述

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

  1. 选择文件夹:用户可以通过GUI界面选择包含Word文档的文件夹。
  2. 转换文档:将选择的文件夹中的所有Word文档转换为PNG图片,并记录转换日志。
  3. 显示消息:在转换完成后,显示一个消息框提示用户检查转换日志。

代码实现

导入必要的库

首先,我们需要导入一些必要的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()

代码解析

  1. 导入库:我们导入了os、wx、win32com、pythoncom、sys、traceback、tempfile和fitz库。这些库分别用于文件操作、创建GUI、与Word应用程序交互、处理异常、创建临时文件以及处理PDF文件。

  2. convert_word_to_pdf函数:这个函数使用win32com库将Word文档转换为PDF格式。它首先初始化COM库,然后创建一个Word应用程序实例,打开指定的Word文档,并将其导出为PDF格式。最后,它关闭文档并退出Word应用程序。

  3. convert_pdf_to_png函数:这个函数使用PyMuPDF库将PDF文件的第一页转换为PNG图片。它打开指定的PDF文件,加载第一页,生成图像并保存为PNG格式。

  4. MyFrame类:这是我们的GUI类,继承自wx.Frame。它包含一个文本框用于显示和输入文件夹路径,一个浏览按钮用于选择文件夹,以及一个转换按钮用于启动转换过程。on_browse方法用于处理浏览按钮点击事件,on_convert方法用于处理转换按钮点击事件。

  5. on_convert方法:这个方法首先获取用户选择的文件夹路径,然后遍历该文件夹中的所有Word文档,依次将其转换为PDF格式,再将PDF文件转换为PNG图片。转换过程中记录日志,并在转换完成后显示消息框。

  6. 主程序:在主程序中,我们创建一个wx.App实例,并创建MyFrame实例来显示GUI。

结果如下

在这里插入图片描述

总结

通过这篇博客,我们介绍了如何使用Python和wxPython库创建一个简单的GUI工具,将指定文件夹中的所有Word文档转换为PNG图片。这个工具使用了win32com库与Word应用程序交互,将Word文档导出为PDF格式,并使用PyMuPDF库将PDF文件转换为PNG图片。希望这篇博客对你有所帮助!

  • 31
    点赞
  • 31
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值