欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

Python PyPDF2模块安装使用解析

程序员文章站 2022-07-18 21:26:17
这篇文章主要介绍了python pypdf2模块安装使用解析,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下 pypdf2模块主要的...

这篇文章主要介绍了python pypdf2模块安装使用解析,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下

pypdf2模块主要的功能是分割或合并pdf文件,裁剪或转换pdf文件中的页面。

0、安装pypdf2的模块

pip install pypdf2

1、常用的函数

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @time  : 2020/1/15 13:38
# @author : suk
# @file  : pyxl.py
# @software: pycharm
import pypdf2

reader = pypdf2.pdffilereader(open('linux.pdf', 'rb'))
print(reader.getnumpages()) # 获取pdf总页数
print(reader.isencrypted) # 判断是否有加密
page = reader.getpage(4) # 获取第四页
print(page.extracttext()) # 获取第四页的内容
print(reader.getdocumentinfo()) # 获取pdf元信息,即创建时间,作者,标题等

2、读取pdf文件,取指定页数,写入到硬盘上的示例

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import pypdf2

reader = pypdf2.pdffilereader(open('linux.pdf', 'rb'))

output = pypdf2.pdffilewriter()

output.addpage(reader.getpage(1))
output.addpage(reader.getpage(4))
output.addpage(reader.getpage(5))
print(output.getnumpages()) # 获取写入页的总页数

output.encrypt('123456')
outputstream = open('pypdf2-output.pdf', 'wb')
output.write(outputstream)
outputstream.close()

3、读取pdf某一页,旋转180度后,写入到新的pdf文件的示例

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import pypdf2

reader = pypdf2.pdffilereader(open('linux.pdf', 'rb'))

page = reader.getpage(0) # 获取第0页
page.rotateclockwise(180) # 旋转180度

writer = pypdf2.pdffilewriter() # 创建pdf写入的对象
writer.addpage(page)

outputstream = open('rotate-page-test.pdf', 'wb') # 创建一个pdf文件
writer.write(outputstream) # 往文件写入pdf数据
outputstream.close() # 写入流

4、pdf增加水印的示例

注意:水印模板可以利用word文档写好文字,转为pdf即可

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import pypdf2

reader = pypdf2.pdffilereader(open('linux.pdf', 'rb')) # 增加水印的原文件

watermark = pypdf2.pdffilereader(open('水印模板.pdf', 'rb')) # 水印的模板

writer = pypdf2.pdffilewriter() # 写入pdf的对象

for i in range(reader.getnumpages()):
  page = reader.getpage(i)
  page.mergepage(watermark.getpage(0)) # 将原文件与水印模板合并
  writer.addpage(page) # 增加到写入对象中

outputstream = open('watermark-test-linux.pdf', 'wb') # 打开一个写入硬盘的文件对象
writer.write(outputstream) # 将合并好的数据,写入硬盘中
outputstream.close() # 关闭文件句柄

测试效果

Python PyPDF2模块安装使用解析

5、合并多个指定的pdf文件的示例

#!/usr/bin/env python
# -*- coding: utf-8 -*-

from pypdf2 import pdffilemerger

merger = pdffilemerger() # 创建一个合并的对象

input1 = open('01pdf.pdf', 'rb')
input2 = open('02pdf.pdf', 'rb')
input3 = open('03pdf.pdf', 'rb')

merger.append(fileobj=input1, pages=(0, 3)) # 合并文件1的0到3页
merger.merge(position=2, fileobj=input2, pages=(0, 1)) # 合并文件2的0到1页
merger.append(fileobj=input3) # 合并文件的所有页

output = open('document-output.pdf', 'wb') # 保存硬盘上
merger.write(output) # 写入到硬盘上
output.close() # 关闭文件句柄

6、批量合并指定目录的pdf文件的示例

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import pypdf2
import os
import glob

def get_all_pdf_files(path):
  """获取指定目录的所有pdf文件名"""
  all_pdfs = glob.glob('{0}/*.pdf'.format(path))
  all_pdfs.sort(key=str.lower) # 排序
  return all_pdfs

def main():
  path = os.getcwd()
  all_pdfs = get_all_pdf_files(path)
  if not all_pdfs:
    raise systemexit('没有可用的pdf类型文件')

  merger = pypdf2.pdffilemerger()

  first_obj = open(all_pdfs[0], 'rb') # 打开第一个pdf文件
  merger.append(first_obj) # 增加到合并的对象中

  file_objs = []
  for pdf in all_pdfs[1:]: # 读取所有的文件对象
    file_objs.append(open(pdf, 'rb'))

  for file_obj in file_objs:
    reader = pypdf2.pdffilereader(file_obj)
    merger.append(fileobj=file_obj, pages=(1, reader.getnumpages()))

  outputstream = open('merge-pdfs.pdf', 'wb')
  merger.write(outputstream)
  outputstream.close()
  for file_obj in file_objs: # 批量关闭文件句柄
    file_obj.close()

if __name__ == '__main__':
  main()

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持。