mypy

util/ffmpeg_utils.py

@@ -1,16 +1,18 @@
 """
 FFmpeg工具模块 - 提供FFmpeg命令构建和处理的公共函数
 """
+
 import logging
 from typing import List, Tuple, Optional
 from config.settings import get_ffmpeg_config
 
 logger = logging.getLogger(__name__)
 
+
 def build_base_ffmpeg_args() -> List[str]:
     """
     构建基础FFmpeg参数
-    
+
     Returns:
         基础参数列表
     """
@@ -20,40 +22,45 @@ def build_base_ffmpeg_args() -> List[str]:
     args.extend(config.loglevel_args)
     return args
 
+
 def build_null_audio_input() -> List[str]:
     """
     构建空音频输入参数
-    
+
     Returns:
         空音频输入参数列表
     """
     config = get_ffmpeg_config()
     return config.null_audio_args
 
+
 def build_amix_filter(input1: str, input2: str, output: str) -> str:
     """
     构建音频混合滤镜
-    
+
     Args:
         input1: 第一个音频输入
-        input2: 第二个音频输入  
+        input2: 第二个音频输入
         output: 输出流名称
-        
+
     Returns:
         混合滤镜字符串
     """
     config = get_ffmpeg_config()
     return f"{input1}[{input2}]{config.amix_args[0]}[{output}]"
 
-def build_overlay_scale_filter(video_input: str, overlay_input: str, output: str) -> str:
+
+def build_overlay_scale_filter(
+    video_input: str, overlay_input: str, output: str
+) -> str:
     """
     构建覆盖层缩放滤镜
-    
+
     Args:
         video_input: 视频输入流
         overlay_input: 覆盖层输入流
         output: 输出流名称
-        
+
     Returns:
         缩放滤镜字符串
     """
@@ -61,12 +68,15 @@ def build_overlay_scale_filter(video_input: str, overlay_input: str, output: str
     if config.overlay_scale_mode == "scale":
         return f"{video_input}[{overlay_input}]scale=iw:ih[{output}]"
     else:
-        return f"{video_input}[{overlay_input}]{config.overlay_scale_mode}=iw:ih[{output}]"
+        return (
+            f"{video_input}[{overlay_input}]{config.overlay_scale_mode}=iw:ih[{output}]"
+        )
+
 
 def get_annexb_filter() -> str:
     """
     获取annexb转换滤镜
-    
+
     Returns:
         annexb滤镜名称
     """
@@ -76,10 +86,11 @@ def get_annexb_filter() -> str:
         return "hevc_mp4toannexb"
     return "h264_mp4toannexb"
 
+
 def build_standard_output_args() -> List[str]:
     """
     构建标准输出参数
-    
+
     Returns:
         输出参数列表
     """
@@ -88,40 +99,63 @@ def build_standard_output_args() -> List[str]:
         *config.video_args,
         *config.audio_args,
         *config.encoder_args,
-        *config.default_args
+        *config.default_args,
     ]
 
+
 def validate_ffmpeg_file_extensions(file_path: str) -> bool:
     """
     验证文件扩展名是否为FFmpeg支持的格式
-    
+
     Args:
         file_path: 文件路径
-        
+
     Returns:
         是否为支持的格式
     """
     supported_extensions = {
-        '.mp4', '.avi', '.mov', '.mkv', '.flv', '.wmv', '.webm',
-        '.ts', '.m2ts', '.mts', '.m4v', '.3gp', '.asf', '.rm',
-        '.mp3', '.wav', '.aac', '.flac', '.ogg', '.m4a', '.wma'
+        ".mp4",
+        ".avi",
+        ".mov",
+        ".mkv",
+        ".flv",
+        ".wmv",
+        ".webm",
+        ".ts",
+        ".m2ts",
+        ".mts",
+        ".m4v",
+        ".3gp",
+        ".asf",
+        ".rm",
+        ".mp3",
+        ".wav",
+        ".aac",
+        ".flac",
+        ".ogg",
+        ".m4a",
+        ".wma",
     }
-    
+
     import os
+
     _, ext = os.path.splitext(file_path.lower())
     return ext in supported_extensions
 
-def estimate_processing_time(input_duration: float, complexity_factor: float = 1.0) -> float:
+
+def estimate_processing_time(
+    input_duration: float, complexity_factor: float = 1.0
+) -> float:
     """
     估算处理时间
-    
+
     Args:
         input_duration: 输入文件时长（秒）
         complexity_factor: 复杂度因子（1.0为普通处理）
-        
+
     Returns:
         预估处理时间（秒）
     """
     # 基础处理速度假设为实时的0.5倍（即处理1秒视频需要2秒）
     base_processing_ratio = 2.0
-    return input_duration * base_processing_ratio * complexity_factor
+    return input_duration * base_processing_ratio * complexity_factor