Python3代码解析:视频信息提取与Excel导出
该代码使用Python3,利用mediainfo工具获取视频信息,并使用openpyxl库将信息存储到Excel表格中。/n/n代码功能:/n/n1. 获取指定文件夹中所有支持的视频文件(包括子目录)。/n2. 使用mediainfo获取每个视频文件的格式、大小、时长、分辨率、码率、帧率、音频码率、音频采样率和音频语言。/n3. 将获取到的视频信息写入Excel表格,并自动调整列宽。/n4. 将生成的Excel文件保存到指定路径。/n/n注意事项:/n/n1. 确保安装了mediainfo软件,否则会出现subprocess.CalledProcessError错误。/n2. 确保输入的文件夹路径存在。/n3. 确保输出Excel文件名不会覆盖已有的文件。如果已有同名文件,代码将直接覆盖。/n4. 确保所使用的Python库已安装。如果缺少某个库,可以使用pip install命令安装。/n/n代码分析:/n/npython/nimport os/nimport re/nimport subprocess/nimport openpyxl/nimport math/nfrom openpyxl.styles import Font, PatternFill, Alignment/n/n/n# 支持的视频格式/nsupported_formats = ['.mp4', '.avi', '.mkv', '.wmv', '.mov', '.flv', '.m2ts', '.ts', '.rm', '.rmvb',/n '.vob', '.3gp', '.webm', '.hdmov', '.mp4v', '.mpv4', '.divx', '.xvid', '.f4v',/n '.mpeg', '.asf', '.asx', '.m2t']/n# 需要处理的视频文件夹/nvideo_folder = input('请输入文件夹路径:')/n# 输出Excel文件夹路径/nexcel_folder = '.'/n# 输出Excel文件名/nexcel_file = 'video_info.xlsx'/n# 正则表达式匹配码率数据中的空格/npattern = re.compile(r'(/d+)/s+(/d+)')/n# 创建Excel文件/nwb = openpyxl.Workbook()/nws = wb.active/nws.title = 'Video Info'/n/n/n# 设置表头样式/nheader_font = Font(bold=True, color='800080') # 紫色/nheader_fill = PatternFill('solid', fgColor='C5E0B4')/nheader_alignment = Alignment(horizontal='center', vertical='center')/nheaders = ['文件名', '文件路径', '格式', '大小', '时长', '分辨率', '码率', '帧率', '音频码率', '音频采样率', '音频语言', '压缩比率']/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# 锁定表头/nws.freeze_panes = 'A2'/n/n# 获取视频文件列表(包括子目录)/nvideo_files = []/nfor root, dirs, files in os.walk(video_folder):/n for file in files:/n for format in supported_formats:/n if file.endswith(format):/n video_files.append(os.path.join(root, file))/n break/n/n# 遍历所有视频文件,获取视频信息、音频信息和音频语言,并写入Excel文件/nfor row, video_file in enumerate(video_files, start=2):/n # 获取文件名/n file_name = os.path.basename(video_file)/n/n # 获取文件大小并进行单位换算/n size = os.path.getsize(video_file)/n if size < 1024:/n size_str = f'{size} B'/n elif size < 1024 * 1024:/n size_str = f'{size / 1024:.2f} KiB'/n elif size < 1024 * 1024 * 1024:/n size_str = f'{size / 1024 / 1024:.2f} MiB'/n else:/n size_str = f'{size / 1024 / 1024 / 1024:.2f} GiB'/n/n # 使用mediainfo获取视频信息 / 音频信息和音频语言/n video_result = subprocess.run(['mediainfo', '--Inform=Video;%Format%|%Width%x%Height%|%BitRate/String%|%FrameRate%|%Duration/String3%', video_file], stdout=subprocess.PIPE)/n video_output = video_result.stdout.decode().strip()/n audio_result = subprocess.run(['mediainfo', '--Inform=Audio;%BitRate/String%|%SamplingRate/String%|%Language/String%', video_file], stdout=subprocess.PIPE)/n audio_output = audio_result.stdout.decode().strip()/n/n # 解析输出结果/n format, resolution, bitrate, framerate, duration = video_output.split('|')/n audiobitrate, audiosamplingrate, audiolang = audio_output.split('|')/n/n # 时长取整/n duration = duration.split('.')[0]/n h, m, s = duration.split(':')/n duration = f'{h}:{m}:{math.ceil(float(s))}'/n/n # 计算压缩比率/n ratio = f'{round(float(duration) / size, 2)}'/n/n # 使用正则表达式替换码率数据中第一个数字和第二个数字之间的空格/n bitrate = re.sub(pattern, r'/1/2', bitrate)/n /n # 写入Excel文件/n ws.cell(row=row, column=1, value=file_name)/n ws.cell(row=row, column=2, value=os.path.dirname(video_file)) # 写入文件夹路径/n ws.cell(row=row, column=3, value=format)/n ws.cell(row=row, column=4, value=size_str) # 写入文件大小/n ws.cell(row=row, column=5, value=duration)/n ws.cell(row=row, column=6, value=resolution)/n ws.cell(row=row, column=7, value=bitrate)/n ws.cell(row=row, column=8, value=framerate)/n ws.cell(row=row, column=9, value=audiobitrate)/n ws.cell(row=row, column=10, value=audiosamplingrate)/n ws.cell(row=row, column=11, value=audiolang)/n ws.cell(row=row, column=12, value=ratio)/n/n# 自动调整列宽/nfor col in ws.columns:/n max_length = 0/n column = col[0].column_letter/n for cell in col:/n try:/n if len(str(cell.value)) > max_length:/n max_length = len(str(cell.value))/n except:/n pass/n adjusted_width = (max_length + 2)/n ws.column_dimensions[column].width = adjusted_width + 1 # 列宽+1/n/n# 保存Excel文件/nif not os.path.exists(excel_folder):/n os.makedirs(excel_folder)/nexcel_file_path = os.path.join(excel_folder, excel_file)/nwb.save(excel_file_path)/nprint(f'视频信息已保存到 {excel_file_path}')/n/n/n代码解释:/n/n1. 导入必要的库:os、re、subprocess、openpyxl、math、openpyxl.styles。/n2. 定义支持的视频格式列表 supported_formats。/n3. 获取用户输入的视频文件夹路径 video_folder。/n4. 设置输出Excel文件夹路径 excel_folder 和文件名 excel_file。/n5. 定义正则表达式 pattern 用于匹配码率数据中的空格。/n6. 创建一个新的Excel工作簿 wb 和工作表 ws,并设置表头样式。/n7. 使用 os.walk 遍历指定文件夹及其子目录,获取所有支持的视频文件。/n8. 循环遍历每个视频文件,使用 subprocess.run 调用 mediainfo 获取视频信息和音频信息。/n9. 解析 mediainfo 的输出结果,并将信息写入Excel工作表中。/n10. 自动调整列宽。/n11. 将Excel工作簿保存到指定路径。/n/n总结:/n/n本代码提供了一个完整的视频信息提取和Excel导出方案,用户可以根据自身需求进行修改和扩展。同时,代码中的注释和分析有助于用户更好地理解代码功能和逻辑。/n
原文地址: https://www.cveoy.top/t/topic/mAKX 著作权归作者所有。请勿转载和采集!