用Python批量给扫描pdf添加书签

import fitz  # PyMuPDF:pip install pymupdf

def copy_bookmarks(source_pdf_path, target_pdf_path, output_pdf_path):
    # 打开源PDF和目标PDF文件
    source_pdf = fitz.open(source_pdf_path)
    target_pdf = fitz.open(target_pdf_path)
  
    # 从源PDF读取书签（大纲）
    bookmarks = source_pdf.get_toc(simple=False)
  
    # 将书签（大纲）添加到目标PDF
    target_pdf.set_toc(bookmarks)
  
    # 保存更改到新的PDF文件
    target_pdf.save(output_pdf_path)
  
    # 关闭PDF文件
    source_pdf.close()
    target_pdf.close()

# 源PDF路径（书签来源）
source_pdf_path = 'source.pdf'
# 目标PDF路径（需要添加书签的PDF）
target_pdf_path = 'target.pdf'
# 输出的PDF路径
output_pdf_path = 'output_with_bookmarks.pdf'

# 调用函数
copy_bookmarks(source_pdf_path, target_pdf_path, output_pdf_path)

print("书签导入完成。")

import fitz  # PyMuPDF:pip install pymupdf
def parse_bookmarks_from_txt(txt_path,offset):
    """
    解析txt文件中的书签信息。
    txt文件的书签信息格式为markdown多级列表，根据缩进来确定层级。
    """
    bookmarks = []
    with open(txt_path, 'r', encoding='utf-8') as file:
        for line in file:
            if line.strip():  # 跳过空行
                # 计算行前的空格数来确定层级
                # 假设每个层级的缩进是2个空格，可以根据实际情况调整
                indent = len(line) - len(line.lstrip())
                level = indent // 2 + 1  # 计算层级，假定每2个空格代表一级缩进
                # 移除星号和空格，然后分割标题和页码
                title, page = line.strip().removeprefix('*').strip().split('@')
                bookmarks.append([level, title.strip(), int(page)+offset])
    return bookmarks




def add_bookmarks_to_pdf(pdf_path, output_pdf_path,bookmarks):
    """
    将格式化后的书签信息添加到指定的PDF文件中。
    :param pdf_path: PDF文件的路径
    :param bookmarks: 要添加的书签列表，格式为[[level, title, page], ...]
    """
    # 打开PDF文件
    doc = fitz.open(pdf_path)
  
    doc.set_toc(bookmarks)
    # 保存更改到新的PDF文件中，以防覆盖原文件
    doc.save(output_pdf_path)
    doc.close()
    print(f"书签已添加到新的PDF文件中：{output_pdf_path}")


# 1.将书签从txt文件解析并转换为pymupdf格式
txt_path = 'bookmarks.txt'  # 假设书签信息存储在bookmarks.txt文件中
bookmarks = parse_bookmarks_from_txt(txt_path,offset=32)

# 2.添加书签到PDF
pdf_path = r"C:\Users\Achuan-2\Downloads\神经科学：探索脑（第四版）OCR.pdf" # 替换为目标PDF文件的路径
output_pdf_path = r"C:\Users\Achuan-2\Downloads\神经科学：探索脑（第四版）test.pdf"
add_bookmarks_to_pdf(pdf_path, output_pdf_path,bookmarks)