Python 3 代码优化：使用函数提高视频信息提取效率

Python 3 代码优化：使用函数提高视频信息提取效率/n/n本代码示例展示如何将 Python 3 代码优化为函数结构，以提高视频信息提取的效率。/n/n原始代码：/n/npython/nimport os/nimport re/nimport subprocess/nimport math/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 = '/home/115/up'/n# 正则表达式匹配码率数据中的空格/npattern = re.compile(r'(/d+)/s+(/d+)')/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# 遍历所有视频文件/nfor video_file in video_files:/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 video_info = video_output.split('|')/n audio_info = audio_output.split('|')/n/n # 处理多音轨情况/n audio_bitrate = []/n audio_sampling_rate = []/n audio_lang = []/n for i in range(len(audio_info) // 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/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 # 获取帧率/n try:/n framerate = float(video_info[3])/n except ValueError:/n framerate = None/n/n # 打印文件信息/n print(f'{file_name}, {size_str}, {duration}, {video_info[1]}, {bitrate}, {framerate}, {' / '.join(audio_lang)}, {ratio}')/n/n/n优化后的代码：/n/npython/nimport os/nimport re/nimport subprocess/nimport math/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 = '/home/115/up'/n# 正则表达式匹配码率数据中的空格/npattern = re.compile(r'(/d+)/s+(/d+)')/n/n# 定义函数获取视频文件列表/ndef get_video_files(video_folder, supported_formats):/n video_files = []/n for 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 return video_files/n/n# 定义函数获取文件大小的字符串表示/ndef get_size_str(size):/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 return size_str/n/n# 定义函数获取视频信息和音频信息/ndef get_video_audio_info(video_file):/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 return video_output, audio_output/n/n# 定义函数处理多音轨情况/ndef process_audio_info(audio_info):/n audio_bitrate = []/n audio_sampling_rate = []/n audio_lang = []/n for i in range(len(audio_info) // 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 return audio_bitrate, audio_sampling_rate, audio_lang/n/n# 定义函数获取帧率/ndef get_framerate(video_info):/n try:/n framerate = float(video_info[3])/n except ValueError:/n framerate = None/n return framerate/n/n# 定义主函数/ndef main():/n # 获取视频文件列表（包括子目录）/n video_files = get_video_files(video_folder, supported_formats)/n/n # 遍历所有视频文件/n for video_file in video_files:/n # 获取文件名/n file_name = os.path.basename(video_file)/n/n # 获取文件大小并进行单位换算/n size = os.path.getsize(video_file)/n size_str = get_size_str(size)/n/n # 使用mediainfo获取视频信息 / 音频信息/n video_output, audio_output = get_video_audio_info(video_file)/n/n # 解析输出结果/n video_info = video_output.split('|')/n audio_info = audio_output.split('|')/n/n # 处理多音轨情况/n audio_bitrate, audio_sampling_rate, audio_lang = process_audio_info(audio_info)/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 # 获取帧率/n framerate = get_framerate(video_info)/n/n # 打印文件信息/n print(f'{file_name}, {size_str}, {duration}, {video_info[1]}, {bitrate}, {framerate}, {' / '.join(audio_lang)}, {ratio}')/n/n# 调用主函数/nif __name__ == '__main__':/n main()/n/n/n优化说明：/n/n1. 将重复的代码段封装为函数，例如获取视频文件列表、计算文件大小、获取视频音频信息、处理多音轨等。/n2. 函数的使用使代码结构更清晰，易于理解和维护。/n3. 提高代码可重用性，这些函数可以在其他程序中重复使用。/n/n优点：/n/n* 代码结构更清晰，更易于理解和维护。/n* 提高代码可重用性。/n* 提高代码效率，减少重复代码。/n/n注意：/n/n* 优化后的代码使用 mediainfo 命令来获取视频信息，需要确保系统安装了 mediainfo 工具。/n* 代码中的视频文件夹路径和支持的视频格式可以根据实际情况进行调整。/n