mirror of
https://github.com/pacnpal/Pac-cogs.git
synced 2025-12-20 02:41:06 -05:00
re-structure
This commit is contained in:
@@ -2,7 +2,7 @@
|
||||
from redbot.core.bot import Red
|
||||
import asyncio
|
||||
import logging
|
||||
from .video_archiver import VideoArchiver
|
||||
from .core.base import VideoArchiver
|
||||
from .exceptions import ProcessingError
|
||||
|
||||
logger = logging.getLogger("VideoArchiver")
|
||||
@@ -10,7 +10,7 @@ logger = logging.getLogger("VideoArchiver")
|
||||
async def setup(bot: Red) -> None:
|
||||
"""Load VideoArchiver."""
|
||||
try:
|
||||
# Load main cog first
|
||||
# Load main cog
|
||||
cog = VideoArchiver(bot)
|
||||
await bot.add_cog(cog)
|
||||
|
||||
@@ -22,23 +22,7 @@ async def setup(bot: Red) -> None:
|
||||
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:
|
||||
if not cog.ready.is_set():
|
||||
logger.error("VideoArchiver failed to initialize")
|
||||
await bot.remove_cog(cog.__class__.__name__)
|
||||
raise ProcessingError("Initialization failed")
|
||||
@@ -50,11 +34,6 @@ async def setup(bot: Red) -> None:
|
||||
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:
|
||||
|
||||
@@ -1,4 +1,4 @@
|
||||
"""Re-export commands from core.commands"""
|
||||
from .core.commands import VideoArchiverCommands
|
||||
"""Re-export commands from core.base"""
|
||||
from .core.base import VideoArchiver
|
||||
|
||||
__all__ = ['VideoArchiverCommands']
|
||||
__all__ = ['VideoArchiver']
|
||||
|
||||
@@ -1,7 +0,0 @@
|
||||
"""VideoArchiver cog for Red-DiscordBot"""
|
||||
|
||||
from .core import VideoArchiver
|
||||
|
||||
def setup(bot):
|
||||
"""Load VideoArchiver cog."""
|
||||
bot.add_cog(VideoArchiver(bot))
|
||||
Reference in New Issue
Block a user