python使用PyPDF2库进行PDF基本操作

# -*- coding =utf-8 -*-
# @Time : 2021/1/26 10:09
# @Author :Mr
# @File :pdftest.py
# @Software :PyCharm
from PyPDF2 import PdfFileReader, PdfFileWriter
import os
# 创建输出文件目录
def mkdir(read_file_path, out_path=None):
if out_path is None:
out_path = (os.path.join(os.path.dirname(read_file_path),
os.path.basename(read_file_path)[:-4].replace(" ", "")))
if not os.path.exists(out_path): # 判断是否存在文件夹如果不存在则创建为文件夹
# print("--- 创建新的文件夹... ---")
os.makedirs(out_path) # makedirs 创建文件时如果路径不存在会创建这个路径
# print("--- OK ---")
else:
# print("--- 文件夹已存在! ---")
pass
return out_path
# 切分每一页
def split_pdf(read_file_path, out_path=None):
out_path = mkdir(read_file_path, out_path)
pdfReader = PdfFileReader(read_file_path)
for page in range(pdfReader.getNumPages()):
pdf_writer = PdfFileWriter()
pdf_writer.addPage(pdfReader.getPage(page))
pdf_name = f'{page}.pdf'
pdf_file = os.path.join(out_path, pdf_name)
with open(pdf_file, 'wb') as output_pdf:
pdf_writer.write(output_pdf)
return
# PDF文件按条件分割
def split_by_pages(read_file_path, out_detail, out_path=None):
out_path = mkdir(read_file_path, out_path)
try:
fp_read_file = open(read_file_path, 'rb')
pdfReader = PdfFileReader(fp_read_file) # 将要分割的PDF内容格式话
page_count = pdfReader.getNumPages() # 获取PDF页数
print(page_count) # 打印页数
with open(out_detail, 'r', True, 'utf-8')as fp:
# print(fp)
txt = fp.readlines()
# print(txt)
for detail in txt: # 打开分割标准文件
# print(type(detail))
pages, write_file = detail.split() # 空格分组
print(pages, write_file)
# write_file, write_ext = os.path.splitext(write_file) # 用于返回文件名和扩展名元组
pdf_file = f'{pages}页.pdf'
pdf_files = os.path.join(out_path, pdf_file)
print(pdf_files)
# liststr=list(map(int, pages.split('-')))
# print(type(liststr))
start_page, end_page = list(map(int, pages.split('-'))) # 将字符串数组转换成整形数组
start_page -= 1
try:
print(f'开始分割{start_page}页-{end_page}页，保存为{pdf_file}......')
pdf_output = PdfFileWriter() # 实例一个 PDF文件编写器
for i in range(start_page, end_page):
pdf_output.addPage(pdfReader.getPage(i))
with open(pdf_files, 'wb') as sub_fp:
pdf_output.write(sub_fp)
print(f'完成分割{start_page}页-{end_page}页，保存为{pdf_file}!')
except IndexError:
print(f'分割页数超过了PDF的页数')
# fp.close()
except Exception as e:
print(e)
finally:
fp_read_file.close()
return
# 合并pdf
def merge_pdfs(out_path=None, out_file='new', *read_file_paths):
# print(type(read_file_path))
if not read_file_paths:
print('请重新选择文件')
else:
read_file_path = read_file_paths[0]
out_path = mkdir(read_file_path, out_path)
pdf_name = f'{out_file}.pdf'
output = os.path.join(out_path, pdf_name)
pdf_writer = PdfFileWriter()
for File in read_file_paths:
# 依次循环打开要合并文件
pdfReader = PdfFileReader(open(File, 'rb'))
numPages = pdfReader.getNumPages()
for index in range(0, numPages):
pageObj = pdfReader.getPage(index)
pdf_writer.addPage(pageObj)
# 把这个已合并了的PDF文档存储起来
with open(output, 'wb') as out:
pdf_writer.write(out)
return
# 删除PDF页
def delete_pdf(read_file_path, index, out_path=None):
out_path = mkdir(read_file_path, out_path)
pdfReader = PdfFileReader(open(read_file_path, "rb"))
pages = pdfReader.getNumPages()
pdf_writer = PdfFileWriter()
pdf_file = os.path.join(out_path, f'{os.path.basename(read_file_path)[:-4].replace(" ", "")}new.pdf')
print(pdf_file)
for page in range(pages):
if page + 1 in index:
continue
pdf_writer.addPage(pdfReader.getPage(page))
with open(pdf_file, 'wb') as out:
pdf_writer.write(out)
return