iartamonov/mautrix-telegram
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Catch individual errors when syncing dialogs

Browse Source
This commit is contained in:
Tulir Asokan
2020-08-06 20:42:19 +03:00
parent eae7bba649
commit 321a7810c4
1 changed files with 12 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mautrix_telegram/user.py
+12 -3
View File
@@ -334,6 +334,12 @@ class User(AbstractUser, BaseUser):
return await self._search_remote(query), True return await self._search_remote(query), True
async def _catch(self, action: str, task: asyncio.Task) -> None:
try:
await task
except Exception:
self.log.exception(f"Error while {action}")
async def sync_dialogs(self) -> None: async def sync_dialogs(self) -> None:
if self.is_bot: if self.is_bot:
return return
@@ -361,11 +367,14 @@ class User(AbstractUser, BaseUser):
if portal.mxid: if portal.mxid:
update_task = portal.update_matrix_room(self, entity) update_task = portal.update_matrix_room(self, entity)
backfill_task = portal.backfill(self, last_id=dialog.message.id) backfill_task = portal.backfill(self, last_id=dialog.message.id)
creators.append(self.loop.create_task(update_task)) creators.append(self._catch(f"updating {portal.tgid_log}",
creators.append(self.loop.create_task(backfill_task)) self.loop.create_task(update_task)))
creators.append(self._catch(f"backfilling {portal.tgid_log}",
self.loop.create_task(backfill_task)))
elif not create_limit or index < create_limit: elif not create_limit or index < create_limit:
create_task = portal.create_matrix_room(self, entity, invites=[self.mxid]) create_task = portal.create_matrix_room(self, entity, invites=[self.mxid])
creators.append(self.loop.create_task(create_task)) creators.append(self._catch(f"creating {portal.tgid_log}",
self.loop.create_task(create_task)))
index += 1 index += 1
await self.save(portals=True) await self.save(portals=True)
await asyncio.gather(*creators) await asyncio.gather(*creators)