Python实战：批量转换文件格式的自动化方案
文件格式转换是日常开发中常见的需求，比如将图片从PNG转为JPEG、将文档从PDF转为TXT，或者处理音频视频文件。手动转换不仅耗时，还容易出错。本文将介绍如何使用Python构建一个自动化文件格式转换工具，支持批量处理，并提供可扩展的代码示例。
环境准备
确保已安装Python 3.6或更高版本。我们将使用几个第三方库来简化转换过程。通过pip安装所需依赖：
bash
pip install Pillow PyPDF2 moviepy
- Pillow：用于图像格式转换（如PNG到JPEG）。
- PyPDF2：用于PDF文本提取（转换为TXT）。
- moviepy：用于视频格式转换（如MP4到GIF）。
这些库覆盖了常见文件类型，你可以根据需要添加其他库，如pydub用于音频转换。
核心代码实现
我们将创建一个Python脚本，定义函数来处理不同文件类型的转换，并实现批量遍历功能。代码结构清晰，易于扩展。
图像格式转换
使用Pillow库转换图像文件。以下函数将指定目录下的PNG文件批量转换为JPEG格式。
python
from PIL import Image
import os
def convert_images(input_dir, output_dir, input_ext=‘.png’, output_ext=‘.jpg’):
“”"
批量转换图像格式。
:param input_dir: 输入目录路径
:param output_dir: 输出目录路径
:param input_ext: 输入文件扩展名
:param output_ext: 输出文件扩展名
“”"
os.makedirs(output_dir, exist_ok=True)
for filename in os.listdir(input_dir):
if filename.endswith(input_ext):
input_path = os.path.join(input_dir, filename)
output_filename = filename.replace(input_ext, output_ext)
output_path = os.path.join(output_dir, output_filename)
try:
with Image.open(input_path) as img:
img.convert('RGB').save(output_path)
print(f"转换成功: {filename} -> {output_filename}")
except Exception as e:
print(f"转换失败 {filename}: {e}")
示例调用
convert_images(‘./images’, ‘./converted_images’, ‘.png’, ‘.jpg’)
PDF转TXT
使用PyPDF2提取PDF文本并保存为TXT文件。注意，此方法适用于文本型PDF，扫描版PDF需要OCR处理。
python
import PyPDF2
def pdf_to_txt(input_path, output_path):
“”"
将PDF文件转换为TXT文本。
:param input_path: PDF文件路径
:param output_path: 输出TXT文件路径
“”"
try:
with open(input_path, ‘rb’) as pdf_file:
reader = PyPDF2.PdfReader(pdf_file)
text = ‘’
for page in reader.pages:
text += page.extract_text() + ‘\\n’
with open(output_path, 'w', encoding='utf-8') as txt_file:
txt_file.write(text)
print(f"转换成功: {input_path} -> {output_path}")
except Exception as e:
print(f"转换失败 {input_path}: {e}")
批量处理示例
import os
def batch_pdf_to_txt(input_dir, output_dir):
os.makedirs(output_dir, exist_ok=True)
for filename in os.listdir(input_dir):
if filename.endswith(‘.pdf’):
input_path = os.path.join(input_dir, filename)
output_filename = filename.replace(‘.pdf’, ‘.txt’)
output_path = os.path.join(output_dir, output_filename)
pdf_to_txt(input_path, output_path)
batch_pdf_to_txt(‘./pdfs’, ‘./texts’)
视频格式转换
使用moviepy库将视频文件转换为GIF动画。此库支持多种视频格式，如MP4、AVI等。
python
from moviepy.editor import VideoFileClip
def video_to_gif(input_path, output_path, fps=10):
“”"
将视频文件转换为GIF。
:param input_path: 输入视频文件路径
:param output_path: 输出GIF文件路径
:param fps: 帧率，控制GIF流畅度
“”"
try:
clip = VideoFileClip(input_path)
clip.write_gif(output_path, fps=fps)
print(f"转换成功: {input_path} -> {output_path}“)
except Exception as e:
print(f"转换失败 {input_path}: {e}”)
批量处理示例
import os
def batch_video_to_gif(input_dir, output_dir, fps=10):
os.makedirs(output_dir, exist_ok=True)
for filename in os.listdir(input_dir):
if filename.endswith(‘.mp4’):
input_path = os.path.join(input_dir, filename)
output_filename = filename.replace(‘.mp4’, ‘.gif’)
output_path = os.path.join(output_dir, output_filename)
video_to_gif(input_path, output_path, fps)
batch_video_to_gif(‘./videos’, ‘./gifs’, fps=5)
集成与扩展
将上述函数整合到一个主脚本中，支持命令行参数，实现更灵活的批量转换。以下是一个简单示例，使用argparse库解析参数。
python
import argparse
import sys
def main():
parser = argparse.ArgumentParser(description=‘批量文件格式转换工具’)
parser.add_argument(‘–type’, required=True, choices=[‘image’, ‘pdf’, ‘video’], help=‘文件类型’)
parser.add_argument(‘–input’, required=True, help=‘输入目录或文件路径’)
parser.add_argument(‘–output’, required=True, help=‘输出目录路径’)
parser.add_argument(‘–input_ext’, default=‘.png’, help=‘输入文件扩展名（仅图像类型）’)
parser.add_argument(‘–output_ext’, default=‘.jpg’, help=‘输出文件扩展名（仅图像类型）’)
parser.add_argument(‘–fps’, type=int, default=10, help=‘视频转换的帧率（仅视频类型）’)
args = parser.parse_args()
if args.type == 'image':
convert_images(args.input, args.output, args.input_ext, args.output_ext)
elif args.type == 'pdf':
batch_pdf_to_txt(args.input, args.output)
elif args.type == 'video':
batch_video_to_gif(args.input, args.output, args.fps)
else:
print("不支持的文件类型")
sys.exit(1)
if name == ‘main’:
main()
运行示例：
bash
python converter.py –type image –input ./images –output ./converted –input_ext .png –output_ext .jpg
注意事项与优化建议
- 错误处理：代码中已包含基本异常捕获，建议根据实际需求添加更详细的日志记录。
- 性能优化：对于大型文件或批量处理，考虑使用多线程或异步处理以提高效率。
- 格式支持：本文示例覆盖了常见格式，你可以通过添加新库（如pydub用于音频）扩展功能。
- 资源管理：确保及时关闭文件句柄和释放内存，特别是在处理视频等大文件时。
总结
通过Python和第三方库，我们可以轻松构建一个自动化文件格式转换工具。本文提供了图像、PDF和视频转换的完整代码示例，并展示了如何集成命令行界面。你可以基于此框架，根据项目需求定制和扩展功能，实现高效的文件处理工作流。
完整代码已上传至GitHub仓库，欢迎下载和贡献改进：https://github.com/example/file-converter-tool（示例链接，实际使用时请替换为你的仓库）。
