self/botalka
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add
All checks were successful
Deploy Dev / Build (pull_request) Successful in 24s
Details
Deploy Dev / Push (pull_request) Successful in 21s
Details
Deploy Dev / Deploy dev (pull_request) Successful in 33s
Details
Deploy Prod / Build (pull_request) Successful in 11s
Details
Deploy Prod / Push (pull_request) Successful in 19s
Details
Deploy Prod / Deploy prod (pull_request) Successful in 22s
Details

Browse Source
This commit is contained in:
Administrator 2024-12-21 19:27:00 +03:00
parent 7ed7afbef4
commit 5a00dad803
1 changed files with 14 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
daemons/poll.py
View File

@ -1,5 +1,5 @@
import telebot import telebot
import threading import multiprocessing
import time import time
from daemons import base from daemons import base
@ -10,45 +10,37 @@ from utils import queues
class Daemon(base.Daemon): class Daemon(base.Daemon):
def __init__(self): def __init__(self):
super().__init__() super().__init__()
self.telegram_bots: dict[str, dict[str, telebot.TeleBot|None]] = {} self.processes: dict[str, multiprocessing.Process|None] = {}
self.threads: dict[str, dict[str, threading.Thread|None]] = {}
def execute(self): def execute(self):
while True: while True:
bots = platform.platform_client.get_config('bots') bots = platform.platform_client.get_config('bots')
for project_name, project in bots.items(): for project_name, project in bots.items():
if project_name not in self.telegram_bots:
self.telegram_bots[project_name] = {}
self.threads[project_name] = {}
for bot_name, bot_info in project.items(): for bot_name, bot_info in project.items():
if bot_name not in self.telegram_bots[project_name]: key = f'{project_name}_{bot_name}'
self.telegram_bots[project_name][bot_name] = None proc = self.processes.get(key)
self.threads[project_name][bot_name] = None
bot = self.telegram_bots[project_name][bot_name]
if bot_info.get('poll_enabled'): if bot_info.get('poll_enabled'):
if bot is not None and self.threads[project_name][bot_name].is_alive(): if proc and proc.is_alive():
print(f'process for {project_name} {bot_name} is alive') print(f'process for {project_name} {bot_name} is alive')
continue continue
print(f'starting process for {project_name} {bot_name}') print(f'starting process for {project_name} {bot_name}')
bot = telebot.TeleBot(bot_info['secrets']['telegram_token']) process = multiprocessing.Process(target=self.start_polling, args=(bot_info['secrets']['telegram_token'], bot_info['queue']))
thread = self.start_polling(bot, bot_info['queue']) process.start()
self.telegram_bots[project_name][bot_name] = bot self.processes[key] = process
self.threads[project_name][bot_name] = thread
print(f'started process for {project_name} {bot_name}') print(f'started process for {project_name} {bot_name}')
else: else:
if bot is None: if proc is None:
print(f'process for {project_name} {bot_name} is not alive') print(f'process for {project_name} {bot_name} is not alive')
continue continue
print(f'terminating process for {project_name} {bot_name}') print(f'terminating process for {project_name} {bot_name}')
bot.stop_bot() proc.terminate()
self.telegram_bots[project_name][bot_name] = None self.processes[key] = None
print(f'terminated process for {project_name} {bot_name}') print(f'terminated process for {project_name} {bot_name}')
time.sleep(10) time.sleep(10)
def start_polling(self, bot: telebot.TeleBot, queue: str) -> threading.Thread: def start_polling(self, token: str, queue: str):
bot = telebot.TeleBot(token)
@bot.message_handler(content_types=['audio', 'photo', 'voice', 'video', 'document', 'animation', 'text', 'location', 'contact', 'sticker', 'video_note']) @bot.message_handler(content_types=['audio', 'photo', 'voice', 'video', 'document', 'animation', 'text', 'location', 'contact', 'sticker', 'video_note'])
def do_action(message: telebot.types.Message): def do_action(message: telebot.types.Message):
queues.set_task(self.stub, queue, message.json, 1) queues.set_task(self.stub, queue, message.json, 1)
thread = threading.Thread(target=bot.polling) bot.polling()
thread.start()
return thread