Merge pull request 'master' (#47) from master into dev

Reviewed-on: #47
This commit is contained in:
emmatveev 2024-12-31 02:50:30 +03:00
commit 5102564b84
2 changed files with 8 additions and 2 deletions

View File

@ -1,6 +1,7 @@
import telebot
import multiprocessing
import time
import json
from daemons import base
from utils import platform
@ -41,5 +42,5 @@ class Daemon(base.Daemon):
bot = telebot.TeleBot(token)
@bot.message_handler(content_types=['audio', 'photo', 'voice', 'video', 'document', 'animation', 'text', 'location', 'contact', 'sticker', 'video_note'])
def do_action(message: telebot.types.Message):
queues.set_task(queue, message.json, 1)
queues.set_task(queue, json.loads(message.json), 1)
bot.polling()

View File

@ -17,7 +17,12 @@ class QueuesException(Exception):
class TasksHandlerMixin:
def poll(self):
while True:
response = requests.get(f'{QUEUES_URL}/api/v1/take', headers={'queue': self.queue_name}).json()
try:
response = requests.get(f'{QUEUES_URL}/api/v1/take', headers={'queue': self.queue_name}).json()
except requests.JSONDecodeError:
print('Unable to decode json')
time.sleep(3)
continue
task = response.get('task')
if not task:
time.sleep(0.2)