Python 代码优化：获取视频信息并取整时长写入 Excel

本文提供了一段 Python 代码，用于获取视频文件信息并将其写入 Excel 文件。代码能够自动获取视频格式、分辨率、码率、帧率、时长和大小等信息，并将它们写入 Excel 文件中。此外，代码还对视频时长进行了取整处理，确保所有信息准确无误。

代码如下：

import os
import subprocess
import openpyxl
import math

# 安装 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)

    # 获取文件大小并进行单位换算
    size = os.path.getsize(video_file)
    if size < 1024:
        size_str = f'{size} B'
    elif size < 1024*1024:
        size_str = f'{size/1024:.2f} KiB'
    elif size < 1024*1024*1024:
        size_str = f'{size/1024/1024:.2f} MiB'
    else:
        size_str = f'{size/1024/1024/1024:.2f} GiB'

    # 使用 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('|')
    
    # 将时长取整
    duration = duration.split('.')[0]
    h, m, s = duration.split(':')
    duration = f'{h}:{m}:{math.ceil(float(s))}'

    # 写入 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_str) # 写入文件大小

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

代码功能说明：

安装依赖库： 代码首先需要安装 openpyxl 库和 mediainfo 命令行工具。
设置变量： 代码定义了视频文件夹路径 video_folder 和输出 Excel 文件名 excel_file。
创建 Excel 文件： 代码使用 openpyxl 库创建了一个新的 Excel 文件，并设置了工作表名称为 Video Info。
设置表头： 代码将表头设置为 文件名, 格式, 分辨率, 码率, 帧率, 时长, 大小。
获取视频文件列表： 代码遍历视频文件夹，获取所有 .mp4 格式的视频文件。
获取视频信息： 代码使用 mediainfo 命令行工具获取每个视频文件的格式、分辨率、码率、帧率、时长等信息。
取整时长： 代码将获取的视频时长进行取整，确保所有信息准确无误。
写入 Excel 文件： 代码将获取到的所有视频信息写入 Excel 文件中。

代码特点：

代码清晰易懂，并附带详细注释，方便读者理解和应用。
代码使用了 mediainfo 命令行工具获取视频信息，确保信息准确可靠。
代码对视频时长进行了取整处理，确保所有信息准确无误。
代码能够自动生成 Excel 文件，方便用户查看和分析视频信息。

使用示例：

将视频文件放置在 video_folder 文件夹中，运行代码后，将在 excel_file 路径下生成一个 Excel 文件，其中包含所有视频文件的详细信息。

注意：

确保已安装 openpyxl 库和 mediainfo 命令行工具。
修改代码中的 video_folder 和 excel_file 变量，设置你的视频文件夹路径和输出 Excel 文件名。

希望本文能够帮助你更好地了解如何使用 Python 获取视频信息并将其写入 Excel 文件。