Python3 代码优化：处理视频信息时正确打印双轨信息

Python3 代码优化：处理视频信息时正确打印双轨信息/n/n这段 Python3 代码用于处理视频文件，获取并打印包括音轨信息在内的各种信息。在处理多音轨视频时，原代码无法正确打印双轨信息，需要进行优化。/n/n问题: 原代码在处理多音轨时，直接将所有音轨的信息拼接在一起，导致无法区分不同的音轨。/n/n解决方法: 对每个音轨进行单独处理，将每个音轨的信息分别写入不同的行，方便用户查看。/n/n修改后的代码：/n/npython/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# 视频文件夹/nvideo_folder = input('请输入文件夹路径：')/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# 获取视频文件列表（包括子目录）/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# 锁定表头/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]/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# 遍历所有视频文件/nsuccess_count = 0/nrow = 2 # 从第二行开始写入数据/nfor video_file in video_files:/n try:/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(/n [/n 'mediainfo',/n '--Inform=Video;%Format%|%Width%x%Height%|%BitRate/String%|%FrameRate%|%Duration/String3%',/n video_file,/n ],/n stdout=subprocess.PIPE,/n )/n video_output = video_result.stdout.decode().strip()/n audio_result = subprocess.run(/n [/n 'mediainfo',/n '--Inform=Audio;%BitRate/String%|%SamplingRate/String%|%Language/String%',/n video_file,/n ],/n stdout=subprocess.PIPE,/n )/n audio_output = audio_result.stdout.decode().strip()/n/n # 解析输出结果/n video_info = video_output.split('|')/n audio_info = audio_output.split('|')/n/n # 处理多音轨情况/n audio_bitrate = []/n audio_sampling_rate = []/n audio_lang = []/n audio_info_len = len(audio_info)/n for i in range(audio_info_len // 3):/n audio_bitrate.append(audio_info[i * 3])/n audio_sampling_rate.append(audio_info[i * 3 + 1])/n audio_lang.append(audio_info[i * 3 + 2])/n if audio_info_len % 3 != 0: # 处理多余的音轨信息/n audio_bitrate.append('')/n audio_sampling_rate.append('')/n audio_lang.append('')/n for i in range(len(audio_bitrate)):/n if audio_bitrate[i] == '':/n audio_bitrate[i] = ' / '/n if audio_sampling_rate[i] == '':/n audio_sampling_rate[i] = ' / '/n ws.cell(row=row, column=9, value=audio_bitrate[0])/n ws.cell(row=row, column=10, value=audio_sampling_rate[0])/n if len(audio_bitrate) > 1:/n ws.cell(row=row, column=11, value='双轨')/n else:/n ws.cell(row=row, column=11, value=audio_lang[0])/n for i in range(1, len(audio_bitrate)):/n row += 1/n ws.cell(row=row, column=1, value='')/n ws.cell(row=row, column=2, value='')/n ws.cell(row=row, column=3, value='')/n ws.cell(row=row, column=4, value='')/n ws.cell(row=row, column=5, value='')/n ws.cell(row=row, column=6, value='')/n ws.cell(row=row, column=7, value='')/n ws.cell(row=row, column=8, value='')/n ws.cell(row=row, column=9, value=audio_bitrate[i])/n ws.cell(row=row, column=10, value=audio_sampling_rate[i])/n ws.cell(row=row, column=11, value=audio_lang[i])/n ws.cell(row=row, column=12, value='')/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 # 以分钟计算的时长 2/n duration_minutes = int(h) * 60 + int(m) + math.ceil(float(s)) / 60/n/n # 计算压缩比率/n ratio = round(duration_minutes / size * 1000000000, 2)/n/n # 使用正则表达式替换码率数据中第一个数字和第二个数字之间的空格/n bitrate = re.sub(pattern, r'/1/2', video_info[2])/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=video_info[0])/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=video_info[1])/n ws.cell(row=row, column=7, value=bitrate)/n ws.cell(row=row, column=8, value=video_info[3])/n # 打印进度/n print(f'已处理 {success_count+1}/{len(video_files)} 共计', end='/r')/n success_count += 1/n row += 1/n/n except Exception as e:/n # 处理失败，跳过并打印文件名和具体错误信息/n failed_count += 1/n print(f'处理文件 {file_name} 时失败：{e}')/n # 将失败的文件写入到err.txt/n with open('err.txt', 'a') as f:/n f.write(f'{file_name}/n')/n continue/n/n# 总计处理文件数/ntotal_count = len(video_files)/n/n# 调整L列左对齐/nl_col = ws['L']/nfor cell in l_col:/n cell.alignment = Alignment(horizontal='left') # 左对齐/n/n# 固定列宽/nfor col in ws.columns:/n col_letter = col[0].column_letter/n if col_letter not in ['']: # 不固定文件夹路径、格式、音频语言列的宽度/n ws.column_dimensions[col_letter].width = 11/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)/n/n# 打印处理失败文件个数和总计处理文件数/nprint(/n f'已完成，处理成功 {success_count} 个文件，处理失败 {failed_count} 个文件，视频信息已保存到 {excel_file_path}'/n)/n/n/n代码解释:/n/n1. 处理多余的音轨信息: 使用 if audio_info_len % 3 != 0: 判断是否有多余的音轨信息，如果有，则用空字符串填充 audio_bitrate, audio_sampling_rate 和 audio_lang 列表。/n2. 将每个音轨的信息分别写入不同的行: 使用一个循环遍历 audio_bitrate, audio_sampling_rate 和 audio_lang 列表，将每个音轨的信息分别写入不同的行。/n/n改进后的代码能够正确打印多音轨视频的音轨信息，方便用户查看。/n/n希望本文能够帮助您解决类似的代码问题。