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

Separate dialog syncing and creation limits and fix bugs

Browse Source
This commit is contained in:
Tulir Asokan
2020-07-29 21:55:51 +03:00
parent 59eb7376c9
commit 9848f8b92c
5 changed files with 32 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mautrix_telegram/user.py
+9 -5
View File
@@ -338,10 +338,13 @@ class User(AbstractUser, BaseUser):
if self.is_bot:
return
creators = []
limit = config["bridge.sync_dialog_limit"] or None
self.log.debug(f"Syncing dialogs (limit={limit})")
update_limit = config["bridge.sync_update_limit"] or None
create_limit = config["bridge.sync_create_limit"]
index = 0
self.log.debug(f"Syncing dialogs (update_limit={update_limit}, "
f"create_limit={create_limit})")
dialog: Dialog
async for dialog in self.client.iter_dialogs(limit=limit, ignore_migrated=True,
async for dialog in self.client.iter_dialogs(limit=update_limit, ignore_migrated=True,
archived=False):
entity = dialog.entity
if isinstance(entity, ChatForbidden):
@@ -357,12 +360,13 @@ class User(AbstractUser, BaseUser):
self.portals[portal.tgid_full] = portal
if portal.mxid:
update_task = portal.update_matrix_room(self, entity)
backfill_task = portal.backfill(self, last_known_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.loop.create_task(backfill_task))
else:
elif not create_limit or index < create_limit:
create_task = portal.create_matrix_room(self, entity, invites=[self.mxid])
creators.append(self.loop.create_task(create_task))
index += 1
self.save(portals=True)
await asyncio.gather(*creators)
self.log.debug("Dialog syncing complete")