import telebot
from wwdb import *
from parsingstocks import *
from webdriver_manager.chrome import ChromeDriverManager
from selenium import webdriver



options = webdriver.ChromeOptions()
options.add_argument("--ignore-certificate-error")
options.add_argument("--ignore-ssl-errors")
#options.add_argument('--incognito')
options.add_argument("--disable-blink-features=AutomationControlled")
options.add_argument('load-extension=C:/Users/Girafi/Desktop/bot/3.13_0')
caps = webdriver.DesiredCapabilities.CHROME.copy()
caps['acceptInsecureCerts'] = True
caps['acceptSslCerts'] = True
driver = webdriver.Chrome(ChromeDriverManager().install(), options=options, desired_capabilities=caps)


TOKEN = "5151507341:AAHeGomTb8b6i3YZaRF6Fb7odmf79IkqSeY"
bot = telebot.TeleBot(TOKEN)

@bot.message_handler(commands=['stop'])
def stop(message):
    closedb()
    driver.close()
    bot.send_message(message.chat.id, "Робота зупинена")
    bot.stop_polling()
    exit()

@bot.message_handler(commands=['start'])
def start(message):
    bot.send_message(message.chat.id, 'Привіт, я бот для перевірки акцій компаній /seestocks')

@bot.message_handler(commands=['addinstrument'])
def addinstrument(message):
    bot.send_message(message.chat.id, 'Почато введення інструментів, для припинення введіть будь-яку цифру')
    bot.register_next_step_handler(message, instrument_add)

def instrument_add(message):
    if (message.text[0] == '/'):
        bot.send_message(message.chat.id, "Схоже ви намагались ввести команду. Щоб це зробити, спочатку зупиніть введення інструментів")
    else:
        if (message.text.isdigit()):
            bot.send_message(message.chat.id, "Введення інструментів припинено")
            return
        bot.send_message(message.chat.id, "Зачекайте, йде перевірка...")
        if (testforinput(message.text, driver)):
            bot.send_message(message.chat.id, insert(message.chat.id, message.text))
            bot.register_next_step_handler_by_chat_id(message, instrument_add)
        else:
            bot.send_message(message.chat.id, "Такого об'єкту не існує")
            bot.register_next_step_handler_by_chat_id(message, instrument_add)

@bot.message_handler(commands=['seedb'])
def seedb(message):
    print(printdb())

@bot.message_handler(commands=['seestocks'])
def seestocks(message):
    instruments = get(message.chat.id)
    if (len(instruments) == 0):
        bot.send_message(message.chat.id, "Схоже ви не додали ні одного об'єкту слідкування, використейте /addinstrument щоб додати їх!")
    else:
        for i in range(len(instruments)):
            bot.send_message(message.chat.id, getinformation(instrument=instruments[i][1], driver=driver))
            makesrcreenshot(instrument=instruments[i][1], driver=driver)
            img = open('C:/Users/Girafi/Desktop/bot/screen.png', 'rb')
            bot.send_photo(message.chat.id, photo=img)
        bot.send_message(message.chat.id, "Це все")

#############

bot.polling()