mirror of
https://github.com/pacnpal/Pac-cogs.git
synced 2025-12-20 02:41:06 -05:00
63 lines
2.2 KiB
Python
63 lines
2.2 KiB
Python
"""VideoArchiver cog for Red-DiscordBot"""
|
|
from redbot.core.bot import Red
|
|
import asyncio
|
|
import logging
|
|
from .video_archiver import VideoArchiver
|
|
from .exceptions import ProcessingError
|
|
|
|
logger = logging.getLogger("VideoArchiver")
|
|
|
|
async def setup(bot: Red) -> None:
|
|
"""Load VideoArchiver."""
|
|
try:
|
|
# Load main cog first
|
|
cog = VideoArchiver(bot)
|
|
await bot.add_cog(cog)
|
|
|
|
# Wait for initialization to complete with timeout
|
|
try:
|
|
await asyncio.wait_for(cog.ready.wait(), timeout=30)
|
|
except asyncio.TimeoutError:
|
|
logger.error("VideoArchiver initialization timed out")
|
|
await bot.remove_cog(cog.__class__.__name__)
|
|
raise ProcessingError("Initialization timed out")
|
|
|
|
# Only load commands if main cog initialized successfully
|
|
if cog.ready.is_set():
|
|
try:
|
|
from .commands import VideoArchiverCommands
|
|
commands_cog = VideoArchiverCommands(
|
|
bot,
|
|
cog.config_manager,
|
|
cog.update_checker,
|
|
cog.processor
|
|
)
|
|
await bot.add_cog(commands_cog)
|
|
except Exception as e:
|
|
logger.error(f"Failed to load commands cog: {str(e)}")
|
|
# Clean up main cog if commands fail to load
|
|
await bot.remove_cog(cog.__class__.__name__)
|
|
raise
|
|
else:
|
|
logger.error("VideoArchiver failed to initialize")
|
|
await bot.remove_cog(cog.__class__.__name__)
|
|
raise ProcessingError("Initialization failed")
|
|
|
|
except Exception as e:
|
|
logger.error(f"Failed to load VideoArchiver: {str(e)}")
|
|
raise
|
|
|
|
async def teardown(bot: Red) -> None:
|
|
"""Clean up when unloading."""
|
|
try:
|
|
# Remove commands cog first
|
|
if "VideoArchiverCommands" in bot.cogs:
|
|
await bot.remove_cog("VideoArchiverCommands")
|
|
|
|
# Then remove main cog
|
|
if "VideoArchiver" in bot.cogs:
|
|
await bot.remove_cog("VideoArchiver")
|
|
except Exception as e:
|
|
logger.error(f"Error during teardown: {str(e)}")
|
|
raise
|