Как правильно одновременно запустить двух Telegram ботов одной программой на Python?

Question

[Аркадий Гальченко]

Есть следующий код на двух билбиотеках: Aiogram, Pyrogram

from threading import Thread
from aiogram import Bot, Dispatcher, executor, types
from pyrogram import Client, filters


def pyrobot():
    print("pyro started")
    @app.on_message(filters.chat("some_chat"))
    async def print_pyrogram():
        print("Pyrogram")
    app.run()


def aiobot():
    print("aio started")
    @dp.message_handler(content_types=['any'])
    async def print_aiogram():
        print("Aiogram")
    executor.start_polling(dp, skip_updates=True)


# Данные и инициализация Юзербота
api_id = "..."
api_hash = "..."
phone_number = "..."
app = Client("Pyrobot", api_id, api_hash, phone_number)

# Данные и инициализация Бота
bot = Bot(token="...")
dp = Dispatcher(bot)

th_bot = Thread(target=pyrobot(), args=())
th_userbot = Thread(target=aiobot(), args=())

th_userbot.start()
th_bot.start()

*за переопределение методов и тд знаю, с этим потом, пока суть имнно запустить

При таком варианте все равно запускается первый бот. И пока он не завершит работу, второй просто ждёт, хотя вроде как два разных потока. Самое интересное - если в функции с "ботами" просто написать циклы, то они работают одновременно.

Моя задача ловить обновления двумя ботами одновременно или по очереди. Какие варианты реализации вы можете предложить?

Answer 1

[AlbertForest]

Глянь тут мой ответ

Comments

[Аркадий Гальченко]

Ну во первых у меня две разные библиотеки, во вторых - бот и юзербот. А самое главное не вижу гарантии, что не будет той же проблемы, что в моём коде. Но ради интереса попробую. Если я правильно понимаю, в моём случае в event_loop нужно запихнуть оба хендлера?

[AlbertForest]

Аркадий Гальченко, Да ты хоть 10 хендлеров из разных библиотек запихни в один event_loop. Это куда адекватнее чем запускать их в потоках. Они и так асинхронны. Зачем это извращение?
А ошибка у тебя вообще в синтаксисе, Vindicar об этом написал

[Аркадий Гальченко]

i3a4em, вопрос - как реализовать? Ответа нет. Твой вариант очень далёк от моего кода. Общие фразы "используй asyncio" я и так уже видел в других темах форумов. Я не понимаю как конкретно мне это сделать. Вариант с потоками мне просто показался более понятным, не утверждаю что он правильный. И после устранения ошибок он не работает.

[AlbertForest]

Аркадий Гальченко, напишу позже, если никто не ответит. Сейсас я не у компа.
И да. Ты не смотрел в сторону telethon? И боты и юзеры в одной либе. Удобно же.

[Аркадий Гальченко]

i3a4em, ок.
Telethon пробовал, не понравилось. В Pyrogram тоже аналогично, есть и боты и юзеры. Но больше всего нравится Aiogram, только она чисто под ботов.

Нужно мне вникать в асинхронность и многопоточность, пока сложно даётся.

[AlbertForest]

Твой вариант очень далёк от моего кода.

Сравни сам)

from aiogram import Bot, Dispatcher, executor, types
from pyrogram import Client, filters
import asyncio
    
def pyrobot(api_id, api_hash, phone_number):
    print("pyro started")
    app = Client("Pyrobot", api_id, api_hash, phone_number)
    
    @app.on_message(filters.chat("some_chat"))
    async def print_pyrogram():
        print("Pyrogram")
    app.run()


def aiobot(event_loop, token):
    bot = Bot(token=token)
    dp = Dispatcher(bot)
    print("aio started")
    
    @dp.message_handler(commands=['start'])
    async def process_start_command(message: types.Message):
        print("Aiogram")
    
    async def start_bot(dp):
        event_loop.create_task(dp.start_polling())
    
    event_loop.run_until_complete(start_bot(dp))

api_id = "..."
api_hash = "..."
phone_number = "..."
token="..."

if __name__ == '__main__':
    event_loop = asyncio.get_event_loop()
    aiobot(event_loop, token)
    pyrobot(api_id, api_hash, phone_number)
    event_loop.run_forever()

Telethon пробовал, не понравилось

Ну тут вопрос скорее не вкуса, а возможностей. Но в твоем случае если не понравилось то и не надо. Уже не помню суть проблемы, но когда-то выбрал его из-за более гибкого функционала какой-то нужной мне плюшки. С тех времен больше вообще ничем не пользуюсь. Если таких проблем не испытываешь - делай как нравиться.

Нужно мне вникать в асинхронность и многопоточность, пока сложно даётся.

Читай доки. Хоть по чуть-чуть. Да, это скучно. На хабре спросить и проще. Но понимать начинаешь только после проб и ошибок. Не умничаю. Просто совет.

[Аркадий Гальченко]

i3a4em, а функция pyrobot почему без изменений? Я пытался писать что-то похожее, но оно не работало. Твой вариант запустился) Спасибо!

[AlbertForest]

Аркадий Гальченко, ну видать не надо раз без изменений. Проблемы были не в функциях, а в том, как ты пытался все запустить

[Аркадий Гальченко]

i3a4em, не, я просто к тому, что в твоём новом коде есть:

async def start_bot(loop, func):
    loop.create_task(func())

И она не вызывается нигде, вроде?

Потом внутри функции aiobot есть своя функция:

async def start_bot(dp):
        event_loop.create_task(dp.start_polling())
    
    event_loop.run_until_complete(start_bot(dp))

При этом в функции pyrobot вообще нет вызова ни того, ни другого. Я просто пытаюсь разобраться, чтобы понять очередность работы

[AlbertForest]

И она не вызывается нигде, вроде?

Ты прав. Перенес ее в aiobot, а удалить забыл. Пятница... сам понимаешь))) (была)

pyrobot (как и aiobot) вызывается.

event_loop = asyncio.get_event_loop() # создаем цикл событий
    aiobot(event_loop, token) # Запускаем бота, передав цикл, в котором он должен существовать
    pyrobot(api_id, api_hash, phone_number) # Запускаем юзера...
    event_loop.run_forever() # Запускаем цикл

Answer 2

[Vindicar]

Ну потому что ты фигню написал.

th_bot = Thread(target=pyrobot(), args=())
th_userbot = Thread(target=aiobot(), args=())

Ты пытаешься запустить в качестве потока значение, возвращаемое функцией pyrobot(). А так как она уходит в цикл и значения не возвращает, то далее ничего не происходит. До вызова конструктора Thread() дело не доходит. С aiobot() аналогично.

Еще раз:
pyrobot() - вызов функции
pyrobot - ссылка на функцию

EDIT:
Оба бота - асинхронные на базе asyncio, им для работы нужен цикл-реактор (event loop). Вообще не факт, что хорошая идея запускать их в потоках.
Тут есть два варианта, сразу даже не скажу, что проще.
Вариант А, лобовой: каждый бот создаёт своё собственные реактор через asyncio.new_event_loop(), потом задаёт его как текущий для своего потока через asyncio.set_event_loop(loop). Если ботам не требуется взаимодействовать, то это может быть проще. Если требуется... будут проблемы. Два реактора в одной программе - это не хорошо.

Вариант Б, адекватный:
И вызов app.run(), и вызов executor.start_polling(dp, skip_updates=True) скорее всего под капотом создают асинхронную функцию (корутину), и запускают её в реакторе. Тогда ты можешь обойтись без потоков, заставив обоих ботов работать на одном реакторе. Нужно будет зарыться в доки, или даже глянуть исходники.

Например, для пирограмма написано такое:

When calling this method (app.run()) without any argument it acts as a convenience method that calls start(), idle() and stop() in sequence. It makes running a single client less verbose.

Т.е. вместо вызова app.run() ты можешь изменить код так:

async def pyrobot():  # обрати внимание, теперь функция асинхронная!
    print("pyro started")
    @app.on_message(filters.chat("some_chat"))
    async def print_pyrogram():
        print("Pyrogram")
    # это вместо вызова app.run(), как написано в доках.
    await app.start()
    try:
        await app.idle()
    finally:
        await app.end()

Затем делаешь аналогичный трюк с aiobot(). Нужно посмотреть в доках на аиограм, как именно.

И потом запускаешь обоих ботов кодом вида...
asyncio.run(asyncio.gather(pyrobot(), aiobot()))

Comments

[Аркадий Гальченко]

спасибо, не заметил) заменил на такое:

th_bot = Thread(target=pyrobot, args=())
th_userbot = Thread(target=aiobot, args=())

теперь валятся ошибки
RuntimeError: There is no current event loop in thread 'Thread-2 (aiobot)'.
RuntimeError: There is no current event loop in thread 'Thread-1 (pyrobot)'.

[Vindicar]

Аркадий Гальченко, потому что оба бота - асинхронные на базе asyncio, им для работы нужен цикл-реактор (event loop). Вообще не факт, что хорошая идея запускать их в потоках.
Тут есть два варианта, сразу даже не скажу, что проще.
Вариант А, лобовой: каждый бот создаёт своё собственные реактор через asyncio.new_event_loop(), потом задаёт его как текущий для своего потока через asyncio.set_event_loop(loop). Если ботам не требуется взаимодействовать, то это может быть проще. Если требуется... будут проблемы. Два реактора в одной программе - это не хорошо.

Вариант Б, адекватный:
И вызов app.run(), и вызов executor.start_polling(dp, skip_updates=True) скорее всего под капотом создают асинхронную функцию (корутину), и запускают её в реакторе. Тогда ты можешь обойтись без потоков, заставив обоих ботов работать на одном реакторе. Нужно будет зарыться в доки, или даже глянуть исходники.

Например, для пирограмма написано такое:

When calling this method (app.run()) without any argument it acts as a convenience method that calls start(), idle() and stop() in sequence. It makes running a single client less verbose.

Т.е. вместо вызова app.run() ты можешь изменить код так:

async def pyrobot():  # обрати внимание, теперь функция асинхронная!
    print("pyro started")
    @app.on_message(filters.chat("some_chat"))
    async def print_pyrogram():
        print("Pyrogram")
    await app.start()
    try:
        await app.idle()
    finally:
        await app.end()

Затем делаешь аналогичный трюк с aiobot(). Нужно посмотреть в доках на аиограм, как именно.

И потом запускаешь обоих ботов кодом вида...
asyncio.run(asyncio.gather(pyrobot(), aiobot()))

[Аркадий Гальченко]

Vindicar, спасибо за объяснение)