xym / VideoHeightHandler

import os import shutil import subprocess import concurrent.futures from pathlib import Path from datetime import datetime # 指定ffmpeg的二进制文件夹位置，如果ffmpeg不在系统PATH中 FFMPEG_BIN = "C:\\ffmpeg-7.0.2-full_build\\bin\\" # 请根据实际情况修改路径 def log_error(message): """将错误消息写入到日志文件中""" error_log_file = f"errorlog-{datetime.now().strftime('%Y%m%d')}.txt" with open(error_log_file, 'a') as f: timestamp = datetime.now().strftime('%Y-%m-%d %H:%M:%S') f.write(f"[{timestamp}] {message}\n") def get_video_resolution_ffprobe(video_path): """使用ffprobe获取视频文件的分辨率""" ffprobe_cmd = os.path.join(FFMPEG_BIN, 'ffprobe') if FFMPEG_BIN else 'ffprobe' cmd = [ ffprobe_cmd, '-v', 'error', '-select_streams', 'v:0', '-show_entries', 'stream=width,height', '-of', 'csv=p=0', video_path ] try: output = subprocess.check_output(cmd).decode().strip() width, height = map(int, output.split(',')) return width, height except subprocess.CalledProcessError as e: error_message = f"ffprobe error for file {video_path}: {e.output.decode()}" log_error(error_message) return None, None def move_video(video_path, base_dir): """移动单个视频文件""" try: # 获取视频分辨率 _, height = get_video_resolution_ffprobe(video_path) if height: # 创建目标目录 target_dir = Path(base_dir).parent / str(height) target_dir.mkdir(exist_ok=True) # 移动文件 target_path = target_dir / Path(video_path).name shutil.move(video_path, target_path) print(f"Moved '{video_path}' to '{target_path}'") except Exception as e: error_message = f"Error moving file {Path(video_path).name}: {e}" log_error(error_message) def move_videos_by_height(base_dir): """根据视频高度并行移动文件""" video_files = [] for root, _, files in os.walk(base_dir): for file in files: if file.lower().endswith(('.mp4', '.mkv', '.avi', '.mov', '.wmv', '.rmvb', '.flv', '.m4v', '.mpg')): video_files.append(os.path.join(root, file)) with concurrent.futures.ThreadPoolExecutor() as executor: # 使用多线程处理文件 executor.map(move_video, video_files, [base_dir] * len(video_files)) # 用法示例 base_directory = "foo:\\bar" move_videos_by_height(base_directory)