import telebot
from telebot import types
from main import global_stat
import json


bot = telebot.TeleBot('1239180030:AAFnw0HKMM3xSTbNYBZuXoum1J0YYbCGWJ4')


@bot.message_handler(commands=['start'])
def start(message):
    markup = types.ReplyKeyboardMarkup(resize_keyboard=True, row_width=2)
    btn1 = types.KeyboardButton('Дані з цілого світу')
    btn2 = types.KeyboardButton('Україна')
    btn3 = types.KeyboardButton('США')
    btn4 = types.KeyboardButton('Німеччина')
    markup.add(btn1, btn2, btn3, btn4)

    send_message = f"<b>Привіт {message.from_user.first_name}!</b>\nЩоб дізнатися статистику по якійсь країні просто напиши мені назву, наприклад: США, Україна, і так дальше... " \
                   f"\nАбо обери зі списку нижче 👇\n"
    bot.send_message(message.chat.id, send_message, parse_mode='html', reply_markup=markup)


@bot.message_handler(content_types=['text'])
def mess(message):
    final_message = ""
    get_message_bot = message.text.strip().lower()
    if get_message_bot == "все":
        location = json.loads(global_stat())

    if final_message == "":
        final_message = f"<u>Данные по стране:</u>\n" \
                        f"<b>Заболевших:</b>  {location['result']['confirmed']:,}\n<b>Сметрей: </b>" \
                        f"{location['result']['deaths']:,}"

    bot.send_message(message.chat.id, final_message, parse_mode='html')


bot.polling(none_stop=True)