Как брать код подтверждения пользователя исходя из его ввода в чат?

Question

[k1nster]

Как брать код подтверждения для подключения телефона из чата в боте Telethon?

Я уже писал подобный вопрос, но требуется помощь в другом. После подвязки телефона, телеграмм требует код подтверждения. Этот код у меня сейчас вводится в консоле, мне нужно реализовать так, дабы пользователь его вводил в боте.

Я знаю как работать с состояниями в телеграмме и т.д, но мне не понятен сам метод user_client.start(phone), как в нём брать код подтверждения, а так же пароль от телеграмма (если он есть) именно из чата?

# Main telegram bot imports
import asyncio
from aiogram import Bot, Dispatcher, F, types
from aiogram.types import Message, CallbackQuery, InputFile
from aiogram.fsm.context import FSMContext
from aiogram.fsm.state import State, StatesGroup
from telethon import TelegramClient, utils
from telethon.errors.rpcerrorlist import PhoneNumberInvalidError

# Import BD
import pymysql

# Import files
import keyboard as kb
from config import TOKEN, API_ID, API_HASH
from texts import MESSAGES
from database import create_connection


api_id = API_ID
api_hash = API_HASH
bot = Bot(token=TOKEN)
dp = Dispatcher()
client = TelegramClient('Kinster', api_id, api_hash)

connection = create_connection()

# DB function
# Function to retrieve user information from the database
# def get_user_info(user_id):
#     with connection.cursor() as cursor:
#         select_user_query = "SELECT * FROM fiverr WHERE user_id = %s"
#         cursor.execute(select_user_query, (user_id,))
#         user_info = cursor.fetchone()
#     return user_info


class Form(StatesGroup):
    first_connection = State()

# Start function
@dp.message(F.text == '/start')
async def cmd_start(message:Message):
    user_name = message.from_user.first_name # get user usermame
    await message.answer(MESSAGES['welcome'].format(name=user_name), reply_markup=kb.main, parse_mode='Markdown')


# @dp.callback_query(F.data == 'first_connection')
# async def first_connection(callback: CallbackQuery):
#     await callback.message.answer(MESSAGES['first_connection'], parse_mode='Markdown')

@dp.message(F.text.startswith('/connect'))
async def cmd_start(message: Message, state: FSMContext):
    # Получаем текст сообщения и разделяем его на части по пробелу
    text = message.text.strip()
    parts = text.split(' ')
    
    if len(parts) == 1:
        # Пользователь отправил только команду /connect, отправляем инструкцию
        await message.answer(MESSAGES['first_connection'], parse_mode='Markdown')
    elif len(parts) == 2:
        phone = parts[1]
        # Попытка авторизации
        try:
            await message.reply(MESSAGES['phone_handler'])
            # Авторизуемся с использованием сессии, связанной с аккаунтом пользователя
            user_client = TelegramClient(str(message.from_user.id), api_id, api_hash)
            await user_client.start(phone)
            
            # Ожидание кода подтверждения от пользователя
            await message.answer("Пожалуйста, введите код подтверждения:")
            await state.set_state(Form.first_connection)
            
            await user_client.disconnect()
        except PhoneNumberInvalidError:
            await message.answer("❌ The phone number is invalid.")


# Тут будет состояние обрабатывающие сообщение

# Основной цикл бота
async def main():
    await dp.start_polling(bot)

if __name__ == '__main__':
    asyncio.run(main())






# Run
async def main():
    await dp.start_polling(bot)

if __name__ == '__main__':
    asyncio.run(main())

Answer 1

[soremix]

Если по простому:

1. Создаете объекта класса TelegramClient
2. Шлете смс через send_code_request
3. Заходите через sign_in

Comments

[k1nster]

Мне кажется я делаю что-то не так, т.к он всё равно запрашивает код от телефона у меня в консоле). После ввода кода в бота - ничего не происходит всё так же). У меня даже сообщение, которое после await user_client.start(phone) не отслылается пользователю.

Работаю с этой библиотекой впервые, по этому задаю столько вопросов xD

@dp.message(F.text.startswith('/connect'))
async def cmd_start(message: Message, state: FSMContext):
    # Получаем текст сообщения и разделяем его на части по пробелу
    text = message.text.strip()
    parts = text.split(' ')
    
    if len(parts) == 1:
        # Пользователь отправил только команду /connect, отправляем инструкцию
        await message.answer(MESSAGES['first_connection'], parse_mode='Markdown')
    elif len(parts) == 2:
        phone = parts[1]
        await state.update_data(phone = phone)
        # Попытка авторизации
        try:
            await message.reply(MESSAGES['phone_handler'])
            # Авторизуемся с использованием сессии, связанной с аккаунтом пользователя
            user_client = TelegramClient(str(message.from_user.id), api_id, api_hash)
            await user_client.start(phone)
            await message.answer('Please enter the login code sent to telegram app or sms (no spaces).\n\nFor example, your login code is 123456, then enter mycode123456')
            sent = await client.send_code_request(phone)
            print(sent)
            await state.set_state(Form.first_connection)
            await user_client.disconnect()
        except PhoneNumberInvalidError:
            await message.answer("❌ The phone number is invalid.")

@dp.message(F.text.startswith('/chat'))
async def cmd_start(message: Message):
    
    phone = 'тут мой телефон'
    
    await client.start(phone)
    
    # get group and channel list
    dialogs = await client.get_dialogs()

    # print group and channel data
    for dialog in dialogs:
        if dialog.is_group or dialog.is_channel:
            print(f"{dialog.name}: {dialog.id}")

# Тут будет состояние обрабатывающие сообщение
@dp.message(Form.first_connection)
async def cmd_phone(message: Message, state: FSMContext):
    data = await state.get_data()
    phone = data['phone']
    code = int(message.text)
    
    await client.sign_in(phone, code)
# Основной цикл бота
async def main():
    await dp.start_polling(bot)

[soremix]

k1nster, .start() не надо вызывать

Сначала send_code_request, затем sign_in. Всё.

[k1nster]

soremix, так я тоже пробывал, получаю ошибку из ряда "Невозможно отправить запрос при отключенном соединении":
ConnectionError: Cannot send requests while disconnected

@dp.message(F.text.startswith('/connect'))
async def cmd_start(message: Message, state: FSMContext):
    # Получаем текст сообщения и разделяем его на части по пробелу
    text = message.text.strip()
    parts = text.split(' ')
    
    if len(parts) == 1:
        # Пользователь отправил только команду /connect, отправляем инструкцию
        await message.answer(MESSAGES['first_connection'], parse_mode='Markdown')
    elif len(parts) == 2:
        phone = parts[1]
        await state.update_data(phone = phone)
        # Попытка авторизации
        try:
            await message.reply(MESSAGES['phone_handler'])
            # Авторизуемся с использованием сессии, связанной с аккаунтом пользователя
            user_client = TelegramClient(str(message.from_user.id), api_id, api_hash)
            await message.answer('Please enter the login code sent to telegram app or sms (no spaces).\n\nFor example, your login code is 123456, then enter mycode123456')
            await user_client.send_code_request(phone)
            await state.set_state(Form.first_connection)
        except PhoneNumberInvalidError:
            await message.answer("❌ The phone number is invalid.")

# Тут будет состояние обрабатывающие сообщение
@dp.message(Form.first_connection)
async def cmd_phone(message: Message, state: FSMContext):
    data = await state.get_data()
    phone = data['phone']
    code = int(message.text)

    await client.sign_in(phone, code)

[soremix]

k1nster, перед этим вызвать
await user_client.connect()

[k1nster]

soremix, Хорошо. Код у меня действительно отправляется, но после его ввода получаю снова ошибку: ConnectionError: Cannot send requests while disconnected.

Перепробовал уже тучу способов, никак не могу исправить. Возможно знаете почему она возникает?

[k1nster]

Скорее всего я как-то через состояние AIOGRAM это неправильно передаю). Но вот решения так и не нашёл/

[k1nster]

soremix,