mirror of
https://github.com/pacnpal/Pac-cogs.git
synced 2025-12-20 02:41:06 -05:00
247381fc8a
from videoarchiver.utils import progress_tracker Updated the main init.py to properly handle module reloading: Added 'videoarchiver.processor' and 'videoarchiver.processor.core' to modules_to_reload Added explicit import and reload of the processor module These changes ensure that: The progress_tracker is properly imported from the utils package All necessary modules are reloaded during initialization Circular imports are avoided The package structure maintains proper dependency flow
130 lines
3.2 KiB
Python
130 lines
3.2 KiB
Python
"""VideoArchiver cog for Red-DiscordBot"""
|
|
|
|
import sys
|
|
import asyncio
|
|
import logging
|
|
import importlib
|
|
from typing import Optional
|
|
from redbot.core.bot import Red
|
|
|
|
# Force reload of all modules
|
|
modules_to_reload = [
|
|
'videoarchiver.utils.exceptions',
|
|
'videoarchiver.utils',
|
|
'videoarchiver.processor',
|
|
'videoarchiver.processor.core',
|
|
'videoarchiver.core.commands.settings_commands',
|
|
'videoarchiver.core.commands.archiver_commands',
|
|
'videoarchiver.core.commands.database_commands'
|
|
]
|
|
|
|
# Remove modules to force fresh import
|
|
for module in modules_to_reload:
|
|
if module in sys.modules:
|
|
del sys.modules[module]
|
|
|
|
# Import and reload utils
|
|
from . import utils
|
|
importlib.reload(utils)
|
|
|
|
# Import and reload processor
|
|
from . import processor
|
|
importlib.reload(processor)
|
|
|
|
from .core.base import VideoArchiver
|
|
from .core.initialization import initialize_cog, init_callback
|
|
from .core.cleanup import cleanup_resources
|
|
from .utils.exceptions import (
|
|
VideoArchiverError,
|
|
CommandError,
|
|
EventError,
|
|
CogError,
|
|
ErrorContext,
|
|
ErrorSeverity,
|
|
ProcessingError
|
|
)
|
|
from .database import VideoArchiveDB
|
|
from .ffmpeg import FFmpegManager
|
|
from .queue import EnhancedVideoQueueManager
|
|
from .processor import VideoProcessor
|
|
from .config_manager import ConfigManager
|
|
from .update_checker import UpdateChecker
|
|
|
|
logger = logging.getLogger("VideoArchiver")
|
|
|
|
# Track initialization task
|
|
_init_task: Optional[asyncio.Task] = None
|
|
|
|
# Version information
|
|
__version__ = "1.0.0"
|
|
__author__ = "VideoArchiver Team"
|
|
__description__ = "Video archiving cog for Red-DiscordBot"
|
|
|
|
async def setup(bot: Red) -> None:
|
|
"""Load VideoArchiver with proper initialization."""
|
|
try:
|
|
# Create cog instance
|
|
cog = VideoArchiver(bot)
|
|
|
|
# Start initialization in background
|
|
global _init_task
|
|
_init_task = asyncio.create_task(initialize_cog(cog))
|
|
_init_task.add_done_callback(lambda t: init_callback(cog, t))
|
|
|
|
# Add cog to bot
|
|
await bot.add_cog(cog)
|
|
|
|
logger.info("VideoArchiver cog loaded successfully")
|
|
|
|
except Exception as e:
|
|
logger.error(f"Failed to load VideoArchiver: {str(e)}")
|
|
if _init_task and not _init_task.done():
|
|
_init_task.cancel()
|
|
raise
|
|
|
|
|
|
async def teardown(bot: Red) -> None:
|
|
"""Clean up when unloading."""
|
|
try:
|
|
# Cancel initialization if still running
|
|
if _init_task and not _init_task.done():
|
|
_init_task.cancel()
|
|
|
|
# Remove cog and clean up resources
|
|
if "VideoArchiver" in bot.cogs:
|
|
cog = bot.get_cog("VideoArchiver")
|
|
if cog:
|
|
await cleanup_resources(cog)
|
|
await bot.remove_cog("VideoArchiver")
|
|
|
|
logger.info("VideoArchiver cog unloaded successfully")
|
|
|
|
except Exception as e:
|
|
logger.error(f"Error during teardown: {str(e)}")
|
|
raise
|
|
|
|
|
|
__all__ = [
|
|
# Core classes
|
|
"VideoArchiver",
|
|
"VideoArchiveDB",
|
|
"FFmpegManager",
|
|
"EnhancedVideoQueueManager",
|
|
"VideoProcessor",
|
|
"ConfigManager",
|
|
"UpdateChecker",
|
|
|
|
# Base exceptions
|
|
"VideoArchiverError",
|
|
"CommandError",
|
|
"EventError",
|
|
"CogError",
|
|
"ErrorContext",
|
|
"ErrorSeverity",
|
|
"ProcessingError",
|
|
|
|
# Setup functions
|
|
"setup",
|
|
"teardown"
|
|
]
|