Python 代码:使用 mediainfo 统计视频文件大小并进行单位换算
以下 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, filesize = output.split('|')
filesize = int(filesize)
size_unit = ['B', 'KB', 'MB', 'GB', 'TB']
i = 0
while filesize > 1024 and i < len(size_unit)-1:
filesize /= 1024
i += 1
filesize_str = f'{filesize:.2f}{size_unit[i]}'
# 写入 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=filesize_str)
# 保存 Excel 文件
wb.save(excel_file)
print(f'视频信息已保存到 {excel_file}')
代码说明:
- 安装依赖库: 首先需要安装
openpyxl模块和mediainfo命令行工具。 - 定义变量: 定义视频文件夹路径
video_folder和输出 Excel 文件路径excel_file。 - 创建 Excel 文件: 使用
openpyxl库创建 Excel 文件并命名为Video Info。 - 设置表头: 设置 Excel 文件的表头,包括 '文件名', '格式', '分辨率', '码率', '帧率', '时长', '大小'。
- 获取视频文件列表: 使用
os.listdir()获取视频文件夹中的所有文件,并过滤出以 '.mp4' 结尾的视频文件。 - 遍历视频文件: 遍历所有视频文件,使用
mediainfo命令行工具获取视频信息,并将其写入 Excel 文件。 - 单位换算: 使用循环将文件大小单位换算为 KB、MB、GB 等。
- 保存 Excel 文件: 保存生成的 Excel 文件。
代码示例:
假设你的视频文件夹路径为 /home/115/up,则运行代码后,将在该目录下生成一个名为 video_info.xlsx 的 Excel 文件,包含所有视频文件的详细信息,包括文件大小和单位换算后的结果。
注意:
- 该代码仅支持 '.mp4' 格式的视频文件。
- 需要确保你的系统中已经安装了
mediainfo命令行工具。 - 可以根据实际情况调整代码中的视频文件夹路径和输出 Excel 文件路径。
其他说明:
mediainfo命令行工具可以获取视频文件的更多信息,例如音频信息、字幕信息等,可以根据需要修改代码中的--Inform参数来获取所需的信息。- 可以使用其他库或方法来进行文件大小的单位换算,例如
humanize库。 - 可以将代码封装成一个函数,方便重复使用。
希望以上代码和说明对你有所帮助!', 'url':
原文地址: https://www.cveoy.top/t/topic/mz1r 著作权归作者所有。请勿转载和采集!