From c3a2d2e9f0e7dea95ab5be21b2673afe57511e62 Mon Sep 17 00:00:00 2001 From: AlberLC Date: Tue, 27 Jun 2023 07:20:55 +0200 Subject: [PATCH] Fix context manager in _get_contacts_ids for telegram userbots --- flanabot/bots/flana_tele_bot.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/flanabot/bots/flana_tele_bot.py b/flanabot/bots/flana_tele_bot.py index d7d0be0..666e392 100644 --- a/flanabot/bots/flana_tele_bot.py +++ b/flanabot/bots/flana_tele_bot.py @@ -8,7 +8,7 @@ from typing import Callable import telethon.tl.functions from flanautils import Media, OrderedSet -from multibot import TelegramBot, find_message, user_client +from multibot import TelegramBot, find_message, use_user_client, user_client from flanabot import constants from flanabot.bots.flana_bot import FlanaBot @@ -48,7 +48,7 @@ class FlanaTeleBot(TelegramBot, FlanaBot): # -------------------------------------------------------- # @user_client async def _get_contacts_ids(self) -> list[int]: - async with self.user_client: + async with use_user_client(self): contacts_data = await self.user_client(telethon.tl.functions.contacts.GetContactsRequest(hash=0)) return [contact.user_id for contact in contacts_data.contacts]