Python3 代码优化：分析视频文件信息

Python3 代码优化：分析视频文件信息/n/n这段代码用于分析视频文件信息，包括文件大小、时长、分辨率、码率、帧率、音频信息等。以下是一些优化建议：/n/n1. 使用列表推导式获取视频文件列表：/n/npython/nvideo_files = [os.path.join(root, file) for root, dirs, files in os.walk(video_folder) for file in files if any(file.endswith(format) for format in supported_formats)]/n/n/n2. 将视频信息和音频信息的获取封装成函数：/n/npython/ndef get_media_info(media_file):/n video_result = subprocess.run(['mediainfo', '--Inform=Video;%Format%|%Width%x%Height%|%BitRate/String%|%FrameRate%|%Duration/String3%', media_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%', media_file], stdout=subprocess.PIPE)/n audio_output = audio_result.stdout.decode().strip()/n return video_output.split('|'), audio_output.split('|')/n/n/n3. 使用 f 字符串格式化输出信息：/n/npython/nprint(f'{file_name}, {size_str}, {duration}, {video_info[1]}, {bitrate}, {framerate or ''}, { /' / /'.join(audio_lang) or ''}, {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# 获取视频文件列表（包括子目录）/nvideo_files = [os.path.join(root, file) for root, dirs, files in os.walk(video_folder) for file in files if any(file.endswith(format) for format in supported_formats)]/n/n# 获取视频信息和音频信息的函数/ndef get_media_info(media_file):/n video_result = subprocess.run(['mediainfo', '--Inform=Video;%Format%|%Width%x%Height%|%BitRate/String%|%FrameRate%|%Duration/String3%', media_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%', media_file], stdout=subprocess.PIPE)/n audio_output = audio_result.stdout.decode().strip()/n return video_output.split('|'), audio_output.split('|')/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 # 获取视频信息和音频信息/n video_info, audio_info = get_media_info(video_file)/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 or ''}, { ' / '.join(audio_lang) or ''}, {ratio}')/n/n/n代码优化总结：/n/n* 使用列表推导式简化获取视频文件列表的代码。/n* 将获取视频信息和音频信息的代码封装成函数，提高代码可读性和可维护性。/n* 使用 f 字符串格式化输出信息，使代码更加简洁易懂。/n/n注意：/n/n* 此代码需要安装 mediainfo 工具。/n* 视频文件夹路径需要根据实际情况进行修改。/n* 代码中的支持格式列表可以根据需要进行调整。/n* 压缩比率的计算方法可以根据需要进行调整。/n* 代码中的正则表达式可以根据需要进行调整。/n* 帧率的获取可能会失败，需要根据实际情况进行处理。/n* 代码中使用的 print 语句可以根据需要进行调整，例如写入文件或输出到数据库。