mirror of
https://github.com/pacnpal/Pac-cogs.git
synced 2025-12-20 02:41:06 -05:00
971d52bd00
Core processing logic in VideoProcessor class Dedicated message handling in MessageHandler Queue operations in QueueHandler Progress tracking in ProgressTracker Reaction management in reactions.py Improved Maintainability: Each component has a single responsibility Easier to test individual components Better code organization and readability Reduced file sizes for better version control Better Resource Management: Centralized progress tracking Isolated queue operations Cleaner cleanup processes Optimized Imports: Components can be imported individually as needed Main processor.py provides backward compatibility Clear module interface through init.py
89 lines
3.6 KiB
Python
89 lines
3.6 KiB
Python
"""Progress tracking for video downloads and compression"""
|
|
|
|
from typing import Dict, Any
|
|
from datetime import datetime
|
|
|
|
class ProgressTracker:
|
|
"""Tracks progress of video downloads and compression operations"""
|
|
|
|
def __init__(self):
|
|
self._download_progress: Dict[str, Dict[str, Any]] = {}
|
|
self._compression_progress: Dict[str, Dict[str, Any]] = {}
|
|
|
|
def update_download_progress(self, url: str, progress_data: Dict[str, Any]) -> None:
|
|
"""Update download progress for a specific URL"""
|
|
if url not in self._download_progress:
|
|
self._download_progress[url] = {
|
|
'active': True,
|
|
'start_time': datetime.utcnow().isoformat(),
|
|
'retries': 0
|
|
}
|
|
|
|
self._download_progress[url].update(progress_data)
|
|
|
|
def complete_download(self, url: str) -> None:
|
|
"""Mark a download as complete"""
|
|
if url in self._download_progress:
|
|
self._download_progress[url]['active'] = False
|
|
self._download_progress[url]['completed_time'] = datetime.utcnow().isoformat()
|
|
|
|
def increment_download_retries(self, url: str) -> None:
|
|
"""Increment retry count for a download"""
|
|
if url in self._download_progress:
|
|
self._download_progress[url]['retries'] = self._download_progress[url].get('retries', 0) + 1
|
|
|
|
def update_compression_progress(self, file_id: str, progress_data: Dict[str, Any]) -> None:
|
|
"""Update compression progress for a specific file"""
|
|
if file_id not in self._compression_progress:
|
|
self._compression_progress[file_id] = {
|
|
'active': True,
|
|
'start_time': datetime.utcnow().isoformat()
|
|
}
|
|
|
|
self._compression_progress[file_id].update(progress_data)
|
|
|
|
def complete_compression(self, file_id: str) -> None:
|
|
"""Mark a compression operation as complete"""
|
|
if file_id in self._compression_progress:
|
|
self._compression_progress[file_id]['active'] = False
|
|
self._compression_progress[file_id]['completed_time'] = datetime.utcnow().isoformat()
|
|
|
|
def get_download_progress(self, url: str = None) -> Dict[str, Any]:
|
|
"""Get download progress for a specific URL or all downloads"""
|
|
if url:
|
|
return self._download_progress.get(url, {})
|
|
return self._download_progress
|
|
|
|
def get_compression_progress(self, file_id: str = None) -> Dict[str, Any]:
|
|
"""Get compression progress for a specific file or all compressions"""
|
|
if file_id:
|
|
return self._compression_progress.get(file_id, {})
|
|
return self._compression_progress
|
|
|
|
def clear_completed(self) -> None:
|
|
"""Clear completed operations from tracking"""
|
|
# Clear completed downloads
|
|
self._download_progress = {
|
|
url: data for url, data in self._download_progress.items()
|
|
if data.get('active', False)
|
|
}
|
|
|
|
# Clear completed compressions
|
|
self._compression_progress = {
|
|
file_id: data for file_id, data in self._compression_progress.items()
|
|
if data.get('active', False)
|
|
}
|
|
|
|
def get_active_operations(self) -> Dict[str, Dict[str, Any]]:
|
|
"""Get all active operations"""
|
|
return {
|
|
'downloads': {
|
|
url: data for url, data in self._download_progress.items()
|
|
if data.get('active', False)
|
|
},
|
|
'compressions': {
|
|
file_id: data for file_id, data in self._compression_progress.items()
|
|
if data.get('active', False)
|
|
}
|
|
}
|