基于Python开发PPT图片提取与合并工具

 更新时间:2024年12月23日 14:25:38   作者:winfredzhang  
在日常工作中,我们经常需要处理PPT中的图片,本文将介绍如何使用Python开发一个图形界面工具,实现PPT图片提取和九宫格合并功能,需要的可以参考下

在日常工作中,我们经常需要处理PPT中的图片,有时需要批量提取,有时需要将多张图片合并成特定布局。本文将介绍如何使用Python开发一个图形界面工具,实现PPT图片提取和九宫格合并功能。

全部代码

import wx
import os
from pptx import Presentation
from PIL import Image
import io

class MainFrame(wx.Frame):
    def __init__(self):
        super().__init__(parent=None, title='PPT图片提取与合并工具', size=(600, 400))
        self.ppt_path = ''
        self.output_dir = ''
        self.InitUI()
        
    def InitUI(self):
        panel = wx.Panel(self)
        vbox = wx.BoxSizer(wx.VERTICAL)
        
        # PPT选择
        hbox1 = wx.BoxSizer(wx.HORIZONTAL)
        self.ppt_text = wx.TextCtrl(panel)
        choose_btn = wx.Button(panel, label='选择PPT')
        choose_btn.Bind(wx.EVT_BUTTON, self.OnChoosePPT)
        hbox1.Add(self.ppt_text, proportion=1, flag=wx.EXPAND|wx.ALL, border=5)
        hbox1.Add(choose_btn, flag=wx.ALL, border=5)
        
        # 输出目录选择
        hbox2 = wx.BoxSizer(wx.HORIZONTAL)
        self.dir_text = wx.TextCtrl(panel)
        dir_btn = wx.Button(panel, label='选择输出目录')
        dir_btn.Bind(wx.EVT_BUTTON, self.OnChooseDir)
        hbox2.Add(self.dir_text, proportion=1, flag=wx.EXPAND|wx.ALL, border=5)
        hbox2.Add(dir_btn, flag=wx.ALL, border=5)
        
        # 提取按钮
        extract_btn = wx.Button(panel, label='提取图片')
        extract_btn.Bind(wx.EVT_BUTTON, self.OnExtract)
        
        # 合并设置
        hbox3 = wx.BoxSizer(wx.HORIZONTAL)
        row_label = wx.StaticText(panel, label='行数:')
        self.row_text = wx.TextCtrl(panel, value='3')
        col_label = wx.StaticText(panel, label='列数:')
        self.col_text = wx.TextCtrl(panel, value='3')
        merge_btn = wx.Button(panel, label='合并图片')
        merge_btn.Bind(wx.EVT_BUTTON, self.OnMerge)
        
        hbox3.Add(row_label, flag=wx.ALL, border=5)
        hbox3.Add(self.row_text, flag=wx.ALL, border=5)
        hbox3.Add(col_label, flag=wx.ALL, border=5)
        hbox3.Add(self.col_text, flag=wx.ALL, border=5)
        hbox3.Add(merge_btn, flag=wx.ALL, border=5)
        
        # 状态显示
        self.status_text = wx.TextCtrl(panel, style=wx.TE_MULTILINE|wx.TE_READONLY)
        
        # 添加到主布局
        vbox.Add(hbox1, flag=wx.EXPAND)
        vbox.Add(hbox2, flag=wx.EXPAND)
        vbox.Add(extract_btn, flag=wx.EXPAND|wx.ALL, border=5)
        vbox.Add(hbox3, flag=wx.EXPAND)
        vbox.Add(self.status_text, proportion=1, flag=wx.EXPAND|wx.ALL, border=5)
        
        panel.SetSizer(vbox)
    
    def OnChoosePPT(self, event):
        dlg = wx.FileDialog(self, "选择PPT文件", "", "", 
                          "PPT files (*.pptx)|*.pptx", wx.FD_OPEN)
        if dlg.ShowModal() == wx.ID_OK:
            self.ppt_path = dlg.GetPath()
            self.ppt_text.SetValue(self.ppt_path)
        dlg.Destroy()
    
    def OnChooseDir(self, event):
        dlg = wx.DirDialog(self, "选择输出目录")
        if dlg.ShowModal() == wx.ID_OK:
            self.output_dir = dlg.GetPath()
            self.dir_text.SetValue(self.output_dir)
        dlg.Destroy()
    
    def OnExtract(self, event):
        if not self.ppt_path or not self.output_dir:
            wx.MessageBox('请选择PPT文件和输出目录', '错误')
            return
            
        try:
            prs = Presentation(self.ppt_path)
            image_count = 0
            
            for slide in prs.slides:
                for shape in slide.shapes:
                    if hasattr(shape, "image"):
                        image_stream = io.BytesIO(shape.image.blob)
                        image = Image.open(image_stream)
                        image_path = os.path.join(self.output_dir, f'image_{image_count}.png')
                        image.save(image_path)
                        image_count += 1
                        
            self.status_text.AppendText(f'成功提取{image_count}张图片\n')
            
        except Exception as e:
            wx.MessageBox(f'提取图片时出错: {str(e)}', '错误')
    
    def OnMerge(self, event):
        try:
            rows = int(self.row_text.GetValue())
            cols = int(self.col_text.GetValue())
        except ValueError:
            wx.MessageBox('请输入有效的行数和列数', '错误')
            return
            
        if not self.output_dir:
            wx.MessageBox('请选择输出目录', '错误')
            return
            
        try:
            # 获取所有图片文件
            image_files = [f for f in os.listdir(self.output_dir) 
                         if f.startswith('image_') and f.endswith('.png')]
            image_files.sort()
            
            if len(image_files) < rows * cols:
                wx.MessageBox('图片数量不足', '错误')
                return
                
            # 读取第一张图片获取尺寸
            first_image = Image.open(os.path.join(self.output_dir, image_files[0]))
            img_width, img_height = first_image.size
            
            # 创建合并后的画布
            merged = Image.new('RGB', (img_width * cols, img_height * rows))
            
            # 拼接图片
            for idx, img_file in enumerate(image_files[:rows * cols]):
                if idx >= rows * cols:
                    break
                    
                img = Image.open(os.path.join(self.output_dir, img_file))
                x = (idx % cols) * img_width
                y = (idx // cols) * img_height
                merged.paste(img, (x, y))
            
            # 保存合并后的图片
            merged_path = os.path.join(self.output_dir, 'merged.png')
            merged.save(merged_path)
            self.status_text.AppendText(f'成功合并图片: {merged_path}\n')
            
        except Exception as e:
            wx.MessageBox(f'合并图片时出错: {str(e)}', '错误')

def main():
    app = wx.App()
    frame = MainFrame()
    frame.Show()
    app.MainLoop()

if __name__ == '__main__':
    main()

功能概述

这个工具主要实现两个核心功能:

从PPT文档中批量提取所有图片

将提取的图片按照指定的行列数合并成九宫格布局

技术栈选择

为了实现这个工具,我们选用了以下Python库:

wxPython:用于创建图形用户界面

python-pptx:用于处理PPT文档

Pillow(PIL):用于图片处理和合并

io:用于处理二进制数据流

详细设计

1. 用户界面设计

我们使用wxPython创建了一个简洁的图形界面,包含以下组件:

  • PPT文件选择区域
  • 输出目录选择区域
  • 图片提取按钮
  • 行列数输入框
  • 图片合并按钮
  • 状态显示文本框

界面布局采用垂直和水平布局器(BoxSizer)组合,确保各个组件能够合理排列和自适应窗口大小。

2. 核心功能实现

PPT图片提取功能

def OnExtract(self, event):
    if not self.ppt_path or not self.output_dir:
        wx.MessageBox('请选择PPT文件和输出目录', '错误')
        return
        
    try:
        prs = Presentation(self.ppt_path)
        image_count = 0
        
        for slide in prs.slides:
            for shape in slide.shapes:
                if hasattr(shape, "image"):
                    image_stream = io.BytesIO(shape.image.blob)
                    image = Image.open(image_stream)
                    image_path = os.path.join(self.output_dir, f'image_{image_count}.png')
                    image.save(image_path)
                    image_count += 1
                    
        self.status_text.AppendText(f'成功提取{image_count}张图片\n')
        
    except Exception as e:
        wx.MessageBox(f'提取图片时出错: {str(e)}', '错误')

这段代码通过python-pptx库打开PPT文档,遍历所有幻灯片和形状,找到图片类型的形状后,将其转换为PIL Image对象并保存到指定目录。

图片合并功能

def OnMerge(self, event):
    try:
        rows = int(self.row_text.GetValue())
        cols = int(self.col_text.GetValue())
    except ValueError:
        wx.MessageBox('请输入有效的行数和列数', '错误')
        return
        
    # ... 获取图片文件列表 ...
    
    # 创建合并后的画布
    merged = Image.new('RGB', (img_width * cols, img_height * rows))
    
    # 拼接图片
    for idx, img_file in enumerate(image_files[:rows * cols]):
        if idx >= rows * cols:
            break
            
        img = Image.open(os.path.join(self.output_dir, img_file))
        x = (idx % cols) * img_width
        y = (idx // cols) * img_height
        merged.paste(img, (x, y))
    
    merged.save(os.path.join(self.output_dir, 'merged.png'))

合并功能首先创建一个足够大的空白画布,然后按照行列顺序将图片粘贴到对应位置。

使用说明

环境准备

在使用此工具前,需要安装必要的Python库:

pip install wxPython python-pptx Pillow

使用步骤

运行程序

点击"选择PPT"按钮,选择要处理的PPT文件

点击"选择输出目录"按钮,选择图片保存位置

点击"提取图片"按钮,程序会自动提取PPT中的所有图片

输入想要的行数和列数(默认3x3)

点击"合并图片"按钮,程序会生成合并后的图片

结果如下

注意事项

  • PPT文件必须是.pptx格式
  • 确保有足够的磁盘空间存储提取的图片
  • 合并时,建议使用相同尺寸的图片,否则可能会出现布局不均匀的情况
  • 图片数量应不少于行数×列数,否则会提示错误

以上就是基于Python开发PPT图片提取与合并工具的详细内容,更多关于Python PPT图片提取与合并的资料请关注脚本之家其它相关文章!

相关文章

最新评论