From 1567239ae67ef114d9d63135a09621749fa52032 Mon Sep 17 00:00:00 2001 From: Tulir Asokan Date: Wed, 28 Oct 2020 18:44:50 +0200 Subject: [PATCH] Update connection metric after logging in --- mautrix_telegram/user.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/mautrix_telegram/user.py b/mautrix_telegram/user.py index 2d553f0c..394dee35 100644 --- a/mautrix_telegram/user.py +++ b/mautrix_telegram/user.py @@ -204,8 +204,6 @@ class User(AbstractUser, BaseUser): if await self.is_logged_in(): self.log.debug(f"Ensuring post_login() for {self.name}") self.loop.create_task(self.post_login()) - if config["metrics.enabled"]: - self._track_connection_task = self.loop.create_task(self._track_connection()) elif delete_unless_authenticated: self.log.debug(f"Unauthenticated user {self.name} start()ed, deleting session...") await self.client.disconnect() @@ -228,6 +226,9 @@ class User(AbstractUser, BaseUser): self._track_metric(METRIC_CONNECTED, False) async def post_login(self, info: TLUser = None, first_login: bool = False) -> None: + if config["metrics.enabled"] and not self._track_connection_task: + self._track_connection_task = self.loop.create_task(self._track_connection()) + try: await self.update_info(info) except Exception: