Python实战：批量文件格式转换的自动化方案

在日常开发中，我们经常需要处理不同格式的文件，比如将图片从PNG转为JPEG，或将文本文件从TXT转为PDF。手动转换不仅耗时，还容易出错。本文将介绍如何使用Python实现批量文件格式转换的自动化方案，涵盖图片、文档和音频等常见格式，并提供完整的代码示例和操作步骤。## 环境准备在开始之前，确保你的系统已安装Python 3.6或更高版本。我们将使用几个流行的Python库来处理不同文件类型，通过pip安装它们。bashpip install Pillow pypdf2 python-docx pydub- Pillow: 用于图像处理，支持PNG、JPEG、BMP等格式。- PyPDF2: 用于PDF文件操作，如合并、拆分或转换为文本。- python-docx: 用于处理Word文档（.docx格式）。- pydub: 用于音频文件转换，如MP3转WAV。这些库覆盖了多种常见文件类型，你可以根据实际需求选择安装。## 图片格式转换假设我们有一个文件夹，里面包含多个PNG图片，需要批量转换为JPEG格式。Pillow库提供了简单易用的接口。### 操作步骤1. 导入Pillow库的Image模块。2. 使用os模块遍历指定文件夹中的所有PNG文件。3. 对每个文件，用Image.open()打开，然后调用save()方法保存为JPEG格式。4. 可选地，可以设置JPEG的质量参数（如压缩级别）。### 代码示例pythonimport osfrom PIL import Imagedef convert_png_to_jpeg(input_folder, output_folder, quality=85): “”" 将指定文件夹中的所有PNG图片转换为JPEG格式。 :param input_folder: 输入文件夹路径 :param output_folder: 输出文件夹路径 :param quality: JPEG质量，范围1-100，默认85 “”" if not os.path.exists(output_folder): os.makedirs(output_folder) for filename in os.listdir(input_folder): if filename.endswith(‘.png’): input_path = os.path.join(input_folder, filename) output_filename = filename.replace(‘.png’, ‘.jpg’) output_path = os.path.join(output_folder, output_filename) try: with Image.open(input_path) as img: # 转换为RGB模式，因为JPEG不支持透明度 if img.mode in (‘RGBA’, ‘LA’): background = Image.new(‘RGB’, img.size, (255, 255, 255)) background.paste(img, mask=img.split()[-1] if img.mode == ‘RGBA’ else None) img = background elif img.mode != ‘RGB’: img = img.convert(‘RGB’) img.save(output_path, ‘JPEG’, quality=quality) print(f"转换成功: {filename} -> {output_filename}“) except Exception as e: print(f"转换失败 {filename}: {e}”)# 使用示例input_folder = ‘./images’output_folder = ‘./converted_images’convert_png_to_jpeg(input_folder, output_folder)这段代码会处理./images文件夹中的所有PNG文件，并将转换后的JPEG文件保存到./converted_images。注意，PNG可能包含透明度，转换为JPEG时需要处理为RGB模式，以避免颜色失真。## 文档格式转换接下来，我们看看如何将文本文件转换为PDF。这里使用PyPDF2库，但注意它主要用于PDF操作，对于TXT转PDF，我们可以结合其他方法。一个简单的方式是先用python-docx创建Word文档，再转换为PDF（需要安装额外的工具如LibreOffice，这里不展开）。作为替代，我们可以用报告生成库如reportlab，但为了简化，这里展示一个基础示例：将多个TXT文件合并为一个PDF。### 操作步骤1. 读取TXT文件内容。2. 使用PyPDF2创建一个新的PDF文档。3. 将文本内容添加到PDF页面（这里简化，实际可能需要更复杂的布局）。### 代码示例pythonimport osfrom PyPDF2 import PdfWriter, PdfReaderfrom reportlab.lib.pagesizes import letterfrom reportlab.pdfgen import canvasfrom io import BytesIOdef txt_to_pdf(input_folder, output_pdf): “”" 将指定文件夹中的TXT文件内容合并到一个PDF中。 :param input_folder: 输入文件夹路径 :param output_pdf: 输出PDF文件路径 “”" writer = PdfWriter() for filename in os.listdir(input_folder): if filename.endswith(’.txt’): input_path = os.path.join(input_folder, filename) try: with open(input_path, ‘r’, encoding=‘utf-8’) as f: text = f.read() # 创建一个临时PDF页面 packet = BytesIO() can = canvas.Canvas(packet, pagesize=letter) can.drawString(100, 750, f"File: {filename}“) # 添加文件名 can.drawString(100, 730, text[:500]) # 添加文本内容（截断） can.save() packet.seek(0) new_pdf = PdfReader(packet) writer.add_page(new_pdf.pages[0]) print(f"添加文件: {filename}”) except Exception as e: print(f"处理失败 {filename}: {e}“) with open(output_pdf, ‘wb’) as output_file: writer.write(output_file) print(f"PDF已保存到: {output_pdf}”)# 使用示例input_folder = ‘./texts’output_pdf = ‘./output.pdf’txt_to_pdf(input_folder, output_pdf)这个示例使用了reportlab来生成PDF页面，然后通过PyPDF2合并。注意，这只是一个基础演示，实际应用中可能需要更精细的文本格式化。## 音频格式转换对于音频文件，pydub库提供了便捷的转换功能。例如，将MP3文件批量转换为WAV格式。### 操作步骤1. 导入pydub库。2. 遍历文件夹中的MP3文件。3. 使用AudioSegment.from_mp3()加载文件，然后调用export()保存为WAV格式。### 代码示例pythonimport osfrom pydub import AudioSegmentdef convert_mp3_to_wav(input_folder, output_folder): “”" 将指定文件夹中的所有MP3音频转换为WAV格式。 :param input_folder: 输入文件夹路径 :param output_folder: 输出文件夹路径 “”" if not os.path.exists(output_folder): os.makedirs(output_folder) for filename in os.listdir(input_folder): if filename.endswith(’.mp3’): input_path = os.path.join(input_folder, filename) output_filename = filename.replace(‘.mp3’, ‘.wav’) output_path = os.path.join(output_folder, output_filename) try: audio = AudioSegment.from_mp3(input_path) audio.export(output_path, format=‘wav’) print(f"转换成功: {filename} -> {output_filename}“) except Exception as e: print(f"转换失败 {filename}: {e}”)# 使用示例input_folder = ‘./audio’output_folder = ‘./converted_audio’convert_mp3_to_wav(input_folder, output_folder)pydub依赖于ffmpeg，确保系统中已安装ffmpeg。在Linux或macOS上，可以通过包管理器安装；在Windows上，可以从官网下载并添加到系统路径。## 扩展与优化以上示例展示了基础的文件格式转换，你可以根据需求进行扩展。例如：- 添加命令行参数，使脚本更灵活。- 支持更多文件格式，如Excel转CSV或视频格式转换（需额外库）。- 实现错误处理和日志记录，提高健壮性。- 使用多线程或异步处理，加速批量转换。一个简单的命令行界面示例：pythonimport argparseparser = argparse.ArgumentParser(description=‘批量文件格式转换工具’)parser.add_argument(’–input’, type=str, required=True, help=‘输入文件夹路径’)parser.add_argument(‘–output’, type=str, required=True, help=‘输出文件夹路径’)parser.add_argument(‘–format’, type=str, choices=[‘png2jpeg’, ‘mp32wav’], help=‘转换格式’)args = parser.parse_args()if args.format == ‘png2jpeg’: convert_png_to_jpeg(args.input, args.output)elif args.format == ‘mp32wav’: convert_mp3_to_wav(args.input, args.output)else: print(“请指定有效的转换格式”)通过这种方式，你可以轻松集成到自动化工作流中。## 总结本文介绍了使用Python实现批量文件格式转换的自动化方案，覆盖了图片、文档和音频等常见类型。代码示例基于Pillow、PyPDF2、python-docx和pydub库，提供了可复用的函数和操作步骤。这种方法能显著提高效率，减少手动错误，适用于数据处理、媒体管理等多种场景。你可以根据实际需求调整和扩展代码，构建更强大的转换工具。记住，处理文件时总是备份原始数据，以防转换过程中出现意外