Python轻松实现PDF页面的自动化旋转

python -m venv pdf_rotate_env
# Windows
.\pdf_rotate_env\Scripts\activate
# macOS/Linux
source pdf_rotate_env/bin/activate

pip install spire.pdf

from spire.pdf.common import *
from spire.pdf import *

def rotate_pdf_page(input_pdf_path, output_pdf_path, page_index, rotation_angle):
    """
    旋转PDF文档中指定页面的函数。

    Args:
        input_pdf_path (str): 输入PDF文件的路径。
        output_pdf_path (str): 输出PDF文件的路径。
        page_index (int): 要旋转的页面索引（从0开始）。
        rotation_angle (int): 旋转角度，可选值为 90, 180, 270。
    """
    # 创建一个PdfDocument对象
    document = PdfDocument()

    try:
        # 加载现有的PDF文档
        document.LoadFromFile(input_pdf_path)

        # 获取指定页面
        # 注意：页面索引从0开始，所以 page_index 对应 Pages[page_index]
        if page_index < 0 or page_index >= document.Pages.Count:
            print(f"错误：页面索引 {page_index} 超出范围。PDF共有 {document.Pages.Count} 页。")
            return

        page = document.Pages[page_index]

        # 设置旋转角度
        # PdfPageRotateAngle是一个枚举类型，提供预定义的旋转角度
        if rotation_angle == 90:
            page.Rotation = PdfPageRotateAngle.RotateAngle90
        elif rotation_angle == 180:
            page.Rotation = PdfPageRotateAngle.RotateAngle180
        elif rotation_angle == 270:
            page.Rotation = PdfPageRotateAngle.RotateAngle270
        else:
            print(f"警告：不支持的旋转角度 {rotation_angle}。仅支持 90, 180, 270。")
            return

        # 保存修改后的PDF文档
        document.SaveToFile(output_pdf_path)
        print(f"成功将 '{input_pdf_path}' 的第 {page_index + 1} 页旋转 {rotation_angle} 度，并保存为 '{output_pdf_path}'。")

    except Exception as e:
        print(f"处理PDF时发生错误：{e}")
    finally:
        # 关闭文档以释放资源
        document.Close()

# --- 示例用法 ---
if __name__ == "__main__":
    input_pdf = "sample.pdf"  # 确保此路径下存在一个PDF文件
    output_pdf_90 = "rotated_page_90.pdf"
    output_pdf_180 = "rotated_page_180.pdf"
    output_pdf_270 = "rotated_page_270.pdf"
    
    # 假设我们要旋转PDF的第1页（索引为0）
    page_to_rotate = 0 

    # 旋转第1页90度
    rotate_pdf_page(input_pdf, output_pdf_90, page_to_rotate, 90)

    # 旋转第1页180度
    rotate_pdf_page(input_pdf, output_pdf_180, page_to_rotate, 180)

    # 旋转第1页270度
    rotate_pdf_page(input_pdf, output_pdf_270, page_to_rotate, 270)

    # 批量旋转所有页面（示例）
    # 假设有一个PDF有3页，我们想全部旋转90度
    all_pages_output = "all_pages_rotated_90.pdf"
    document_all = PdfDocument()
    try:
        document_all.LoadFromFile(input_pdf)
        for i in range(document_all.Pages.Count):
            document_all.Pages[i].Rotation = PdfPageRotateAngle.RotateAngle90
        document_all.SaveToFile(all_pages_output)
        print(f"成功将 '{input_pdf}' 的所有页面旋转 90 度，并保存为 '{all_pages_output}'。")
    except Exception as e:
        print(f"批量处理时发生错误：{e}")
    finally:
        document_all.Close()

import os

def batch_rotate_pdfs(input_folder, output_folder, page_index, rotation_angle):
    if not os.path.exists(output_folder):
        os.makedirs(output_folder)

    for filename in os.listdir(input_folder):
        if filename.lower().endswith(".pdf"):
            input_path = os.path.join(input_folder, filename)
            output_path = os.path.join(output_folder, f"rotated_{filename}")
            print(f"正在处理文件: {filename}")
            rotate_pdf_page(input_path, output_path, page_index, rotation_angle)

# 示例：批量旋转 'input_pdfs' 文件夹中所有PDF的第一页90度，并保存到 'output_pdfs'
# batch_rotate_pdfs("input_pdfs", "output_pdfs", 0, 90)