import osimport PyPDF2import pdfplumberfrom pdf2image import convert_from_pathfrom datetime import datetime# ===================== 基础配置 =====================# 生成文件的保存路径（默认当前目录）SAVE_PATH = "./PDF处理结果"# 创建保存目录if not os.path.exists(SAVE_PATH):    os.makedirs(SAVE_PATH)# ===================== PDF合并 =====================def merge_pdfs(pdf_files, output_name="合并后的PDF.pdf"):    """    合并多个PDF文件    :param pdf_files: PDF文件列表（按合并顺序）    :param output_name: 输出文件名    """    if not pdf_files:        print("❌ 没有选择要合并的PDF文件！")        return    pdf_merger = PyPDF2.PdfMerger()    # 逐个添加PDF文件    for pdf_file in pdf_files:        if pdf_file.endswith(".pdf") and os.path.exists(pdf_file):            pdf_merger.append(pdf_file)            print(f"✅ 已添加：{pdf_file}")        else:            print(f"❌ 文件无效，跳过：{pdf_file}")    # 保存合并后的文件    output_path = os.path.join(SAVE_PATH, output_name)    pdf_merger.write(output_path)    pdf_merger.close()    print(f"\n🎉 PDF合并完成！文件保存至：{output_path}")# ===================== PDF拆分 =====================def split_pdf(pdf_file, split_pages=None):    """    拆分PDF文件    :param pdf_file: 要拆分的PDF文件    :param split_pages: 拆分页数列表（如[5, 10]表示拆成1-5、6-10、11+页）    """    if not os.path.exists(pdf_file) or not pdf_file.endswith(".pdf"):        print("❌ 请选择有效的PDF文件！")        return    # 获取PDF总页数    with open(pdf_file, "rb") as f:        pdf_reader = PyPDF2.PdfReader(f)        total_pages = len(pdf_reader.pages)    if split_pages is None:        # 默认按单页拆分        split_pages = list(range(1, total_pages+1))    # 补充最后一页    if split_pages[-1] < total_pages:        split_pages.append(total_pages)    start_page = 0    file_count = 1    # 开始拆分    for end_page in split_pages:        end_page = min(end_page, total_pages)        pdf_writer = PyPDF2.PdfWriter()        # 添加指定页数的内容        for page_num in range(start_page, end_page):            pdf_writer.add_page(pdf_reader.pages[page_num])        # 保存拆分后的文件        output_name = f"{os.path.splitext(os.path.basename(pdf_file))[0]}_拆分{file_count}.pdf"        output_path = os.path.join(SAVE_PATH, output_name)        with open(output_path, "wb") as f:            pdf_writer.write(f)        print(f"✅ 拆分文件{file_count}：{start_page+1}-{end_page}页，保存至：{output_name}")        start_page = end_page        file_count += 1    print(f"\n🎉 PDF拆分完成！共生成{file_count-1}个文件，保存至：{SAVE_PATH}")# ===================== 提取PDF文字 =====================def extract_pdf_text(pdf_file, start_page=1, end_page=None):    """    提取PDF中的文字    :param pdf_file: 要提取的PDF文件    :param start_page: 起始页（默认第1页）    :param end_page: 结束页（默认全部）    """    if not os.path.exists(pdf_file) or not pdf_file.endswith(".pdf"):        print("❌ 请选择有效的PDF文件！")        return    with pdfplumber.open(pdf_file) as pdf:        total_pages = len(pdf.pages)        end_page = end_page if end_page else total_pages        end_page = min(end_page, total_pages)        # 提取文字        text_content = ""        for page_num in range(start_page-1, end_page):            page = pdf.pages[page_num]            text = page.extract_text()            if text:                text_content += f"\n===== 第{page_num+1}页 =====\n{text}"        # 保存文字内容        output_name = f"{os.path.splitext(os.path.basename(pdf_file))[0]}_提取文字.txt"        output_path = os.path.join(SAVE_PATH, output_name)        with open(output_path, "w", encoding="utf-8") as f:            f.write(text_content)        print(f"✅ 文字提取完成！共提取{start_page}-{end_page}页，保存至：{output_path}")# ===================== 提取PDF图片 =====================def extract_pdf_images(pdf_file, start_page=1, end_page=None):    """    提取PDF中的图片    :param pdf_file: 要提取的PDF文件    :param start_page: 起始页（默认第1页）    :param end_page: 结束页（默认全部）    """    if not os.path.exists(pdf_file) or not pdf_file.endswith(".pdf"):        print("❌ 请选择有效的PDF文件！")        return    # 创建图片保存子目录    img_save_path = os.path.join(SAVE_PATH, f"{os.path.splitext(os.path.basename(pdf_file))[0]}_提取图片")    if not os.path.exists(img_save_path):        os.makedirs(img_save_path)    with pdfplumber.open(pdf_file) as pdf:        total_pages = len(pdf.pages)        end_page = end_page if end_page else total_pages        end_page = min(end_page, total_pages)        img_count = 1        # 提取图片        for page_num in range(start_page-1, end_page):            page = pdf.pages[page_num]            for img in page.images:                # 保存图片                img_data = img["stream"].get_data()                img_ext = img["ext"] if "ext" in img else "png"                img_name = f"第{page_num+1}页_图片{img_count}.{img_ext}"                img_path = os.path.join(img_save_path, img_name)                with open(img_path, "wb") as f:                    f.write(img_data)                print(f"✅ 提取图片：{img_name}")                img_count += 1    print(f"\n🎉 图片提取完成！共提取{img_count-1}张图片，保存至：{img_save_path}")# ===================== PDF转图片 =====================def pdf_to_images(pdf_file, start_page=1, end_page=None, img_format="png"):    """    PDF转换为图片    :param pdf_file: 要转换的PDF文件    :param start_page: 起始页（默认第1页）    :param end_page: 结束页（默认全部）    :param img_format: 图片格式（png/jpg）    """    if not os.path.exists(pdf_file) or not pdf_file.endswith(".pdf"):        print("❌ 请选择有效的PDF文件！")        return    # 创建图片保存子目录    img_save_path = os.path.join(SAVE_PATH, f"{os.path.splitext(os.path.basename(pdf_file))[0]}_转图片")    if not os.path.exists(img_save_path):        os.makedirs(img_save_path)    # 获取PDF总页数    with open(pdf_file, "rb") as f:        pdf_reader = PyPDF2.PdfReader(f)        total_pages = len(pdf_reader.pages)    end_page = end_page if end_page else total_pages    end_page = min(end_page, total_pages)    # 转换图片    pages = convert_from_path(        pdf_file,        first_page=start_page,        last_page=end_page,        fmt=img_format    )    # 保存图片    for i, page in enumerate(pages):        page_num = start_page + i        img_name = f"第{page_num}页.{img_format}"        img_path = os.path.join(img_save_path, img_name)        page.save(img_path, img_format.upper())        print(f"✅ 转换图片：{img_name}")    print(f"\n🎉 PDF转图片完成！共转换{len(pages)}页，保存至：{img_save_path}")# ===================== 主菜单 =====================def main():    """主程序菜单"""    print("="*60)    print("📄 Python PDF批量处理工具 📄")    print("="*60)    while True:        print("\n请选择要执行的操作：")        print("1. 合并多个PDF文件")        print("2. 拆分PDF文件")        print("3. 提取PDF中的文字")        print("4. 提取PDF中的图片")        print("5. PDF转换为图片")        print("6. 退出程序")        choice = input("\n输入操作编号（1-6）：")        # 1. 合并PDF        if choice == "1":            print("\n===== PDF合并 =====")            pdf_dir = input("请输入PDF文件所在目录（默认当前目录，直接回车）：") or "./"            pdf_files = [f for f in os.listdir(pdf_dir) if f.endswith(".pdf")]            if not pdf_files:                print("❌ 该目录下没有PDF文件！")                continue            print("📋 目录下的PDF文件：")            for i, f in enumerate(pdf_files):                print(f"{i+1}. {f}")            # 选择要合并的文件（默认全部）            select = input("请选择要合并的文件序号（如1,2,3，默认全部）：")            if select:                selected_index = [int(i)-1 for i in select.split(",") if i.isdigit()]                selected_files = [os.path.join(pdf_dir, pdf_files[i]) for i in selected_index if 0<=i<len(pdf_files)]            else:                selected_files = [os.path.join(pdf_dir, f) for f in pdf_files]            output_name = input("请输入合并后的文件名（默认：合并后的PDF.pdf）：") or "合并后的PDF.pdf"            merge_pdfs(selected_files, output_name)        # 2. 拆分PDF        elif choice == "2":            print("\n===== PDF拆分 =====")            pdf_file = input("请输入要拆分的PDF文件路径（如：test.pdf）：")            if not os.path.exists(pdf_file):                print("❌ 文件不存在！")                continue            split_input = input("请输入拆分页数（如5,10表示拆成1-5、6-10、11+页，直接回车按单页拆分）：")            if split_input:                split_pages = [int(p) for p in split_input.split(",") if p.isdigit()]                split_pages.sort()            else:                split_pages = None            split_pdf(pdf_file, split_pages)        # 3. 提取文字        elif choice == "3":            print("\n===== 提取PDF文字 =====")            pdf_file = input("请输入要提取文字的PDF文件路径：")            if not os.path.exists(pdf_file):                print("❌ 文件不存在！")                continue            start_page = input("请输入起始页（默认1）：") or "1"            end_page = input("请输入结束页（默认全部）：")            extract_pdf_text(                pdf_file,                start_page=int(start_page) if start_page.isdigit() else 1,                end_page=int(end_page) if end_page.isdigit() else None            )        # 4. 提取图片        elif choice == "4":            print("\n===== 提取PDF图片 =====")            pdf_file = input("请输入要提取图片的PDF文件路径：")            if not os.path.exists(pdf_file):                print("❌ 文件不存在！")                continue            start_page = input("请输入起始页（默认1）：") or "1"            end_page = input("请输入结束页（默认全部）：")            extract_pdf_images(                pdf_file,                start_page=int(start_page) if start_page.isdigit() else 1,                end_page=int(end_page) if end_page.isdigit() else None            )        # 5. PDF转图片        elif choice == "5":            print("\n===== PDF转图片 =====")            pdf_file = input("请输入要转换的PDF文件路径：")            if not os.path.exists(pdf_file):                print("❌ 文件不存在！")                continue            start_page = input("请输入起始页（默认1）：") or "1"            end_page = input("请输入结束页（默认全部）：")            img_format = input("请输入图片格式（png/jpg，默认png）：") or "png"            pdf_to_images(                pdf_file,                start_page=int(start_page) if start_page.isdigit() else 1,                end_page=int(end_page) if end_page.isdigit() else None,                img_format=img_format            )        # 6. 退出        elif choice == "6":            print("\n👋 感谢使用，再见！")            break        # 输入错误        else:            print("❌ 输入错误，请输入1-6之间的数字！")if __name__ == "__main__":    main()

python pdf_tool.py