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

def zip_to_pdf(zip_path, output_folder=None):
    """
    将ZIP文件中的JPG图片合并为PDF
    
    参数:
        zip_path (str): ZIP文件路径
        output_folder (str, optional): 输出PDF的目录，默认为ZIP所在目录
    """
    # 检查ZIP文件是否存在
    if not os.path.exists(zip_path):
        raise FileNotFoundError(f"ZIP文件不存在: {zip_path}")
    
    # 设置输出目录
    if output_folder is None:
        output_folder = os.path.dirname(zip_path)
    
    # 获取ZIP文件名(不带扩展名)
    zip_name = os.path.splitext(os.path.basename(zip_path))[0]
    pdf_path = os.path.join(output_folder, f"{zip_name}.pdf")
    
    # 创建临时目录存放解压文件
    temp_dir = os.path.join(output_folder, f"temp_{zip_name}")
    os.makedirs(temp_dir, exist_ok=True)
    
    try:
        # 解压ZIP文件
        with zipfile.ZipFile(zip_path, 'r') as zip_ref:
            zip_ref.extractall(temp_dir)
        
        # 获取所有JPG文件
        jpg_files = glob.glob(os.path.join(temp_dir, "*.jpg")) + glob.glob(os.path.join(temp_dir, "*.jpeg"))
        
        if not jpg_files:
            raise ValueError("ZIP文件中没有找到JPG图片")
        
        # 按文件名排序
        jpg_files.sort()
        
        # 创建PDF
        pdf = FPDF(unit="pt")  # 使用点(point)作为单位
        
        for jpg_file in jpg_files:
            # 打开图片获取尺寸
            with Image.open(jpg_file) as img:
                width, height = img.size
                
                # 添加新页面并设置尺寸
                pdf.add_page(format=[width, height])
                pdf.image(jpg_file, x=0, y=0, w=width, h=height)
        
        # 保存PDF
        pdf.output(pdf_path)
        print(f"成功创建PDF: {pdf_path}")
        
    finally:
        # 清理临时文件
        import time
        time.sleep(1)  # 等待1秒确保文件被释放
        
        for root, dirs, files in os.walk(temp_dir, topdown=False):
            for name in files:
                file_path = os.path.join(root, name)
                try:
                    os.remove(file_path)
                except PermissionError:
                    print(f"无法删除文件 {file_path}，可能仍被占用")
            for name in dirs:
                os.rmdir(os.path.join(root, name))
        try:
            os.rmdir(temp_dir)
        except OSError:
            print(f"无法删除临时目录 {temp_dir}，可能仍有文件被占用")

if __name__ == "__main__":
    # 示例用法
    zip_file = input("请输入ZIP文件路径: ").strip('"\' ')  # 处理可能的引号
    zip_to_pdf(zip_file)