pip install PyPDF2
pip install PyMuPDF

import fitz # PyMuPDF庫
import os

def merge_pdfs(pdf_list, output_pdf_path):
    """
    批量合并多個PDF文件為一個PDF文件。
    這是Python處理PDF和PDF合并分割的核心功能。
    :param pdf_list: 要合并的PDF文件路徑列表
    :param output_pdf_path: 合并后PDF文件的輸出路徑
    """
    if not pdf_list:
        print("? 沒有指定任何PDF文件進(jìn)行合并。")
        return False
    
    os.makedirs(os.path.dirname(output_pdf_path), exist_ok=True)

    print(f"?? 正在合并 {len(pdf_list)} 個PDF文件到 '{os.path.basename(output_pdf_path)}'...")
    
    # 創(chuàng)建一個新的PDF文檔
    new_pdf = fitz.open() 
    
    try:
        for pdf_path in pdf_list:
            if not os.path.exists(pdf_path):
                print(f"?? 文件不存在，跳過：{os.path.basename(pdf_path)}")
                continue
            
            # 打開每個PDF文件，并將其所有頁面插入到新PDF中
            with fitz.open(pdf_path) as doc:
                new_pdf.insert_pdf(doc) # 插入PDF的所有頁面
                print(f"   ? 已合并：{os.path.basename(pdf_path)}")
        
        new_pdf.save(output_pdf_path) # 保存合并后的PDF
        print(f"? PDF合并成功！結(jié)果保存到：'{output_pdf_path}'")
        return True
    except Exception as e:
        print(f"? PDF合并失?。簕e}")
        return False
    finally:
        new_pdf.close() # 確保關(guān)閉PDF對象

if __name__ == "__main__":
    # 準(zhǔn)備測試PDF文件
    pdf1_path = os.path.expanduser("~/Desktop/report_part1.pdf")
    pdf2_path = os.path.expanduser("~/Desktop/report_part2.pdf")
    output_merged_pdf = os.path.expanduser("~/Desktop/merged_annual_report.pdf")

    # 簡單創(chuàng)建模擬PDF文件 (實(shí)際請用你的真實(shí)PDF)
    if not os.path.exists(pdf1_path):
        doc1 = fitz.open(); doc1.new_page().insert_text(fitz.Point(50,50), "這是第一部分"); doc1.save(pdf1_path); doc1.close()
    if not os.path.exists(pdf2_path):
        doc2 = fitz.open(); doc2.new_page().insert_text(fitz.Point(50,50), "這是第二部分"); doc2.save(pdf2_path); doc2.close()
    
    pdfs_to_merge = [pdf1_path, pdf2_path]
    merge_pdfs(pdfs_to_merge, output_merged_pdf)

import fitz # PyMuPDF
import os

def split_pdf_by_pages(input_pdf_path, output_folder, page_ranges=None):
    """
    將單個PDF文件按頁碼范圍分割為多個PDF文件。
    這是Python處理PDF和PDF合并分割的高級功能。
    :param input_pdf_path: 源PDF文件路徑
    :param output_folder: 分割后PDF文件的輸出文件夾
    :param page_ranges: 要分割的頁碼范圍列表，例如 [[0, 2], [5, 5]] 表示分割第1-3頁和第6頁
                        頁碼從0開始計(jì)數(shù)。如果為None，則每頁生成一個PDF。
    """
    if not os.path.exists(input_pdf_path):
        print(f"? PDF文件不存在：{input_pdf_path}")
        return False
    
    os.makedirs(output_folder, exist_ok=True)

    print(f"?? 正在分割PDF文件 '{os.path.basename(input_pdf_path)}'...")
    
    try:
        with fitz.open(input_pdf_path) as doc:
            if page_ranges is None: # 如果沒有指定范圍，則每頁分割一個PDF
                page_ranges = [[i, i] for i in range(doc.page_count)]

            for i, (start_page, end_page) in enumerate(page_ranges):
                if start_page < 0 or end_page >= doc.page_count or start_page > end_page:
                    print(f"?? 頁碼范圍無效，跳過：[{start_page}, {end_page}]")
                    continue
                
                # 創(chuàng)建新PDF來存儲分割的頁面
                new_pdf = fitz.open()
                new_pdf.insert_pdf(doc, from_page=start_page, to_page=end_page)
                
                output_filename = f"{os.path.splitext(os.path.basename(input_pdf_path))[0]}_part_{i+1}_pages_{start_page+1}-{end_page+1}.pdf"
                output_full_path = os.path.join(output_folder, output_filename)
                
                new_pdf.save(output_full_path)
                new_pdf.close()
                print(f"   ? 已分割：'{output_filename}' (頁碼: {start_page+1}-{end_page+1})")
        
        print(f"? PDF分割完成！結(jié)果保存到：'{output_folder}'")
        return True
    except Exception as e:
        print(f"? PDF分割失?。簕e}")
        return False

if __name__ == "__main__":
    # 準(zhǔn)備測試PDF文件 (多頁)
    long_report_path = os.path.expanduser("~/Desktop/long_report.pdf")
    # 簡單創(chuàng)建模擬PDF文件 (例如創(chuàng)建5頁的PDF)
    if not os.path.exists(long_report_path):
        doc = fitz.open()
        for i in range(5):
            page = doc.new_page(width=595, height=842) # A4
            page.insert_text(fitz.Point(50,50), f"這是第 {i+1} 頁")
        doc.save(long_report_path); doc.close()
    
    output_split_folder = os.path.expanduser("~/Desktop/分割后PDF")
    
    # 示例1：將每一頁分割為單獨(dú)的文件
    print("\n--- 示例1：按頁分割所有PDF ---")
    split_pdf_by_pages(long_report_path, os.path.join(output_split_folder, "ByPage"))

    # 示例2：按指定范圍分割 (例如，只取第2頁到第4頁，對應(yīng)索引1到3)
    print("\n--- 示例2：按頁碼范圍分割PDF (取第2到4頁) ---")
    split_pdf_by_pages(long_report_path, os.path.join(output_split_folder, "Range"), page_ranges=[[1, 3]]) # 頁碼索引是0-based，范圍是[start, end)

import fitz # PyMuPDF
from PIL import Image, ImageDraw, ImageFont # 用于文字渲染到圖片 (如果需要)，或處理Logo圖片
import os
import io # 用于內(nèi)存中的文件操作

# 定義一個中文字體路徑，確保你的系統(tǒng)中有，否則中文可能顯示為方塊
CHINESE_FONT_PATH = "C:/Windows/Fonts/simhei.ttf" # Windows黑體示例

def add_pdf_watermark(pdf_path, output_path, watermark_type="text",
                      text_content="", font_size=30, text_color=(0.5, 0.5, 0.5), # RGB (0.0-1.0)
                      image_watermark_path=None, image_opacity=0.3, image_size_ratio=0.1,
                      position="bottom_right", rotate_degree=0): # 新增旋轉(zhuǎn)參數(shù)
    """
    為PDF文件添加文字水印或圖片水印。
    這是PDF自動化和PDF加水印的核心功能。
    :param pdf_path: 源PDF文件路徑
    :param output_path: 輸出PDF文件路徑
    :param watermark_type: "text" (文字水印) 或 "image" (圖片水印)
    :param text_content: 文字水印內(nèi)容
    :param font_size: 文字水印字體大小
    :param text_color: 文字水印顏色 (RGB元組，范圍0.0-1.0)
    :param image_watermark_path: 圖片水印文件路徑
    :param image_opacity: 圖片水印透明度 (0.0-1.0)
    :param image_size_ratio: 圖片水印相對于頁面寬度的比例
    :param position: 水印位置 ("top_left", "top_right", "bottom_left", "bottom_right", "center")
    :param rotate_degree: 文字水印旋轉(zhuǎn)角度 (度)
    """
    if not os.path.exists(pdf_path): return print(f"? PDF文件不存在：{pdf_path}")
    os.makedirs(os.path.dirname(output_path), exist_ok=True)

    doc = fitz.open(pdf_path) # 打開PDF文檔
    print(f"?? 正在為 '{os.path.basename(pdf_path)}' 添加水印...")

    # 嘗試加載中文字體
    font_name_in_pdf = "helv" # 默認(rèn)內(nèi)置字體
    if pdf_font_path and os.path.exists(CHINESE_FONT_PATH):
        try:
            # register_font 返回字體對象的xref，可以直接用于insert_text
            font_xref = doc.register_font(fontfile=CHINESE_FONT_PATH, fontname="MyCustomFont")
            font_name_in_pdf = "MyCustomFont"
            print(f"   ?? 已注冊自定義字體 '{font_name_in_pdf}'。")
        except Exception as e:
            print(f"?? 注冊中文字體失敗：{e}。中文水印可能無法正確顯示。")
    
    # 加載圖片水?。ㄡ槍D片水?。?
    watermark_img_obj = None
    if watermark_type == "image":
        if not os.path.exists(image_watermark_path):
            print(f"? 水印圖片文件不存在：{image_watermark_path}")
            return False
        try:
            watermark_img_obj = Image.open(image_watermark_path).convert("RGBA")
            print(f"   ?? 已加載水印圖片：{os.path.basename(image_watermark_path)}")
        except Exception as e:
            print(f"? 無法加載水印圖片：{e}")
            return False

    for page_num in range(doc.page_count):
        page = doc.load_page(page_num)
        rect = page.rect # 獲取頁面矩形
        
        if watermark_type == "text":
            # 插入文字水印
            # 注意：fitz.insert_text直接處理透明度和顏色 (R,G,B浮點(diǎn)數(shù)0.0-1.0)
            # 計(jì)算文字尺寸和位置 (需要實(shí)際渲染才能準(zhǔn)確獲取，這里估算)
            # fitz.Font().text_bbox() 更精確，但需先實(shí)例化Font
            text_len_estimate = fitz.Font(fontname=font_name_in_pdf).text_bbox(text_content, fontsize=font_size)[2]
            text_height_estimate = font_size * 1.2 # 粗略估算高度

            x, y = 0, 0
            if position == "top_left": x, y = rect.x0 + 20, rect.y0 + 20
            elif position == "top_right": x, y = rect.x1 - text_len_estimate - 20, rect.y0 + 20
            elif position == "bottom_left": x, y = rect.x0 + 20, rect.y1 - text_height_estimate - 20
            elif position == "bottom_right": x, y = rect.x1 - text_len_estimate - 20, rect.y1 - text_height_estimate - 20
            elif position == "center": x, y = (rect.width - text_len_estimate) / 2, (rect.height - text_height_estimate) / 2
            
            # 使用 insert_text
            page.insert_text(
                (x, y),
                text_content,
                fontname=font_name_in_pdf,
                fontsize=font_size,
                color=text_color, # 顏色為RGB浮點(diǎn)元組 (0.0, 0.0, 0.0) 到 (1.0, 1.0, 1.0)
                # rotate=rotate_degree, # 旋轉(zhuǎn)在insert_text中不直接支持，需要transform矩陣
                overlay=True, # 確保水印在內(nèi)容之上
            )
            # 對于旋轉(zhuǎn)，需要更復(fù)雜的PyMuPDF變換矩陣，這里為簡潔不展開

        elif watermark_type == "image" and watermark_img_obj:
            # 插入圖片水印
            page_width, page_height = rect.width, rect.height

            # 調(diào)整水印圖片大小 (相對于頁面寬度比例)
            wm_width = int(page_width * image_size_ratio)
            wm_height = int(wm_width * watermark_img_obj.height / watermark_img_obj.width) # 保持比例
            
            # 確保水印不超出頁面且有最小尺寸
            if wm_width == 0 or wm_height == 0:
                print(f"?? 水印尺寸計(jì)算為零，跳過頁面 {page_num+1}。")
                continue
            
            resized_watermark_pil = watermark_img_obj.resize((wm_width, wm_height), Image.LANCZOS)
            
            # 調(diào)整水印圖片透明度
            alpha = resized_watermark_pil.split()[-1]
            alpha = Image.eval(alpha, lambda x: x * image_opacity)
            resized_watermark_pil.putalpha(alpha)
            
            # 將處理后的水印圖片轉(zhuǎn)為BytesIO，以便PyMuPDF插入
            img_byte_arr = io.BytesIO()
            resized_watermark_pil.save(img_byte_arr, format='PNG')
            img_byte_arr_value = img_byte_arr.getvalue()

            # 計(jì)算粘貼坐標(biāo)
            insert_x, insert_y = 0, 0
            if position == "top_left": insert_x, insert_y = 10, 10
            elif position == "top_right": insert_x, insert_y = page_width - wm_width - 10, 10
            elif position == "bottom_left": insert_x, insert_y = 10, page_height - wm_height - 10
            elif position == "bottom_right": insert_x, insert_y = page_width - wm_width - 10, page_height - wm_height - 10
            elif position == "center": insert_x, insert_y = (page_width - wm_width) // 2, (page_height - wm_height) // 2
            
            insert_rect = fitz.Rect(insert_x, insert_y, insert_x + wm_width, insert_y + wm_height)
            
            page.insert_image(insert_rect, stream=img_byte_arr_value, overlay=True)

        print(f"   - 頁面 {page_num+1} 已添加水印。")
    
    doc.save(output_path)
    print(f"? PDF水印添加完成！文件已保存到：'{output_path}'")
    return True
    finally:
        if 'doc' in locals(): doc.close()
        if 'watermark_img_obj' in locals(): watermark_img_obj.close() # 關(guān)閉水印圖片

if __name__ == "__main__":
    # 準(zhǔn)備測試PDF文件 (多頁)
    source_pdf_path = os.path.expanduser("~/Desktop/sample_report.pdf")
    # 簡單創(chuàng)建模擬PDF文件
    if not os.path.exists(source_pdf_path):
        doc = fitz.open(); doc.new_page().insert_text(fitz.Point(50,50), "這是測試報告第一頁\n請勿復(fù)制"); doc.new_page().insert_text(fitz.Point(50,50), "這是第二頁"); doc.save(source_pdf_path); doc.close()
    
    # 準(zhǔn)備Logo圖片 (如果測試圖片水印)
    logo_path = os.path.expanduser("~/Desktop/company_logo.png")
    if not os.path.exists(logo_path):
        Image.new('RGBA', (100, 50), color = (255, 0, 0, 128)).save(logo_path) # 模擬透明紅色Logo
        print(f"臨時Logo文件 '{os.path.basename(logo_path)}' 已創(chuàng)建。")

    output_text_wm_pdf = os.path.expanduser("~/Desktop/report_with_text_wm.pdf")
    output_image_wm_pdf = os.path.expanduser("~/Desktop/report_with_logo_wm.pdf")

    # 示例1：批量添加文字水印
    print("\n--- 示例1：添加文字水印 ---")
    add_pdf_watermark(
        source_pdf_path, 
        output_text_wm_pdf, 
        watermark_type="text", 
        text_content="【機(jī)密文件 嚴(yán)禁外傳】", 
        font_size=20, # 調(diào)整字體大小
        text_color=(1, 0, 0), # 紅色 (0.0-1.0)
        position="center" # 居中
    )

    # 示例2：批量添加圖片水印 (Logo)
    print("\n--- 示例2：添加圖片Logo水印 ---")
    add_pdf_watermark(
        source_pdf_path, 
        output_image_wm_pdf, 
        watermark_type="image", 
        image_watermark_path=logo_path, 
        image_opacity=0.3, # 更透明
        image_size_ratio=0.2 # 尺寸更大
    )

mport fitz # PyMuPDF
import os

# 確保中文字體路徑正確，用于PDF中文頁碼渲染
CHINESE_FONT_PATH = "C:/Windows/Fonts/simhei.ttf" # Windows黑體示例

def add_page_numbers_to_pdf(input_pdf_path, output_pdf_path, position="bottom_center", font_size=9, color=(0,0,0)):
    """
    為PDF文件的每一頁批量添加頁碼。
    這是Python處理PDF和PDF頁碼自動化的核心功能。
    :param input_pdf_path: 源PDF文件路徑
    :param output_pdf_path: 添加頁碼后PDF文件的輸出路徑
    :param position: 頁碼位置 ("top_left", "top_right", "bottom_left", "bottom_right", "center", "bottom_center", "top_center")
    :param font_size: 字體大小
    :param color: 顏色 (R, G, B)，范圍0.0-1.0
    """
    if not os.path.exists(input_pdf_path): return print(f"? PDF文件不存在：{input_pdf_path}")
    os.makedirs(os.path.dirname(output_pdf_path), exist_ok=True)

    print(f"?? 正在為 '{os.path.basename(input_pdf_path)}' 批量添加頁碼...")
    try:
        doc = fitz.open(input_pdf_path)
        
        # 嘗試加載中文字體 (用于中文頁碼，如“第X頁”)
        font_name_in_pdf = "helv" # 默認(rèn)內(nèi)置字體 (英文數(shù)字)
        if CHINESE_FONT_PATH and os.path.exists(CHINESE_FONT_PATH):
            try:
                doc.register_font(fontfile=CHINESE_FONT_PATH, fontname="MyCustomFont")
                font_name_in_pdf = "MyCustomFont"
                print(f"   ?? 已注冊自定義字體 '{font_name_in_pdf}'。")
            except Exception as e:
                print(f"?? 注冊中文字體失?。簕e}。中文頁碼可能無法正確顯示。")
        
        total_pages = doc.page_count
        for page_num in range(total_pages):
            page = doc.load_page(page_num)
            
            # 頁碼文本：例如 "Page 1 / 10" 或 "第1頁 / 共10頁" (根據(jù)是否成功加載中文顯示)
            page_text = f"Page {page_num + 1} / {total_pages}" 
            if font_name_in_pdf == "MyCustomFont": # 如果成功加載了中文字體
                 page_text = f"第 {page_num + 1} 頁 / 共 {total_pages} 頁"

            # 計(jì)算頁碼位置
            rect = page.rect # 頁面矩形
            # fitz.Font().text_bbox() 更精確計(jì)算文本寬度，這里使用估算
            # text_len_estimate = fitz.Font(fontname=font_name_in_pdf).text_bbox(page_text, fontsize=font_size)[2]
            text_len_estimate = font_size * len(page_text) * 0.6 # 粗略估算寬度
            text_height_estimate = font_size * 1.2 # 粗略估算高度

            x, y = 0, 0
            if position == "top_left": x, y = rect.x0 + 20, rect.y0 + 20
            elif position == "top_right": x, y = rect.x1 - text_len_estimate - 20, rect.y0 + 20
            elif position == "bottom_left": x, y = rect.x0 + 20, rect.y1 - text_height_estimate - 20
            elif position == "bottom_right": x, y = rect.x1 - text_len_estimate - 20, rect.y1 - text_height_estimate - 20
            elif position == "center": x, y = (rect.width - text_len_estimate) / 2, (rect.height - text_height_estimate) / 2
            elif position == "bottom_center": x, y = (rect.width - text_len_estimate) / 2, rect.y1 - text_height_estimate - 20
            elif position == "top_center": x, y = (rect.width - text_len_estimate) / 2, rect.y0 + 20
            
            page.insert_text(
                (x, y),
                page_text,
                fontname=font_name_in_pdf, # 使用注冊的字體名或內(nèi)置字體
                fontsize=font_size,
                color=color, # 顏色 (R, G, B) 范圍0.0-1.0
                overlay=True # 確保頁碼在內(nèi)容之上
            )
            print(f"   - 頁面 {page_num+1} 已添加頁碼。")
    
    doc.save(output_path)
    print(f"? PDF頁碼添加完成！文件已保存到：'{output_path}'")
    return True
    finally:
        if 'doc' in locals(): doc.close()

if __name__ == "__main__":
    input_pdf = os.path.expanduser("~/Desktop/long_report_no_pages.pdf")
    output_pdf = os.path.expanduser("~/Desktop/report_with_page_numbers.pdf")

    # 簡單創(chuàng)建模擬PDF文件 (多頁)
    if not os.path.exists(input_pdf):
        doc = fitz.open()
        for i in range(5):
            page = doc.new_page()
            page.insert_text(fitz.Point(50,50), f"這是第 {i+1} 頁的內(nèi)容。")
        doc.save(input_pdf); doc.close()

    # 示例1：添加右下角居中頁碼
    print("\n--- 示例1：添加居中底部頁碼 ---")
    add_page_numbers_to_pdf(
        input_pdf, 
        output_pdf, 
        position="bottom_center", 
        font_size=10, 
        color=(0, 0, 0) # 黑色
    )

from PyPDF2 import PdfReader, PdfWriter
import os
import shutil # 用于復(fù)制文件，如果未加密則直接復(fù)制

def encrypt_pdf_with_password(pdf_path, output_path, password):
    """
    使用密碼加密PDF文件。
    這是PDF自動化和PDF加密的核心功能。
    :param pdf_path: 源PDF文件路徑
    :param output_path: 加密后PDF文件的輸出路徑
    :param password: 加密密碼 (字符串)
    """
    if not os.path.exists(pdf_path): return print(f"? PDF文件不存在：{pdf_path}")
    os.makedirs(os.path.dirname(output_path), exist_ok=True)

    reader = PdfReader(pdf_path)
    writer = PdfWriter()

    print(f"?? 正在加密PDF文件 '{os.path.basename(pdf_path)}'...")

    for page in reader.pages:
        writer.add_page(page) # 將所有頁面添加到writer

    try:
        writer.encrypt(password) # 設(shè)置加密密碼
        with open(output_path, "wb") as output_pdf_file:
            writer.write(output_pdf_file)
        print(f"? PDF文件加密成功！保存到：'{output_path}'")
        return True
    except Exception as e:
        print(f"? PDF加密失?。簕e}")
        return False

def decrypt_pdf_with_password(encrypted_pdf_path, output_path, password):
    """
    使用密碼解密PDF文件。
    這是PDF自動化和PDF解密的核心功能。
    :param encrypted_pdf_path: 加密PDF文件路徑
    :param output_path: 解密后PDF文件的輸出路徑
    :param password: 解密密碼 (字符串)
    """
    if not os.path.exists(encrypted_pdf_path): return print(f"? 加密PDF文件不存在：{encrypted_pdf_path}")
    os.makedirs(os.path.dirname(output_path), exist_ok=True)

    reader = PdfReader(encrypted_pdf_path)
    
    if reader.is_encrypted:
        if not reader.decrypt(password): # 嘗試解密
            print(f"? 解密失?。?{os.path.basename(encrypted_pdf_path)}' 密碼不正確或文件已損壞。")
            return False
    else:
        print(f"?? 文件 '{os.path.basename(encrypted_pdf_path)}' 未加密，直接復(fù)制到輸出路徑。")
        shutil.copyfile(encrypted_pdf_path, output_path) # 未加密則直接復(fù)制
        print(f"文件已直接復(fù)制到：'{output_path}'")
        return True
     
    writer = PdfWriter()
    for page in reader.pages:
        writer.add_page(page)

    try:
        with open(output_path, "wb") as output_pdf_file:
            writer.write(output_pdf_file)
        print(f"? PDF文件解密成功！保存到：'{output_path}'")
        return True
    except Exception as e:
        print(f"? PDF解密失敗：{e}")
        return False

if __name__ == "__main__":
    # 準(zhǔn)備測試PDF文件
    source_pdf = os.path.expanduser("~/Desktop/secret_document.pdf")
    if not os.path.exists(source_pdf):
        doc = fitz.open(); doc.new_page().insert_text(fitz.Point(50,50), "這是機(jī)密文檔"); doc.save(source_pdf); doc.close()
    
    encrypted_pdf = os.path.expanduser("~/Desktop/secret_document_encrypted.pdf")
    decrypted_pdf = os.path.expanduser("~/Desktop/secret_document_decrypted.pdf")
    
    password = "MySecurePassword123"

    # 示例1：加密PDF
    print("\n--- 示例1：加密PDF文件 ---")
    encrypt_pdf_with_password(source_pdf, encrypted_pdf, password)

    # 示例2：解密PDF
    print("\n--- 示例2：解密PDF文件 ---")
    decrypt_pdf_with_password(encrypted_pdf, decrypted_pdf, password)
    
    # 示例3：嘗試用錯誤密碼解密
    print("\n--- 示例3：嘗試用錯誤密碼解密 ---")
    decrypt_pdf_with_password(encrypted_pdf, os.path.expanduser("~/Desktop/failed_decryption.pdf"), "WrongPassword")