import telebot
import os

from daemons import base
from utils import queues


class Daemon(base.Daemon, queues.TasksHandlerMixin):
    def __init__(self):
        super().__init__()
        self.bot = telebot.TeleBot(os.getenv("TELEGRAM_TOKEN"))

    @property
    def queue_name(self):
        return 'ruz_bot_mailbox'
    
    def execute(self):
        self.poll()

    def process(self, payload):
        body = {
            'chat_id': payload['chat_id'],
            'text': payload['text'],
        }
        reply_markup = payload.get('reply_markup')
        if reply_markup:
            body['reply_markup'] = reply_markup
        try:
            self.bot.send_message(**body, parse_mode='Markdown')
        except Exception as exc:
            print('Error', str(exc))