import telebot
from telebot import types
import COVID19Py



covid19 = COVID19Py.COVID19()
bot = telebot.TeleBot('токен')

# Функция, что сработает при отправке команды Старт
# Здесь мы создаем быстрые кнопки, а также сообщение с привествием
@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.replace("ё", "е").lower()
    if get_message_bot == "во всем мире":
        try:
            location = covid19.getLatest()
            print(location)
            final_message = f"<u>Данные по всему миру:</u>\n<b>Заболевших: </b>{location['confirmed']:,}\n<b>Сметрей: </b>{location['deaths']:,}"
            bot.send_message(message.chat.id, final_message, parse_mode='html')
        except Exception as e:
            print(e)
    if get_message_bot == "россия":
        try:
            location = covid19.getLocationByCountryCode("RU")
            print(location[0].get('latest')['confirmed'])
            final_message = f"<u>Данные из России:</u>\n<b>Заболевших: </b>{location[0].get('latest')['confirmed']:,}\n<b>Сметрей: </b>{location[0].get('latest')['deaths']:,}"
            bot.send_message(message.chat.id, final_message, parse_mode='html')
        except Exception as e:
            print(e)

    if get_message_bot == "украина":
        try:
            location = covid19.getLocationByCountryCode("UA")
            print(location[0].get('latest')['confirmed'])
            final_message = f"<u>Данные с Украины:</u>\n<b>Заболевших: </b>{location[0].get('latest')['confirmed']:,}\n<b>Сметрей: </b>{location[0].get('latest')['deaths']:,}"
            bot.send_message(message.chat.id, final_message, parse_mode='html')
        except Exception as e:
            print(e)

    if get_message_bot == "сша":
        try:
            location = covid19.getLocationByCountryCode("US")
            print(location[0].get('latest')['confirmed'])
            final_message = f"<u>Данные из США:</u>\n<b>Заболевших: </b>{location[0].get('latest')['confirmed']:,}\n<b>Сметрей: </b>{location[0].get('latest')['deaths']:,}"
            bot.send_message(message.chat.id, final_message, parse_mode='html')
        except Exception as e:
            print(e)

    # Вот это зачем так и не понял
    # if final_message == "":
    #   date = location[0]['last_updated'].split("T")
    #   time = date[1].split(".")
    #   final_message = f"<u>Данные по стране:</u>\nНаселение: {location[0]['country_population']:,}\n" \
    #           f"Последнее обновление: {date[0]} {time[0]}\nПоследние данные:\n<b>" \
    #           f"Заболевших: </b>{location[0]['latest']['confirmed']:,}\n<b>Сметрей: </b>" \
    #           f"{location[0]['latest']['deaths']:,}"


# Это нужно чтобы бот работал всё время
bot.polling(none_stop=True)