Python 视频信息提取工具：自动生成视频信息 Excel 表格

python/nimport os/nimport re/nimport subprocess/nimport openpyxl/nimport math/nfrom openpyxl.styles import Font, PatternFill, Alignment/n/n# 视频格式/nsupported_formats = [/n '.mp4',/n '.avi',/n '.mkv',/n '.wmv',/n '.mov',/n '.flv',/n '.m2ts',/n '.ts',/n '.rm',/n '.rmvb',/n '.vob',/n '.3gp',/n '.webm',/n '.hdmov',/n '.mp4v',/n '.mpv4',/n '.divx',/n '.xvid',/n '.f4v',/n '.mpeg',/n '.asf',/n '.asx',/n '.m2t',/n]/n/n# 输出Excel路径/nexcel_folder = '.'/n/n# 输出Excel文件名/nexcel_file = 'video_info.xlsx'/n/n# 正则表达式匹配码率数据中的空格/npattern = re.compile(r'(/d+)/s+(/d+)')/n/n# 创建Excel文件/nwb = openpyxl.Workbook()/nws = wb.active/nws.title = 'Video Info'/n/n# 锁定表头/nws.freeze_panes = 'A2'/n/n# 设置表头样式/nheader_font = Font(bold=True, color='800080') # 紫色/nheader_fill = PatternFill('solid', fgColor='C5E0B4')/nheader_alignment = Alignment(horizontal='center', vertical='center')/nheaders = [/n '文件名',/n '文件路径',/n '格式',/n '大小',/n '时长',/n '分辨率',/n '码率',/n '帧率',/n '音频码率',/n '音频采样率',/n '音频语言',/n '压缩比率',/n '文件总数',/n]/nfor col, header in enumerate(headers, start=1):/n cell = ws.cell(row=1, column=col, value=header)/n cell.font = header_font/n cell.fill = header_fill/n cell.alignment = header_alignment/n/n# 处理失败文件个数/nfailed_count = 0/n/n# 总计处理文件数/ntotal_count = 0/n/n# 遍历所有视频文件/nsuccess_count = 0/nrow = 2 # 从第二行开始写入数据/n/ndef process_video(video_file):/n global success_count, row, failed_count/n try:/n # 获取文件名/n file_name = os.path.basename(video_file)/n/n # 获取文件大小/n size_str = convert_size(os.path.getsize(video_file))/n/n # 使用mediainfo获取视频信息 / 音频信息/n with subprocess.Popen(/n [/n 'mediainfo',/n '--Inform=Video;%Format%|%Width%x%Height%|%BitRate/String%|%FrameRate%|%Duration/String3%',/n video_file/n ], /n stdout=subprocess.PIPE) as video_proc, /n subprocess.Popen(/n ['mediainfo',/n '--Inform=Audio;%BitRate/String%|%SamplingRate/String%|%Language/String%',/n video_file/n ],/n stdout=subprocess.PIPE) as audio_proc:/n video_output = video_proc.communicate()[0].decode().strip()/n audio_output = audio_proc.communicate()[0].decode().strip()/n/n # 解析输出结果/n video_info = video_output.split('|')/n audio_info = audio_output.split('|')/n/n # 处理多音轨情况/n audio_bitrate, audio_sampling_rate, audio_lang = zip(*[(audio_info[i*3], /n audio_info[i*3+1], audio_info[i*3+2]) for i in range(len(audio_info)//3)])/n/n # 时长取整/n duration = video_info[4].split('.')[0]/n h, m, s = duration.split(':')/n duration = f'{h}:{m}:{str(s).zfill(2)}'/n/n # 以分钟计算的时长/n duration_minutes = int(h) * 60 + int(m) + math.ceil(float(s)) / 60/n/n # 计算压缩比率/n ratio = round(duration_minutes / os.path.getsize(video_file) * 1000000000, 2)/n/n # 使用正则表达式替换码率数据中第1、2数字之间的空格/n bitrate = re.sub(pattern, r'//1//2', video_info[2])/n/n # 写入Excel文件/n write_cell(ws, row, 1, file_name)/n write_cell(ws, row, 2, os.path.dirname(video_file)) # 写入文件路径/n write_cell(ws, row, 3, video_info[0])/n write_cell(ws, row, 4, size_str)/n write_cell(ws, row, 5, duration)/n write_cell(ws, row, 6, video_info[1])/n write_cell(ws, row, 7, bitrate)/n write_cell(ws, row, 8, video_info[3])/n write_cell(ws, row, 9, ' / '.join(audio_bitrate))/n write_cell(ws, row, 10, ' / '.join(audio_sampling_rate))/n write_cell(ws, row, 11, ' / '.join(audio_lang))/n write_cell(ws, row, 12, ratio)/n write_cell(ws, 2, 13, total_count) # 写入文件总数/n/n # 打印进度/n success_count += 1/n row += 1/n/n except Exception as e:/n # 处理失败, 跳过并打印文件/n failed_count += 1/n print(f'处理文件 {video_file} 时失败：{e}')/n # 写入失败到err.txt/n with open('err.txt', 'a') as f:/n f.write(f'{video_file}/n')/n return False/n/n return True/n/ndef process_folder(video_folder):/n global total_count/n # 获取视频文件递归/n video_files = []/n for root, dirs, files in os.walk(video_folder):/n for file in files:/n if os.path.splitext(file)[1] in supported_formats:/n video_files.append(os.path.join(root, file))/n total_count = len(video_files)/n if total_count == 0:/n print('无视频文件或路径为空')/n else:/n for i, video_file in enumerate(video_files, start=1):/n if process_video(video_file):/n print(f'已处理 {success_count}/{total_count} 共计', end='//r')/n else:/n print(f'处理文件 {i}/{total_count} 时失败', end='//r')/n print(f'已处理 {success_count}/{total_count} 共计')/n/ndef convert_size(size):/n # 转换单位/n units = ['B', 'KiB', 'MiB', 'GiB']/n i = 0/n while size >= 1024 and i < len(units)-1:/n size /= 1024/n i += 1/n return f'{size:.2f} {units[i]}'/n/ndef write_cell(ws, row, col, value):/n # 写入Excel/n cell = ws.cell(row=row, column=col, value=value)/n if col == 13:/n cell.font = Font(bold=True, color='FF0000')/n if col in [10, 11, 13]:/n cell.alignment = Alignment(horizontal='left')/n else:/n cell.alignment = Alignment(horizontal='center')/n if col not in [3, 8, 9, 10, 11, 12, 13]:/n ws.column_dimensions[cell.column_letter].width = 11/n/n# 文件路径/npath = input('请输入文件路径：')/nif os.path.isfile(path):/n # 1个文件 total_count += 1/n total_count = 1 /n process_video(path)/nelif os.path.isdir(path):/n process_folder(path) /n/nif total_count > 0:/n # 保存Excel文件/n os.makedirs(excel_folder, exist_ok=True)/n excel_file_path = os.path.join(excel_folder, excel_file)/n wb.save(excel_file_path)/n/n # 打印处理失败文件个数和总计处理文件数/n print(/n f'成功处理 {success_count} 个文件, 失败 {failed_count} 个文件, 视频信息已保存到 {excel_file_path}'/n )/n/n/n## 功能介绍/n/n该 Python 脚本可用于批量提取视频信息并生成 Excel 表格，支持以下功能：/n/n* 支持多种视频格式： 包括 .mp4、.avi、.mkv 等常见格式。/n* 获取丰富视频信息： 包括文件名、文件路径、格式、大小、时长、分辨率、码率、帧率、音频码率、音频采样率、音频语言、压缩比率等信息。/n* 生成 Excel 表格： 将所有信息整理成易于阅读的 Excel 表格。/n* 批量处理： 支持处理单个视频文件或整个文件夹。/n* 进度显示： 实时显示已处理的文件数量和总计文件数。/n* 错误处理： 记录处理失败的文件并输出到 err.txt 文件。/n/n## 使用方法/n/n1. 确保已安装 Python 和 openpyxl 库：/nbash/npip install openpyxl/n/n2. 将代码保存为 .py 文件，例如 video_info.py。/n3. 运行代码并输入目标文件或文件夹路径：/nbash/npython video_info.py/n/n4. 代码会自动生成一个名为 video_info.xlsx 的 Excel 文件，保存到当前目录。/n/n## 示例/n/n假设您要提取名为 videos 的文件夹中的所有视频信息，则可以按照以下步骤操作：/n/n1. 运行代码。/n2. 在提示符中输入 'videos' 并按回车键。/n3. 代码会自动处理 videos 文件夹中的所有视频文件，并生成 video_info.xlsx 文件。/n/n## 注意事项/n/n* 确保目标路径中包含视频文件，否则程序不会生成任何输出。/n* 脚本依赖 mediainfo 工具，请确保已安装并配置好环境变量。/n* 处理大文件或大量文件可能需要较长时间，请耐心等待。/n/n## 总结/n/n该 Python 脚本为用户提供了一个方便快捷的工具，可以自动提取视频信息并生成 Excel 表格，节省大量手动操作时间。用户可以根据自己的需要调整代码，例如添加更多信息项或自定义输出格式。/n