Python给PDF文件添加书签

这段代码主要做了以下几件事:

  1. 定义一个generate_bookmarks函数,该函数从一个文本文件中读取书签信息,并将这些信息存储在一个列表中。每个书签是一个元组,包含三个元素:级别、标题和页码。

  2. 在generate_bookmarks函数中,首先打开书签文件并读取所有行。对于每一行,如果它不是空行,就提取出标题和页码。标题是通过分割行并去除前后的'.'字符得到的。页码是通过分割行并去除前后的'. ()'字符得到的。

  3. 对于每个标题,使用正则表达式检查它是否只包含中文字符。如果是,就将书签的级别设置为2。否则,书签的级别由'.'的数量决定。

  4. 使用unicodedata.normalize('NFKC', title)将标题中的全角字符转换为半角字符。

  5. 定义了一个add_bookmarks_to_pdf函数,该函数将书签添加到PDF文件中。首先,创建一个PdfWriter对象,并从输入的PDF文件中读取所有页面。然后,对于每个书签,根据其级别将其添加到PDF文件的大纲中。最后,将带有书签的PDF文件写入到输出路径。

  6. 最后,调用generate_bookmarks函数生成书签,然后调用add_bookmarks_to_pdf函数将书签添加到PDF文件中。

from PyPDF2 import PdfReader, PdfWriter
import re
import unicodedata

def generate_bookmarks(bookmark_file_path):
    bookmarks = []
    with open(bookmark_file_path, 'r', encoding='utf-8') as file:
        lines = file.readlines()
        for line in lines:
            if line.strip() == '':
                continue
            title = line.split('!')[0].strip('.').strip()
            title = unicodedata.normalize('NFKC', title) #将全角字符改成半角字符
            page_num_str = line.strip('. ()').split('(')[-1].strip(')\n')
            page_num = int(page_num_str) if page_num_str.isdigit() else 0
            if re.match("^[\u4e00-\u9fa5]*$", title):#全中文正则表达式
                level = 2
            else:
                level = line.count('.') + 1
            bookmark = (level,title,page_num)
            bookmarks.append(bookmark)
    return bookmarks

bookmarks = generate_bookmarks('c:\\Users\\user1\\Desktop\\bookmark.txt')

def add_bookmarks_to_pdf(input_pdf_path, output_pdf_path, bookmarks):
    pdf_writer = PdfWriter()
    pdf_reader = PdfReader(input_pdf_path)

    for page in range(len(pdf_reader.pages)):
        pdf_writer.add_page(pdf_reader.pages[page])

    parent_bookmark = None
    for level,title, pagenum in bookmarks:
        if level == 1:
            parent_bookmark = pdf_writer.add_outline_item(title, pagenum+15)#15是偏移量-1
        elif level == 2:
            pdf_writer.add_outline_item(title, pagenum+15, parent=parent_bookmark)

    with open(output_pdf_path, 'wb') as fh:
        pdf_writer.write(fh)


add_bookmarks_to_pdf('D:\\Private\\课本\\材料科学基础_张联盟.pdf', 'D:\\Private\\课本\\材料科学基础_张联盟_with_bookmarks.pdf', bookmarks)

书签文本文件

bookmark.txt

文本直接从PDF粘贴,书签列表获取方式因人而异

效果

书签效果

代码by cursor 调试by me

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 194,457评论 5 459
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 81,837评论 2 371
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 141,696评论 0 319
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 52,183评论 1 263
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 61,057评论 4 355
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 46,105评论 1 272
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 36,520评论 3 381
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 35,211评论 0 253
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 39,482评论 1 290
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 34,574评论 2 309
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 36,353评论 1 326
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 32,213评论 3 312
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 37,576评论 3 298
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 28,897评论 0 17
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,174评论 1 250
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 41,489评论 2 341
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 40,683评论 2 335

推荐阅读更多精彩内容