Zergling_man/moment
Zergling_man/moment
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

NioCallbacks: add self.models/user_id aliases

Browse Source
This commit is contained in:
miruka 2020-05-17 13:57:09 -04:00
parent 14ffe0ba0b
commit f23a83b0d5
1 changed files with 13 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
src/backend/nio_callbacks.py
View File

@ -38,6 +38,8 @@ class NioCallbacks:
def __post_init__(self) -> None: def __post_init__(self) -> None:
"""Register our methods as callbacks.""" """Register our methods as callbacks."""
self.models = self.client.models
for name, class_ in utils.classes_defined_in(nio.responses).items(): for name, class_ in utils.classes_defined_in(nio.responses).items():
with suppress(AttributeError): with suppress(AttributeError):
method = getattr(self, f"on{name}") method = getattr(self, f"on{name}")
@ -53,10 +55,15 @@ class NioCallbacks:
) )
@property
def user_id(self) -> str:
return self.client.user_id
# Response callbacks # Response callbacks
async def onSyncResponse(self, resp: nio.SyncResponse) -> None: async def onSyncResponse(self, resp: nio.SyncResponse) -> None:
room_model = self.client.models[self.client.user_id, "rooms"] room_model = self.models[self.user_id, "rooms"]
for room_id, info in resp.rooms.join.items(): for room_id, info in resp.rooms.join.items():
if room_id not in room_model: if room_id not in room_model:
@ -89,7 +96,7 @@ class NioCallbacks:
self.client.first_sync_done.set() self.client.first_sync_done.set()
self.client.first_sync_date = datetime.now() self.client.first_sync_date = datetime.now()
account = self.client.models["accounts"][self.client.user_id] account = self.models["accounts"][self.user_id]
account.first_sync_done = True account.first_sync_done = True
@ -160,7 +167,7 @@ class NioCallbacks:
async def onRedactionEvent(self, room, ev) -> None: async def onRedactionEvent(self, room, ev) -> None:
model = self.client.models[self.client.user_id, room.room_id, "events"] model = self.models[self.user_id, room.room_id, "events"]
event = None event = None
for existing in model._sorted_data: for existing in model._sorted_data:
@ -325,8 +332,8 @@ class NioCallbacks:
if changed: if changed:
# Update our account profile if the event is newer than last update # Update our account profile if the event is newer than last update
if ev.state_key == self.client.user_id: if ev.state_key == self.user_id:
account = self.client.models["accounts"][self.client.user_id] account = self.models["accounts"][self.user_id]
if account.profile_updated < ev_date: if account.profile_updated < ev_date:
account.profile_updated = ev_date account.profile_updated = ev_date
@ -453,7 +460,7 @@ class NioCallbacks:
room_id = room.room_id room_id = room.room_id
room_item = self.client.models[self.client.user_id, "rooms"][room_id] room_item = self.models[self.user_id, "rooms"][room_id]
room_item.typing_members = sorted( room_item.typing_members = sorted(
room.user_name(user_id) or user_id for user_id in ev.users room.user_name(user_id) or user_id for user_id in ev.users