fix #35

Merged
emmatveev merged 1 commits from master into dev 2024-12-08 13:45:29 +03:00
2 changed files with 3 additions and 2 deletions
Showing only changes of commit 0b6dc3af1a - Show all commits

View File

@ -9,7 +9,7 @@ from utils import queues
class Daemon(base.Daemon):
def __init__(self):
super().__init__(self)
super().__init__()
self.telegram_bots: dict[str, dict[str, telebot.TeleBot|None]] = {}
self.threads: dict[str, dict[str, threading.Thread|None]] = {}

View File

@ -13,7 +13,8 @@ class TasksHandlerMixin:
def poll(self):
while True:
response: tasks_pb2.TakeResponse = self.stub.Take(tasks_pb2.TakeRequest(queue=self.queue_name))
if not response.task:
task = response.task
if not task:
time.sleep(0.2)
continue
try: