设为首页收藏本站
切换到窄版

鱼C论坛

 找回密码
 立即注册
鱼C论坛»论坛 › 编程语言专区 › Python交流 › 用Python写了一个PDF合并的小工具
返回列表 发新帖
查看: 3753|回复: 2

[技术交流] 用Python写了一个PDF合并的小工具

[复制链接]
JBpeople
发表于 2022-1-18 09:37:13 | 显示全部楼层 |阅读模式

马上注册，结交更多好友，享用更多功能^_^

您需要 登录 才可以下载或查看，没有账号？立即注册

x
  1. # -*- coding: utf-8 -*-
  2. # @Time : 2021-12-6 9:10
  3. # @Author : Administrator
  4. # @Email : lcmmljs@foxmail.com
  5. # @File : main.py
  6. # @Project : Project

  8. import glob
  9. import PySimpleGUI as sg
  10. from PyPDF2 import PdfFileReader, PdfFileWriter

  12. # 创建GUI布局
  13. fra1 = sg.Frame('打开文件夹', [[sg.InputText(key='-IN-'), sg.FolderBrowse('文件位置')], [sg.Button('确认')]])
  14. fra2 = sg.Frame('文件列表', [[sg.Listbox(values=[''], s=(54, 30), key='-LIST-')],
  15.                          [sg.Button('上移'), sg.Button('下移'), sg.Button('合并')]])

  17. col = sg.Column([[fra1], [fra2]], pad=(0, 0))

  19. layout = [[col]]
  20. window = sg.Window('合并PDF', layout, keep_on_top=True)

  22. # 创建程序主循环
  23. while True:
  24.     event, value = window.read()
  25.     if event is None:
  26.         break
  27.     elif event == '确认':
  28.         if value['-IN-'] != '':
  29.             path = window['-IN-'].get() + '/*.pdf'
  30.             file = glob.glob(path)
  31.             if file:
  32.                 window['-LIST-'].update(file)
  33.             else:
  34.                 sg.popup_ok('文件夹内未识别到PDF文件~', title='错误', keep_on_top=True)
  35.         else:
  36.             sg.popup_ok('请选择文件夹~', title='错误', keep_on_top=True)
  37.     elif event == '上移':
  38.         if value['-LIST-']:
  39.             choice = value['-LIST-']
  40.             index = file.index(choice[0])
  41.             file[index], file[index-1] = file[index-1], file[index]
  42.             window['-LIST-'].update(file)
  43.     elif event == '下移':
  44.         if value['-LIST-']:
  45.             choice = value['-LIST-']
  46.             index = file.index(choice[0])
  47.             file[index], file[index+1] = file[index+1], file[index]
  48.             window['-LIST-'].update(file)
  49.     elif event == '合并':
  50.         if file:
  51.             pdfFileWriter = PdfFileWriter()
  52.             for pdf in file:
  53.                 pdfReader = PdfFileReader(open(pdf, 'rb'))
  54.                 numPages = pdfReader.getNumPages()
  55.                 for i in range(0, numPages):
  56.                     pageObj = pdfReader.getPage(i)
  57.                     pdfFileWriter.addPage(pageObj)
  58.             folder = sg.popup_get_folder('请选择文件保存位置：', title='提示', keep_on_top=True)
  59.             pdfFileWriter.write(open(folder + '//合并文件.pdf', 'wb'))
  60. window.close()
复制代码
小甲鱼最新课程 -> https://ilovefishc.com
回复

使用道具 举报

JBpeople
 楼主| 发表于 2022-1-18 11:19:52 | 显示全部楼层
稍微更新一下~
  1. # -*- coding: utf-8 -*-
  2. # @Time : 2021-12-6 9:10
  3. # @Author : Administrator
  4. # @Email : lcmmljs@foxmail.com
  5. # @File : main.py
  6. # @Project : Project

  8. import glob
  9. import PySimpleGUI as sg
  10. from PyPDF2 import PdfFileReader, PdfFileWriter

  12. # 创建GUI布局
  13. fra1 = sg.Frame('打开文件夹', [[sg.InputText(key='-IN-'), sg.FolderBrowse('文件位置')], [sg.Button('确认')]])
  14. fra2 = sg.Frame('文件列表', [[sg.Listbox(values=[''], s=(54, 30), key='-LIST-')],
  15.                          [sg.Button('上移'), sg.Button('下移'), sg.Button('移除'), sg.Button('合并')]])

  17. col = sg.Column([[fra1], [fra2]], pad=(0, 0))

  19. layout = [[col]]
  20. window = sg.Window('合并PDF', layout, keep_on_top=True)

  22. # 创建程序主循环
  23. while True:
  24.     event, value = window.read()
  25.     if event is None:
  26.         break
  27.     elif event == '确认':
  28.         if value['-IN-'] != '':
  29.             path = window['-IN-'].get() + '/*.pdf'
  30.             file = glob.glob(path)
  31.             if file:
  32.                 filefloder = file[0].split('\\')[0] + '/'
  33.                 filename = [i.split('\\')[-1] for i in file]
  34.                 window['-LIST-'].update(filename)
  35.             else:
  36.                 sg.popup_ok('文件夹内未识别到PDF文件~', title='错误', keep_on_top=True)
  37.         else:
  38.             sg.popup_ok('请选择文件夹~', title='错误', keep_on_top=True)
  39.     elif event == '上移':
  40.         try:
  41.             choice = value['-LIST-']
  42.             index = filename.index(choice[0])
  43.             filename[index], filename[index-1] = filename[index-1], filename[index]
  44.             window['-LIST-'].update(filename)
  45.         except NameError:
  46.             pass
  47.     elif event == '下移':
  48.         try:
  49.             choice = value['-LIST-']
  50.             index = filename.index(choice[0])
  51.             filename[index], filename[index+1] = filename[index+1], filename[index]
  52.             window['-LIST-'].update(filename)
  53.         except NameError:
  54.             pass
  55.     elif event == '移除':
  56.         try:
  57.             choice = value['-LIST-'][0]
  58.             filename.remove(choice)
  59.             window['-LIST-'].update(filename)
  60.         except NameError:
  61.             pass
  62.         except IndexError:
  63.             pass
  64.     elif event == '合并':
  65.         try:
  66.             if filename:
  67.                 file = [filefloder + i for i in filename]
  68.                 pdfFileWriter = PdfFileWriter()
  69.                 for pdf in file:
  70.                     pdfReader = PdfFileReader(open(pdf, 'rb'))
  71.                     numPages = pdfReader.getNumPages()
  72.                     for i in range(0, numPages):
  73.                         pageObj = pdfReader.getPage(i)
  74.                         pdfFileWriter.addPage(pageObj)
  75.                 folder = sg.popup_get_folder('请选择文件保存位置：', title='提示', keep_on_top=True)
  76.                 pdfFileWriter.write(open(folder + '//合并文件.pdf', 'wb'))
  77.                 sg.popup_ok('文件已合并完成~', keep_on_top=True)
  78.         except NameError:
  79.             pass
  80. window.close()
复制代码
小甲鱼最新课程 -> https://ilovefishc.com
回复 支持 反对

使用道具 举报

楪祈53409
发表于 2023-6-8 19:26:49 | 显示全部楼层
本帖最后由 楪祈53409 于 2023-6-8 19:30 编辑

你好,我对你的代码进行了维护,修改了部分被废弃的函数.
我在py3.0上运行的,初次运行没安装库的先win+r分别运行
  1. pip install PySimpleGUI
复制代码
和
  1. pip install PyPDF2
复制代码

安装所需的库.
以下是我维护好的代码:
  1. import glob
  2. import PySimpleGUI as sg
  3. from PyPDF2 import PdfWriter, PdfReader

  5. fra1 = sg.Frame('打开文件夹', [[sg.InputText(key='-IN-'), sg.FolderBrowse('文件位置')], [sg.Button('确认')]])
  6. fra2 = sg.Frame('文件列表', [[sg.Listbox(values=[''], size=(54, 30), key='-LIST-')],
  7.                          [sg.Button('上移'), sg.Button('下移'), sg.Button('移除'), sg.Button('合并')]])

  9. col = sg.Column([[fra1], [fra2]], pad=(0, 0))

  11. layout = [[col]]
  12. window = sg.Window('合并PDF', layout, keep_on_top=True)

  14. filename = []

  16. while True:
  17.     event, value = window.read()
  18.     if event is None:
  19.         break
  20.     elif event == '确认':
  21.         if value['-IN-'] != '':
  22.             path = value['-IN-'] + '/*.pdf'
  23.             file = glob.glob(path)
  24.             if file:
  25.                 filefolder = file[0].split('\\')[0] + '/'
  26.                 filename = [i.split('\\')[-1] for i in file]
  27.                 window['-LIST-'].update(filename)
  28.             else:
  29.                 sg.popup_ok('文件夹内未识别到PDF文件~', title='错误', keep_on_top=True)
  30.         else:
  31.             sg.popup_ok('请选择文件夹~', title='错误', keep_on_top=True)
  32.     elif event == '上移':
  33.         try:
  34.             choice = value['-LIST-']
  35.             index = filename.index(choice[0])
  36.             filename[index], filename[index-1] = filename[index-1], filename[index]
  37.             window['-LIST-'].update(filename)
  38.         except NameError:
  39.             pass
  40.     elif event == '下移':
  41.         try:
  42.             choice = value['-LIST-']
  43.             index = filename.index(choice[0])
  44.             filename[index], filename[index+1] = filename[index+1], filename[index]
  45.             window['-LIST-'].update(filename)
  46.         except NameError:
  47.             pass
  48.         except IndexError:
  49.             pass
  50.     elif event == '移除':
  51.         try:
  52.             choice = value['-LIST-'][0]
  53.             filename.remove(choice)
  54.             window['-LIST-'].update(filename)
  55.         except NameError:
  56.             pass
  57.         except IndexError:
  58.             pass
  59.     elif event == '合并':
  60.         try:
  61.             if filename:
  62.                 file = [filefolder + i for i in filename]
  63.                 pdfFileWriter = PdfWriter()
  64.                 for pdf in file:
  65.                     pdfReader = PdfReader(open(pdf, 'rb'))
  66.                     for page in pdfReader.pages:
  67.                         pdfFileWriter.add_page(page)
  68.                 folder = sg.popup_get_folder('请选择文件保存位置：', title='提示', keep_on_top=True)
  69.                 with open(folder + '/合并文件.pdf', 'wb') as output_pdf:
  70.                     pdfFileWriter.write(output_pdf)
  71.                 sg.popup_ok('文件已合并完成~', keep_on_top=True)
  72.         except NameError:
  73.             pass

  75. window.close()
复制代码

小甲鱼最新课程 -> https://ilovefishc.com
回复 支持 反对

使用道具 举报

返回列表 发新帖
高级模式
B Color Image Link Quote Code Smilies
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

小黑屋|手机版|Archiver|鱼C工作室 ( 粤ICP备18085999号-1 | 粤公网安备 44051102000585号)

GMT+8, 2025-4-24 00:08

Powered by Discuz! X3.4

© 2001-2023 Discuz! Team.

快速回复 返回顶部 返回列表