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

Add MatriwClient.pause_while_offline()

Browse Source
This commit is contained in:
miruka 2020-07-10 12:46:22 -04:00
parent 5b9e84a9f7
commit 7580686598
1 changed files with 12 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
src/backend/matrix_client.py
View File

@ -424,6 +424,15 @@ class MatrixClient(nio.AsyncClient):
return (await self.content_repository_config()).upload_size return (await self.content_repository_config()).upload_size
async def pause_while_offline(self) -> None:
"""Block until our account is online."""
while (
self.models["accounts"][self.user_id].presence ==
Presence.State.offline
):
await asyncio.sleep(0.2)
async def can_kick(self, room_id: str, target_user_id: str) -> bool: async def can_kick(self, room_id: str, target_user_id: str) -> bool:
"""Return whether we can kick a certain user in a room.""" """Return whether we can kick a certain user in a room."""
@ -522,12 +531,7 @@ class MatrixClient(nio.AsyncClient):
mentions = mentions, mentions = mentions,
) )
while ( await self.pause_while_offline()
self.models["accounts"][self.user_id].presence ==
Presence.State.offline
):
await asyncio.sleep(0.2)
await self._send_message(room_id, content, tx_id) await self._send_message(room_id, content, tx_id)
@ -602,11 +606,7 @@ class MatrixClient(nio.AsyncClient):
monitor.on_transferred = on_transferred monitor.on_transferred = on_transferred
monitor.on_speed_changed = on_speed_changed monitor.on_speed_changed = on_speed_changed
while ( await self.pause_while_offline()
self.models["accounts"][self.user_id].presence ==
Presence.State.offline
):
await asyncio.sleep(0.2)
try: try:
url, mime, crypt_dict = await self.upload( url, mime, crypt_dict = await self.upload(
@ -1190,12 +1190,7 @@ class MatrixClient(nio.AsyncClient):
event.event_type = nio.RedactedEvent event.event_type = nio.RedactedEvent
while ( await self.pause_while_offline()
self.models["accounts"][self.user_id].presence ==
Presence.State.offline
):
await asyncio.sleep(0.2)
return await asyncio.gather(*tasks) return await asyncio.gather(*tasks)