Python 代码优化:使用 os.path.getsize() 获取视频文件大小
可以使用 os.path.getsize() 函数获取文件大小,具体修改如下:
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%', video_file], stdout=subprocess.PIPE)
output = result.stdout.decode().strip()
# 解析输出结果
format, resolution, bitrate, framerate, duration = output.split('|')
# 获取文件大小
file_size = os.path.getsize(video_file)
# 写入 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=file_size)
# 保存 Excel 文件
wb.save(excel_file)
print(f'视频信息已保存到 {excel_file}')
该代码使用 os.path.getsize(video_file) 获取每个视频文件的大小,并将大小信息写入 Excel 文件的对应列。
原文地址: https://www.cveoy.top/t/topic/mz1e 著作权归作者所有。请勿转载和采集!