Как можно передать данные из одной функции в другую в Playwright?

Я использую библиотеку Playwright для асинхронного запуска нескольких сессий и aiogram 3 для получения данных через телеграм-бота. Я пытаюсь реализовать процесс входа на сайт через SMS-код. Для этого мне нужно сначала вызвать функцию, которая запрашивает у пользователя номер телефона, затем в следующей функции обработать этот номер, после чего получить SMS-код от пользователя и в другой функции с помощью состояния обработать его и ввести на сайте. Я не знаю, как в Playwright реализовать ожидание и последующую отправку SMS-кода. Можете помочь с этой задачей?
  • 30 декабря 2024 г. 12:43
Ответы на вопрос 2
Конечно! Чтобы передать данные между функциями в Playwright и использовать их в асинхронном контексте, вы можете воспользоваться несколькими подходами. Я расскажу о простом подходе, используя замыкания и асинхронные функции.

### Пример реализации

Предположим, у вас есть несколько функций: одна для запроса номера телефона, вторая для отправки SMS-кода, и третья для ввода кода на сайте. Мы можем использовать замыкания, чтобы передавать данные между этими функциями.

```python
import asyncio
from playwright.async_api import async_playwright
from aiogram import Bot, Dispatcher, types

# Инициализация бота
API_TOKEN = 'YOUR_API_TOKEN'
bot = Bot(token=API_TOKEN)
dp = Dispatcher(bot)

# Глобальная переменная для хранения номера телефона и SMS-кода
user_data = {'phone_number': None, 'sms_code': None}

# Функция для запроса номера телефона
async def request_phone_number(chat_id):
    await bot.send_message(chat_id, "Введите ваш номер телефона:")
    # Здесь можно ожидать ответа пользователя
    # Используем состояние для обработки - пример с использованием aiogram
    @dp.message_handler(lambda message: message.from_user.id == chat_id)
    async def receive_phone_number(message: types.Message):
        user_data['phone_number'] = message.text
        await bot.send_message(chat_id, "Введите SMS-код:")
        await request_sms_code(chat_id)

# Функция для запроса SMS-кода
async def request_sms_code(chat_id):
    @dp.message_handler(lambda message: message.from_user.id == chat_id)
    async def receive_sms_code(message: types.Message):
        user_data['sms_code'] = message.text
        await process_login(chat_id)

# Функция для входа
async def process_login(chat_id):
    async with async_playwright() as p:
        browser = await p.chromium.launch()
        page = await browser.new_page()
        
        # Здесь логина на сайт
        await page.goto('http://example.com/login')
        
        # Ввод номер телефона и отправка
        await page.fill('input[name="phone"]', user_data['phone_number'])
        await page.click('button[type="submit"]')
        
        # Ввод SMS-кода и отправка
        await page.fill('input[name="sms_code"]', user_data['sms_code'])
        await page.click('button[type="submit"]')
        
        # Здесь вы можете продолжить обработку или проверить результаты входа
        await browser.close()

# Запуск бота и логики
async def main():
    # Это может быть ваш обработчик команд /start в aiogram, например
    @dp.message_handler(commands=['start'])
    async def start_command(message: types.Message):
        await request_phone_number(message.from_user.id)

    await dp.start_polling()

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

### Объяснение кода:
1. **Глобальные переменные**: Я использую `user_data` для хранения номера телефона и SMS-кода.
2. **Асинхронные функции**: Каждая функция отвечает за определенный этап процесса. `request_phone_number` запрашивает номер телефона, `request_sms_code` — SMS-код, а `process_login` выполняет вход на сайт.
3. **Пока ожидаем ответа пользователя**: Используется `@dp.message_handler`, чтобы обрабатывать входящие сообщения от конкретного пользователя, и каждая функция подписывается на новое сообщение.
4. **Состояние бота**: В случае более сложной логики, вы можете использовать состояния в aiogram для управления диалогами.

Этот подход позволяет хранить данные между асинхронными вызовами. Вы можете модифицировать пример, добавляя дополнительные проверки и обработку ошибок в зависимости от ваших требований.
Я разобрался кое-как, можно после создания страницы и браузера возвращать эти объекты из функции создании браузера и страницы, затем это объекты записывать в стейт (FSMContext) и уже после введенного кода делать вводить его на странице в стейте. 

import asyncio
from playwright.async_api import async_playwright
from aiogram import Bot, Dispatcher, types
from aiogram.contrib.fsm_storage.memory import MemoryStorage
from aiogram.dispatcher import FSMContext
from aiogram.dispatcher import State, StatesGroup

API_TOKEN = 'YOUR_TELEGRAM_BOT_TOKEN'

# Инициализация бота и диспетчера
bot = Bot(token=API_TOKEN)
storage = MemoryStorage()
dp = Dispatcher(bot, storage=storage)

# Определение состояний для FSM
class Form(StatesGroup):
    waiting_for_phone = State()
    waiting_for_code = State()

async def login_with_playwright(phone_number: str):
    async with async_playwright() as p:
        browser = await p.chromium.launch(headless=False)  # Установите headless=True для запуска без GUI
        page = await browser.new_page()
        
        await page.goto('URL_ВАШЕГО_САЙТА')
        
        # Ввод номера телефона
        await page.fill('input[name="phone"]', phone_number)
        await page.click('button#send-code')  # Замените селектор на ваш
        
        # Ждем, пока код придет (например, 30 секунд)
        await asyncio.sleep(30)  # Увеличьте время ожидания, если нужно
        
        return page  # Возвращаем страницу для дальнейшего использования

@dp.message_handler(commands=['start'])
async def cmd_start(message: types.Message):
    await Form.waiting_for_phone.set()
    await message.reply("Введите ваш номер телефона:")

@dp.message_handler(state=Form.waiting_for_phone)
async def process_phone(message: types.Message, state: FSMContext):
    phone_number = message.text
    await state.update_data(phone_number=phone_number)

    # Отправляем номер телефона на сайт и получаем страницу
    page = await login_with_playwright(phone_number)
    
    # После отправки номера сообщаем пользователю о необходимости ввести код
    await message.reply("Код был отправлен на ваш номер. Пожалуйста, введите его:")
    
    await state.update_data(page=page)  # Сохраняем страницу в состоянии
    await Form.waiting_for_code.set()

@dp.message_handler(state=Form.waiting_for_code)
async def process_code(message: types.Message, state: FSMContext):
    user_data = await state.get_data()
    phone_number = user_data.get('phone_number')
    sms_code = message.text
    page = user_data.get('page')

    # Здесь можно добавить логику для ввода кода на сайте
    await page.fill('input[name="code"]', sms_code)  # Замените селектор на ваш
    await page.click('button#submit-code')  # Замените селектор на ваш

    await state.finish()
    await message.reply("Вы успешно вошли в систему!")

if __name__ == '__main__':
    from aiogram import executor
    executor.start_polling(dp, skip_updates=True)
Похожие вопросы