spawnrys/flanabot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix context manager in _get_contacts_ids for telegram userbots

Browse Source
This commit is contained in:
AlberLC
2023-06-27 07:20:55 +02:00
parent bb9b5caa55
commit c3a2d2e9f0
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
flanabot/bots/flana_tele_bot.py
View File

@@ -8,7 +8,7 @@ from typing import Callable
import telethon.tl.functions import telethon.tl.functions
from flanautils import Media, OrderedSet 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 import constants
from flanabot.bots.flana_bot import FlanaBot from flanabot.bots.flana_bot import FlanaBot
@@ -48,7 +48,7 @@ class FlanaTeleBot(TelegramBot, FlanaBot):
# -------------------------------------------------------- # # -------------------------------------------------------- #
@user_client @user_client
async def _get_contacts_ids(self) -> list[int]: 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)) contacts_data = await self.user_client(telethon.tl.functions.contacts.GetContactsRequest(hash=0))
return [contact.user_id for contact in contacts_data.contacts] return [contact.user_id for contact in contacts_data.contacts]