diff --git a/videoarchiver/__init__.py b/videoarchiver/__init__.py index e820e48..bce7718 100644 --- a/videoarchiver/__init__.py +++ b/videoarchiver/__init__.py @@ -17,9 +17,49 @@ modules_to_reload = [ 'videoarchiver.queue', 'videoarchiver.queue.models', 'videoarchiver.queue.manager', - 'videoarchiver.core.commands.settings_commands', + 'videoarchiver.queue.cleaners', + 'videoarchiver.queue.cleaners.guild_cleaner', + 'videoarchiver.queue.cleaners.history_cleaner', + 'videoarchiver.queue.cleaners.tracking_cleaner', + 'videoarchiver.queue.monitoring', + 'videoarchiver.queue.recovery_manager', + 'videoarchiver.queue.state_manager', + 'videoarchiver.ffmpeg', + 'videoarchiver.ffmpeg.binary_manager', + 'videoarchiver.ffmpeg.encoder_params', + 'videoarchiver.ffmpeg.exceptions', + 'videoarchiver.ffmpeg.ffmpeg_downloader', + 'videoarchiver.ffmpeg.ffmpeg_manager', + 'videoarchiver.ffmpeg.gpu_detector', + 'videoarchiver.ffmpeg.process_manager', + 'videoarchiver.ffmpeg.verification_manager', + 'videoarchiver.ffmpeg.video_analyzer', + 'videoarchiver.database', + 'videoarchiver.database.connection_manager', + 'videoarchiver.database.query_manager', + 'videoarchiver.database.schema_manager', + 'videoarchiver.database.video_archive_db', + 'videoarchiver.config', + 'videoarchiver.config.channel_manager', + 'videoarchiver.config.exceptions', + 'videoarchiver.config.role_manager', + 'videoarchiver.config.settings_formatter', + 'videoarchiver.config.validation_manager', + 'videoarchiver.core', + 'videoarchiver.core.base', + 'videoarchiver.core.cleanup', + 'videoarchiver.core.commands', 'videoarchiver.core.commands.archiver_commands', - 'videoarchiver.core.commands.database_commands' + 'videoarchiver.core.commands.database_commands', + 'videoarchiver.core.commands.settings_commands', + 'videoarchiver.core.component_manager', + 'videoarchiver.core.error_handler', + 'videoarchiver.core.events', + 'videoarchiver.core.guild', + 'videoarchiver.core.initialization', + 'videoarchiver.core.lifecycle', + 'videoarchiver.core.response_handler', + 'videoarchiver.core.settings' ] # Remove modules to force fresh import @@ -39,6 +79,22 @@ importlib.reload(videoarchiver.processor) import videoarchiver.queue importlib.reload(videoarchiver.queue) +# Import and reload ffmpeg +import videoarchiver.ffmpeg +importlib.reload(videoarchiver.ffmpeg) + +# Import and reload database +import videoarchiver.database +importlib.reload(videoarchiver.database) + +# Import and reload config +import videoarchiver.config +importlib.reload(videoarchiver.config) + +# Import and reload core +import videoarchiver.core +importlib.reload(videoarchiver.core) + from .core.base import VideoArchiver from .core.initialization import initialize_cog, init_callback from .core.cleanup import cleanup_resources diff --git a/videoarchiver/queue/__init__.py b/videoarchiver/queue/__init__.py index c827d04..8b124a2 100644 --- a/videoarchiver/queue/__init__.py +++ b/videoarchiver/queue/__init__.py @@ -14,6 +14,14 @@ from .health_checker import HealthChecker # Importing from cleaners subdirectory from .cleaners import GuildCleaner, HistoryCleaner, TrackingCleaner +# Corrected relative imports from utils +from ..utils.compression_handler import CompressionHandler +from ..utils.directory_manager import DirectoryManager +from ..utils.download_manager import DownloadManager +from ..utils.file_operations import FileOperations +from ..utils.progress_tracker import ProgressTracker +from ..utils.url_validator import UrlValidator + __all__ = [ 'QueueItem', 'QueueMetrics', @@ -32,4 +40,10 @@ __all__ = [ 'GuildCleaner', 'HistoryCleaner', 'TrackingCleaner', + 'CompressionHandler', + 'DirectoryManager', + 'DownloadManager', + 'FileOperations', + 'ProgressTracker', + 'UrlValidator', ]