Python 代码优化：将视频时长取整并添加单位

使用 Python 代码获取视频信息并取整时长

本文提供一段 Python 代码，可以获取视频文件夹内所有视频的信息，并写入 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)

    # 获取文件大小并进行单位换算
    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 = str(round(float(duration[:-1]))) + '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 模块，这两个模块分别用于操作 Excel 文件和获取视频信息。
定义需要处理的视频文件夹路径和输出 Excel 文件路径。
创建一个新的 Excel 文件，并设置表头。
获取视频文件夹内的所有 .mp4 文件。
循环遍历每个视频文件，使用 mediainfo 获取视频信息，并解析输出结果。
将获取到的视频信息写入 Excel 文件中。

代码修改:

在第 35 行代码中，将 duration 变量的值进行取整，并将单位 's' 添加到字符串末尾。这样就将视频时长取整为秒，并加上单位。

注意:

以上代码需要在安装了 mediainfo 命令行工具的系统上运行。
您需要根据您的实际情况修改视频文件夹路径和输出 Excel 文件路径。

希望本文对您有所帮助！如果您有任何问题，请随时留言。