Python代码统计视频大小并写入Excel表格

本文介绍了如何使用Python代码，结合mediainfo命令行工具，统计视频文件的大小信息，并将其写入Excel表格。

代码示例

import os
import subprocess
import openpyxl

# 安装openpyxl模块
# pip install openpyxl

# 安装mediainfo命令行工具
# sudo apt-get install mediainfo

# 需要处理的视频文件夹
video_folder = '/home/115/up'

# 输出Excel文件名
excel_file = '/home/115/up/video_info.xlsx'

# 创建Excel文件
wb = openpyxl.Workbook()
ws = wb.active
ws.title = 'Video Info'

# 设置表头
headers = ['文件名', '格式', '分辨率', '码率', '帧率', '时长', '大小']
for col, header in enumerate(headers, start=1):
    ws.cell(row=1, column=col, value=header)

# 获取视频文件列表
video_files = [os.path.join(video_folder, f) for f in os.listdir(video_folder) if f.endswith('.mp4')]

# 遍历所有视频文件，获取视频信息并写入Excel文件
for row, video_file in enumerate(video_files, start=2):
    # 获取文件名
    file_name = os.path.basename(video_file)

    # 使用mediainfo获取视频信息
    result = subprocess.run(['mediainfo', '--Inform=Video;%Format%|%Width%x%Height%|%BitRate/String%|%FrameRate%|%Duration/String3%|%FileSize%', video_file], stdout=subprocess.PIPE)
    output = result.stdout.decode().strip()

    # 解析输出结果
    format, resolution, bitrate, framerate, duration, size = output.split('|')

    # 写入Excel文件
    ws.cell(row=row, column=1, value=file_name)
    ws.cell(row=row, column=2, value=format)
    ws.cell(row=row, column=3, value=resolution)
    ws.cell(row=row, column=4, value=bitrate)
    ws.cell(row=row, column=5, value=framerate)
    ws.cell(row=row, column=6, value=duration)
    ws.cell(row=row, column=7, value=size)

# 保存Excel文件
wb.save(excel_file)
print(f'视频信息已保存到 {excel_file}')

代码说明

导入必要的库：os、subprocess、openpyxl。
安装依赖库：使用pip install openpyxl安装openpyxl库，使用sudo apt-get install mediainfo安装mediainfo命令行工具。
设置视频文件夹路径和输出Excel文件名。
创建一个新的Excel文件，并设置工作表名称。
设置Excel表格的表头信息。
获取视频文件列表。
遍历所有视频文件，使用mediainfo命令获取视频信息，包括文件名、格式、分辨率、码率、帧率、时长、大小。
解析输出结果，并将信息写入Excel表格。
保存Excel文件。

总结

通过以上代码示例，可以方便地统计视频文件的大小信息，并将其写入Excel表格。该方法可以帮助用户更好地管理和分析视频文件。