import os
import glob
from fpdf import FPDF
from PIL import Image

def images_to_pdf_fpdf(
    image_paths, 
    output_pdf_path, 
    paper_size="A4",  # 紙張大小（默認(rèn)A4）
    direction="縱向"   # 布局方向（默認(rèn)縱向，支持"橫向"）
):
    """
    將多張圖片合并為PDF，支持自定義紙張大小和方向（縱向/橫向）
    
    參數(shù):
        image_paths: 圖片路徑列表（支持通配符，如 '*.jpg'）
        output_pdf_path: 輸出PDF路徑
        paper_size: 紙張大?。J(rèn)'A4'，支持'A3'/'Letter'/'Legal'或自定義元組，如(200, 300)）
        direction: 布局方向（默認(rèn)'縱向'，可選'橫向'）
    """
    # 定義標(biāo)準(zhǔn)紙張尺寸（單位：毫米）
    paper_sizes = {
        "A4": (210, 297),
        "A3": (297, 420),
        "Letter": (215.9, 279.4),
        "Legal": (215.9, 355.6),
    }

    # 校驗(yàn)紙張大小參數(shù)
    if isinstance(paper_size, str):
        if paper_size.upper() in paper_sizes:
            width_mm, height_mm = paper_sizes[paper_size.upper()]
        else:
            raise ValueError(f"未知的紙張大小: {paper_size}。支持的值: {list(paper_sizes.keys())} 或自定義元組（如 (200, 300)）")
    elif isinstance(paper_size, (tuple, list)) and len(paper_size) == 2:
        width_mm, height_mm = paper_size
    else:
        raise ValueError("paper_size 必須是預(yù)定義名稱(chēng)（如 'A4'）或 (width_mm, height_mm) 元組")

    # 校驗(yàn)方向參數(shù)
    direction = direction.lower()
    if direction not in ["縱向", "橫向"]:
        raise ValueError("direction 必須是 '縱向' 或 '橫向'")

    # 根據(jù)方向調(diào)整寬高（橫向時(shí)交換寬高）
    if direction == "橫向":
        width_mm, height_mm = height_mm, width_mm

    # 創(chuàng)建 PDF 對(duì)象
    pdf = FPDF(unit="mm", format=(width_mm, height_mm))  # 直接傳入寬高

    for image_path in image_paths:
        try:
            # 打開(kāi)圖片并獲取尺寸（轉(zhuǎn)換為毫米）
            img = Image.open(image_path)
            img_width, img_height = img.size
            img_width_mm = img_width * 0.264583  # 像素 → 毫米（1像素 ≈ 0.264583毫米）
            img_height_mm = img_height * 0.264583

            # 計(jì)算縮放比例（適應(yīng)當(dāng)前頁(yè)面尺寸）
            scale_width = width_mm / img_width_mm
            scale_height = height_mm / img_height_mm
            scale = min(scale_width, scale_height)

            # 計(jì)算縮放后的圖片尺寸
            new_width_mm = img_width_mm * scale
            new_height_mm = img_height_mm * scale

            # 添加頁(yè)面并插入圖片（居中）
            pdf.add_page()
            x = (width_mm - new_width_mm) / 2
            y = (height_mm - new_height_mm) / 2
            pdf.image(image_path, x=x, y=y, w=new_width_mm, h=new_height_mm)

        except Exception as e:
            print(f"警告：圖片 {image_path} 處理失敗，跳過(guò)。錯(cuò)誤: {e}")

    # 保存 PDF
    pdf.output(output_pdf_path)
    print(f"PDF 已生成：{output_pdf_path}")

# 示例用法
if __name__ == "__main__":
    # 獲取所有JPG圖片（使用raw字符串避免轉(zhuǎn)義）
    image_paths = glob.glob(r'E:\Python\learning\圖片轉(zhuǎn)PDF\img\*流水.jpg')
    output_pdf_path = "1-3月銀行流水.pdf"

    # 調(diào)用方式1：默認(rèn)A4縱向
    images_to_pdf_fpdf(image_paths, output_pdf_path)

    # 調(diào)用方式2：A4橫向
    images_to_pdf_fpdf(image_paths, output_pdf_path, paper_size="A4", direction="橫向")

    # 調(diào)用方式3：自定義紙張大?。?00x300毫米）縱向
    images_to_pdf_fpdf(image_paths, output_pdf_path, paper_size=(200, 300), direction="縱向")