Python3 视频信息提取并写入 Excel 脚本

以下是一个使用 pymediainfo 库统计视频信息并将其写入 Excel 文件的 Python3 脚本示例：

import os
import sys
import argparse
import csv
from pprint import pprint
from pymediainfo import MediaInfo

def get_video_info(file_path):
    media_info = MediaInfo.parse(file_path)
    video_info = {}
    for track in media_info.tracks:
        if track.track_type == 'Video':
            video_info['codec'] = track.codec
            video_info['bitrate'] = track.bit_rate
            video_info['width'] = track.width
            video_info['height'] = track.height
            video_info['duration'] = track.duration
    return video_info

def write_to_excel(video_files, output_file):
    with open(output_file, 'w', newline='') as f:
        writer = csv.writer(f)
        writer.writerow(['File Name', 'Codec', 'Bitrate', 'Width', 'Height', 'Duration'])
        for file_path in video_files:
            video_info = get_video_info(file_path)
            writer.writerow([os.path.basename(file_path), video_info.get('codec', ''),
                             video_info.get('bitrate', ''), video_info.get('width', ''),
                             video_info.get('height', ''), video_info.get('duration', '')])

if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='Get video information and write to Excel')
    parser.add_argument('input_folder', help='Input folder containing video files')
    parser.add_argument('output_file', help='Output Excel file')
    args = parser.parse_args()

    video_files = [os.path.join(args.input_folder, f) for f in os.listdir(args.input_folder) if f.endswith('.mp4')]
    write_to_excel(video_files, args.output_file)

该脚本将遍历指定文件夹中的所有.mp4文件，并将它们的基本信息写入一个Excel文件中，包括文件名、编解码器、比特率、宽度、高度和持续时间。要运行脚本，请按照以下示例命令行参数运行：

python3 video_info_to_excel.py /path/to/input/folder /path/to/output/file.xlsx

其中，/path/to/input/folder 是要遍历的文件夹路径，/path/to/output/file.xlsx 是要写入的Excel文件路径。