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

View File

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