Unregister old chat when a group is upgraded

This commit is contained in:
Tulir Asokan
2020-08-02 00:54:16 +03:00
parent bc4610af04
commit eace9b4ef6
4 changed files with 13 additions and 9 deletions
+4 -2
View File
@@ -372,6 +372,7 @@ class User(AbstractUser, BaseUser):
self.log.debug("Dialog syncing complete")
def register_portal(self, portal: po.Portal) -> None:
self.log.trace(f"Registering portal {portal.tgid_full}")
try:
if self.portals[portal.tgid_full] == portal:
return
@@ -380,9 +381,10 @@ class User(AbstractUser, BaseUser):
self.portals[portal.tgid_full] = portal
self.save(portals=True)
def unregister_portal(self, portal: po.Portal) -> None:
def unregister_portal(self, tgid: int, tg_receiver: int) -> None:
self.log.trace(f"Unregistering portal {(tgid, tg_receiver)}")
try:
del self.portals[portal.tgid_full]
del self.portals[(tgid, tg_receiver)]
self.save(portals=True)
except KeyError:
pass