分享

使用pdfrw库对pdf文件进行读写操作

 大邓的Python 2021-02-23

大家好,前几天分享了一篇读取pdf文章,但是依然无法少部分的pdf进行操作。今天看Python Weekly最新一期推送的文章中《Creating and Manipulate PDFs with pdfrw》,发现该库对pdf操作性能极好,且api更简洁,更适合我这样的小白使用。所以今天我就将该文章翻译一下,做成notebook分享给大家。

在本篇文章中,我们会学到以下内容:

  • 从pdf文件中提取特定的信息

  • 分割pdf文件

  • 合并pdf文件

  • 对pdf中的内容进行旋转

更多功能请查看pdfrw文档

https://github.com/pmaupin/pdfrw#examples

1. 安装

首先我们安装pdfrw库,在终端中输入命令

pip3 install pdfrw

如果你使用的jupyter notebook运行本教程notebook出错提示你没有该库,你需要进行以下操作。

在终端中输入命令

conda install -c nsidc pdfrw

可能网速比较慢,大概1-5min,终端会出现Proceed[y/n],你需要输入y回车

2. 从PDF中提取信息

pdfrw库会从pdf文件中提取并返回pdf文件相关信息,一码胜千言,直接上代码看运行结果可能比我说更好。这里我们使用了w9.pdf文件作为测试文件。

from pdfrw import PdfReader, PdfWriter


def get_pdf_info(path):
   pdf = PdfReader(path)

   #PdfReader的关键词keys
   print(pdf.keys())
   print(pdf.Info)
   print(pdf.Root.keys())

   #pdf文件页数
   print('PDF has {} pages'.format(len(pdf.pages)))


get_pdf_info('w9.pdf')
['/Root', '/Info', '/ID', '/Size']
{'/Author': '(SE:W:CAR:MP)', '/CreationDate': "(D:20171109144422-05'00')", '/Creator': '(Adobe LiveCycle Designer ES 9.0)', '/Keywords': '(Fillable)', '/ModDate': "(D:20171109144521-05'00')", '/Producer': '(Adobe LiveCycle Designer ES 9.0)', '/SPDF': '(1112)', '/Subject': '(Request for Taxpayer Identification Number and Certification)', '/Title': '(Form W-9 \\(Rev. November 2017\\))'}
['/AcroForm', '/Extensions', '/MarkInfo', '/Metadata', '/Names', '/Pages', '/StructTreeRoot', '/Type', '/Perms']
PDF has 6 pages

呈现方式可能不太友好,继续看下面的内容,pdfrw是不会让你失望的。

3. 分割pdf文件

我们可以使用pdfrw库将pdf文件分割为多份文件。比如,你有可能只想要pdf书籍中的封面,或者你想将这本书按照章节分割为多个文件。这就是pdfrw吸引大邓分享给你的初衷,真的是听强悍的一个pdf操作库。

split函数:

  • path: 待分割的pdf文件路径

  • page:待分割pdf文件第page页

  • output:第page页保存的新pdf文件路径

这里我使用一个 Introduction to Text Mining and Natural Language Processing.pdf 作为例子,将其第一页和第二页分割并保存为第一页.pdf、第二页.pdf两个pdf文件。

def split(path, page, output):
   pdf_obj = PdfReader(path)
   total_pages = len(pdf_obj.pages)

   writer = PdfWriter()
   if page <= total_pages:
       writer.addpage(pdf_obj.pages[page])
       writer.write(output)


path = 'Introduction to Text Mining and Natural Language Processing.pdf'
pages = [1, 2]
outputs = ['第一页.pdf', '第二页.pdf']
for page,output in zip(pages, outputs):
   split(path, page, output)

具体效果大家运行后看看文件夹中新生成的第一页.pdf、第二页.pdfIntroduction to Text Mining and Natural Language Processing.pdf做对比,完美的运行实现了我们的分割实验。

4. 合并多个pdf文件

pdfrw可以很容易的将多个pdf文件进行合并,Let's go

在上面的例子中我们创建了concatenate函数,

  • paths参数为待合并的多个pdf文件路径

  • output为合并后pdf文件路径

from pdfrw import PdfReader, PdfWriter, IndirectPdfDict

def concatenate(paths, output):
   writer = PdfWriter()

   for path in paths:
       reader = PdfReader(path)
       writer.addpages(reader.pages)

   #我们给新的pdf文件加一些看不到的信息。
   #Title、author这些词你可以换成你喜欢的,不会影响程序运行
   #这里我们将这些信息隐藏到新的pdf文件中。
   writer.trailer.Info = IndirectPdfDict(
       Title = '合并pdf文件',
       Author = '生成者:大邓',
       Subject= '多个PDF文件合并',
       HelloPdfrw = '真好用',
       PythonIsFun ='YES')

   writer.write(output)

paths = ['第一页.pdf', '第二页.pdf']
concatenate(paths, '合并后的文件.pdf')

我们打开合并后的文件.pdf后,发现确实是对 第一页.pdf、第二页.pdf进行的合并。而且也看不到

Title = '合并pdf文件',
Author = '生成者:大邓',
Subject= '多个PDF文件合并',
HelloPdfrw = '真好用',
PythonIsFun ='YES'

这些信息。

那我们使用get_pdf_info(path)函数查看下合并后的文件.pdf的信息

get_pdf_info('合并后的文件.pdf')
['/Info', '/Root', '/Size']
{'/Author': '<FEFF751F62108005FF1A59279093>', '/HelloPdfrw': '<FEFF771F597D7528>', '/PythonIsFun': '(YES)', '/Subject': '<FEFF591A4E2A00500044004665874EF654085E76>', '/Title': '<FEFF54085E7600700064006665874EF6>'}
['/Pages', '/Type']
PDF has 2 pages

我们发现上面多了

Author/HelloPdfrw/HelloPdfrw/PythonIsFun,

这些关键词都是我自己随便写的,居然都隐藏近新生产的文件了。

这个方法非常棒,比如我以后分享的内容如果以pdf文件保存,我可以将作者信息隐藏到pdf文件中。如果以后谁盗版我的内容,我可以读取这个pdf文件隐藏的信息证明这是我写的。

5. 旋转

有时候你见到的pdf文件某页可能是是旋转90度后的,需要你旋转屏幕才能阅读,那么我们就需要对该pdf文件进行选择。这里我没有这么奇怪的pdf文件,那么我们就对'Introduction to Text Mining and Natural Language Processing.pdf' 文件的奇数页面进行旋转并存储到新的奇数页内容(顺时针旋转90度).pdf,生成奇怪的文件,来试试旋转功能。

from pdfrw import PdfReader, PdfWriter, IndirectPdfDict

def rotate_odd(path, output):

   reader = PdfReader(path)
   writer = PdfWriter()
   pages = reader.pages

   for page in range(len(pages)):
       if page % 2:
           pages[page].Rotate = 90
           writer.addpage(pages[page])

   writer.write(output)

path = 'Introduction to Text Mining and Natural Language Processing.pdf'
output = '奇数页内容(顺时针旋转90度).pdf'
rotate_odd(path, output)

    转藏 分享 献花(0

    0条评论

    发表

    请遵守用户 评论公约

    类似文章 更多