fix: Add profiles module attribute.

This commit is contained in:
2025-09-04 04:14:56 +10:00
parent bcd19d8018
commit 996494ebb0
2 changed files with 12 additions and 0 deletions

View File

@@ -26,6 +26,7 @@ from .monitor import SystemMonitor, ComponentMonitor, StatusLevel, ComponentStat
if TYPE_CHECKING:
from core.cog import CoreCog
from modules.profiles.profiles.discord.cog import ProfilesCog
logger = logging.getLogger(__name__)
@@ -64,6 +65,10 @@ class LionBot(Bot):
def core(self):
return self.get_cog('CoreCog')
@property
def profiles(self):
return self.get_cog('ProfilesCog')
async def _monitor_status(self):
if self.is_closed():
level = StatusLevel.ERRORED
@@ -110,6 +115,10 @@ class LionBot(Bot):
def get_cog(self, name: Literal['CoreCog']) -> 'CoreCog':
...
@overload
def get_cog(self, name: Literal['ProfilesCog']) -> 'ProfilesCog':
...
@overload
def get_cog(self, name: str) -> Optional[Cog]:
...

View File

@@ -9,6 +9,7 @@ from discord.ext.commands import Context
if TYPE_CHECKING:
from .LionBot import LionBot
from modules.profiles.profiles.data import UserProfile, Community
logger = logging.getLogger(__name__)
@@ -46,6 +47,8 @@ class LionContext(Context['LionBot']):
Extends Context to add Lion-specific methods and attributes.
Also adds several contextual wrapped utilities for simpler user during command invocation.
"""
profile: 'UserProfile'
community: 'Community'
def __repr__(self):
parts = {}