Python3 代码优化：提高视频信息提取效率

以下是对代码的优化建议：/n/n1. 使用列表推导式替代循环，以提高代码的可读性和性能。/n/n2. 使用f-strings替代字符串拼接，以提高代码的可读性和性能。/n/n3. 使用os.scandir替代os.walk，以提高代码的性能。/n/n4. 将视频格式和视频文件夹作为函数参数，以提高代码的灵活性。/n/n5. 将获取文件大小和单位换算的代码封装成一个函数，以提高代码的复用性和可读性。/n/n6. 将使用mediainfo获取视频信息和音频信息的代码封装成一个函数，以提高代码的复用性和可读性。/n/n7. 将解析输出结果的代码封装成一个函数，以提高代码的复用性和可读性。/n/n8. 使用typing模块添加类型提示，以提高代码的可读性。/n/n9. 在变量名和函数名中使用更具有描述性的名称，以提高代码的可读性。/n/n下面是优化后的代码：/n/n/nimport os/nimport re/nimport subprocess/nimport math/nfrom typing import List, Tuple/n/n/ndef get_video_files(video_folder: str, supported_formats: List[str]) -> List[str]:/n '''/n 获取视频文件列表（包括子目录）/n '''/n video_files = []/n for entry in os.scandir(video_folder):/n if entry.is_file() and any(entry.name.endswith(format) for format in supported_formats):/n video_files.append(entry.path)/n elif entry.is_dir():/n video_files.extend(get_video_files(entry.path, supported_formats))/n return video_files/n/n/ndef get_size_str(size: int) -> str:/n '''/n 获取文件大小并进行单位换算/n '''/n if size < 1024:/n return f'{size} B'/n elif size < 1024 * 1024:/n return f'{size / 1024:.2f} KiB'/n elif size < 1024 * 1024 * 1024:/n return f'{size / 1024 / 1024:.2f} MiB'/n else:/n return f'{size / 1024 / 1024 / 1024:.2f} GiB'/n/n/ndef get_video_audio_info(video_file: str) -> Tuple[List[str], List[str]]:/n '''/n 使用mediainfo获取视频信息 / 音频信息/n '''/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 return video_info, audio_info/n/n/ndef parse_video_audio_info(video_info: List[str], audio_info: List[str]) -> Tuple[List[str], float, List[str]]:/n '''/n 解析输出结果/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 # 以分钟计算的时长/n duration_minutes = int(h) * 60 + int(m) + math.ceil(float(s)) / 60/n/n # 使用正则表达式替换码率数据中第一个数字和第二个数字之间的空格/n pattern = re.compile(r'(/d+)/s+(/d+)')/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 return audio_lang, duration_minutes, bitrate, framerate, audio_bitrate, audio_sampling_rate/n/n/ndef print_video_info(video_file: str, size_str: str, duration: str, video_info: List[str], audio_info: List[str]) -> None:/n '''/n 打印文件信息/n '''/n audio_lang, duration_minutes, bitrate, framerate, audio_bitrate, audio_sampling_rate = parse_video_audio_info(video_info, audio_info)/n/n # 计算压缩比率/n ratio = round(duration_minutes / os.path.getsize(video_file) * 1000000000, 2)/n/n print(f'{os.path.basename(video_file)}, {size_str}, {duration}, {video_info[1]}, {bitrate}, {framerate}, {' / '.join(audio_lang)}, {ratio}')/n/n/nif name == 'main':/n # 视频格式/n supported_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 # 视频文件夹/n video_folder = '/home/115/up'/n/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_str = get_size_str(os.path.getsize(video_file))/n/n # 使用mediainfo获取视频信息 / 音频信息/n video_info, audio_info = get_video_audio_info(video_file)/n/n # 打印文件信息/n print_video_info(video_file, size_str, video_info, audio_info