mirror of
https://github.com/pacnpal/Pac-cogs.git
synced 2025-12-20 10:51:05 -05:00
a4ca6e8ea6
Component-based architecture with lifecycle management Enhanced error handling and recovery mechanisms Comprehensive state management and tracking Event-driven architecture with monitoring Queue Management: Multiple processing strategies for different scenarios Advanced state management with recovery Comprehensive metrics and health monitoring Sophisticated cleanup system with multiple strategies Processing Pipeline: Enhanced message handling with validation Improved URL extraction and processing Better queue management and monitoring Advanced cleanup mechanisms Overall Benefits: Better code organization and maintainability Improved error handling and recovery Enhanced monitoring and reporting More robust and reliable system
128 lines
4.1 KiB
Python
128 lines
4.1 KiB
Python
"""Module for managing FFmpeg processes"""
|
|
|
|
import logging
|
|
import psutil
|
|
import subprocess
|
|
import time
|
|
from typing import Set, Optional
|
|
|
|
logger = logging.getLogger("FFmpegProcessManager")
|
|
|
|
class ProcessManager:
|
|
"""Manages FFmpeg process execution and lifecycle"""
|
|
|
|
def __init__(self):
|
|
self._active_processes: Set[subprocess.Popen] = set()
|
|
|
|
def add_process(self, process: subprocess.Popen) -> None:
|
|
"""Add a process to track"""
|
|
self._active_processes.add(process)
|
|
|
|
def remove_process(self, process: subprocess.Popen) -> None:
|
|
"""Remove a process from tracking"""
|
|
self._active_processes.discard(process)
|
|
|
|
def kill_all_processes(self) -> None:
|
|
"""Kill all active FFmpeg processes"""
|
|
try:
|
|
# First try graceful termination
|
|
self._terminate_processes()
|
|
|
|
# Give processes a moment to terminate
|
|
time.sleep(0.5)
|
|
|
|
# Force kill any remaining processes
|
|
self._kill_remaining_processes()
|
|
|
|
# Find and kill any orphaned FFmpeg processes
|
|
self._kill_orphaned_processes()
|
|
|
|
self._active_processes.clear()
|
|
logger.info("All FFmpeg processes terminated")
|
|
|
|
except Exception as e:
|
|
logger.error(f"Error killing FFmpeg processes: {e}")
|
|
|
|
def _terminate_processes(self) -> None:
|
|
"""Attempt graceful termination of processes"""
|
|
for process in self._active_processes:
|
|
try:
|
|
if process.poll() is None: # Process is still running
|
|
process.terminate()
|
|
except Exception as e:
|
|
logger.error(f"Error terminating FFmpeg process: {e}")
|
|
|
|
def _kill_remaining_processes(self) -> None:
|
|
"""Force kill any remaining processes"""
|
|
for process in self._active_processes:
|
|
try:
|
|
if process.poll() is None: # Process is still running
|
|
process.kill()
|
|
except Exception as e:
|
|
logger.error(f"Error killing FFmpeg process: {e}")
|
|
|
|
def _kill_orphaned_processes(self) -> None:
|
|
"""Find and kill any orphaned FFmpeg processes"""
|
|
for proc in psutil.process_iter(['pid', 'name', 'cmdline']):
|
|
try:
|
|
if 'ffmpeg' in proc.info['name'].lower():
|
|
proc.kill()
|
|
except (psutil.NoSuchProcess, psutil.AccessDenied, psutil.ZombieProcess):
|
|
pass
|
|
except Exception as e:
|
|
logger.error(f"Error killing orphaned FFmpeg process: {e}")
|
|
|
|
def execute_command(
|
|
self,
|
|
command: list,
|
|
timeout: Optional[int] = None,
|
|
check: bool = False
|
|
) -> subprocess.CompletedProcess:
|
|
"""Execute an FFmpeg command with proper process management
|
|
|
|
Args:
|
|
command: Command list to execute
|
|
timeout: Optional timeout in seconds
|
|
check: Whether to check return code
|
|
|
|
Returns:
|
|
subprocess.CompletedProcess: Result of command execution
|
|
"""
|
|
process = None
|
|
try:
|
|
process = subprocess.Popen(
|
|
command,
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE,
|
|
text=True
|
|
)
|
|
self.add_process(process)
|
|
|
|
stdout, stderr = process.communicate(timeout=timeout)
|
|
result = subprocess.CompletedProcess(
|
|
args=command,
|
|
returncode=process.returncode,
|
|
stdout=stdout,
|
|
stderr=stderr
|
|
)
|
|
|
|
if check and process.returncode != 0:
|
|
raise subprocess.CalledProcessError(
|
|
returncode=process.returncode,
|
|
cmd=command,
|
|
output=stdout,
|
|
stderr=stderr
|
|
)
|
|
|
|
return result
|
|
|
|
except subprocess.TimeoutExpired:
|
|
if process:
|
|
process.kill()
|
|
_, stderr = process.communicate()
|
|
raise
|
|
|
|
finally:
|
|
if process:
|
|
self.remove_process(process)
|