From a33cdae4c3d7e88e05febe03c69026bc31eea39d Mon Sep 17 00:00:00 2001 From: Tulir Asokan Date: Fri, 16 Oct 2020 17:11:26 +0300 Subject: [PATCH] Add missing parameter to get_user --- mautrix_telegram/__main__.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/mautrix_telegram/__main__.py b/mautrix_telegram/__main__.py index 53322931..11c2ebd8 100644 --- a/mautrix_telegram/__main__.py +++ b/mautrix_telegram/__main__.py @@ -111,8 +111,11 @@ class TelegramBridge(Bridge): self.manhole.close() self.manhole = None - async def get_user(self, user_id: UserID) -> User: - return await User.get_by_mxid(user_id).ensure_started() + async def get_user(self, user_id: UserID, create: bool = True) -> User: + user = User.get_by_mxid(user_id, create=create) + if user: + await user.ensure_started() + return user async def get_portal(self, room_id: RoomID) -> Portal: return Portal.get_by_mxid(room_id)