diff --git a/litecord/presence.py b/litecord/presence.py index 9b53c4a..a8c267e 100644 --- a/litecord/presence.py +++ b/litecord/presence.py @@ -78,11 +78,11 @@ async def _pres(storage, user_id: int, status_obj: dict) -> dict: class PresenceManager: """Presence related functions.""" - def __init__(self, storage, user_storage, state_manager, dispatcher): - self.storage = storage - self.user_storage = user_storage - self.state_manager = state_manager - self.dispatcher = dispatcher + def __init__(self, app): + self.storage = app.storage + self.user_storage = app.user_storage + self.state_manager = app.state_manager + self.dispatcher = app.dispatcher async def guild_presences(self, member_ids: List[int], guild_id: int) -> List[Dict[Any, str]]: diff --git a/run.py b/run.py index 4413b6e..7210469 100644 --- a/run.py +++ b/run.py @@ -227,12 +227,7 @@ def init_app_managers(app_): app_.icons = IconManager(app) app_.dispatcher = EventDispatcher(app) - - # TODO: only pass app - app_.presence = PresenceManager( - app.storage, app.user_storage, - app.state_manager, app.dispatcher - ) + app_.presence = PresenceManager(app_) app_.storage.presence = app.presence