import asyncio
from CONFIG import token_id
import os
import openpyxl
from aiogram import Bot, Dispatcher, types, executor
from aiogram.dispatcher.filters import Command
from selenium import webdriver
from selenium.webdriver.common.by import By
import shutil
import schedule
import time
import datetime



bot = Bot(token=token_id)
dp = Dispatcher(bot)
a = [421048811, 582897416]


"""
1. Считывание файлы
2. Берем ссылки из файла
3. По ссылкам из файла начинаем парсить раз в день, записывая цены
    3.1 две колонки:


"""




@dp.message_handler(Command('admin1703'))
async def admin(message: types.Message):
    a.append(message.from_user.id)
    print(a)
    await message.answer('ID добавлен: Вам доступны функции (загрузка файла и /getFile)')
admins_ids = a


@dp.message_handler(Command('my_id'))
async def id(message: types.Message):
    await message.answer(message.from_user.id)

@dp.message_handler(Command('start'))
async def start(message: types.Message):
    await message.answer("All Commands:\n/my_id\n/getFile")


@dp.message_handler(content_types=['document'])
async def doc(message: types.Message):
    if message.from_user.id in admins_ids:
        # os.mkdir('doc')
        path = os.path.abspath(os.path.dirname('doc'))

        if os.path.exists(os.path.join(path, f'{message.from_user.id}')):
            print(f'{path}\{message.from_user.id}')
                # await message.document.download(f'{path}\{message.from_user.id}')
                # shutil.rmtree(os.path.join(path, f'{message.from_user.id}'))
            await message.document.download(os.path.join(path, f'{message.from_user.id}'))

        else:

            os.mkdir(f'{message.from_user.id}')

            await message.document.download(os.path.join(path, f'{message.from_user.id}'))
    else:
        await message.answer("В доступе отказано!")

async def parser(url, time):
    options = webdriver.ChromeOptions()
    options.add_argument('--headless')
    browser = webdriver.Chrome('chromedriver', options=options)
    browser.get(url)
    book = openpyxl.Workbook()
    sheet = book.active
    sheet.cell(row=1, column=1).value = "URL"
    sheet.cell(row=1, column=2).value = "Name"
    sheet.cell(row=1, column=3).value = "Data"
    sheet.cell(row=2, column=3).value = "Price"
    n = 3
    # book.save("Result.xlsx")
    # print("Файл успешно сохранён!")
    # book.close()
    while True:
        await asyncio.sleep(time)
        try:
            browser.get(url)
        except Exception as err:
            pass
        #        print(f"Get error {err=}, {type(err)=}\n")
        try:
            product = browser.find_element(By.TAG_NAME, 'h1')
        except Exception as err:
            product = ""
        #        print(f"Find element TAG  {err=}, {type(err)=}\n")
        try:
            price = browser.find_element(By.CLASS_NAME, 'new-price')
        except Exception as err:
            price = ""
        #        print(f"Find element CLASS_NAME NEW  {err=}, {type(err)=}\n")
        try:
            product = product.text
        except Exception as err:
            product = 'Not Found'
        #        print(f"product.text  {err=}, {type(err)=}\n")
        try:
            price1 = price.text.split(" ")[0]
        except Exception as err:
            price1 = '0'
        print(product, price1)

        sheet.cell(row=n, column=4).value = price1
        sheet.cell(row=n, column=2).value = product
        sheet.cell(row=n, column=3).value = "Price"
        # sheet.cell(row=1, column=4).value = cur_date
        book.save("Result.xlsx")
        n += 1


@dp.message_handler(Command('getFile'))
async def get_file(message: types.Message):
    if message.from_user.id in admins_ids:
        msg = await message.answer("Началась обработка запроса!")
        #path = r'C:\Users\North\PycharmProjects\Torg_Otdel_Ulibka_06_05_2022\documents'
        #path = r'/home/daniil/Torg_Otdel_Ulibka_06_05_2022/documents'
        path = os.path.abspath(os.path.dirname('doc'))
        files = os.listdir(os.path.join(os.path.join(path,f'{message.from_user.id}\documents') ,f''))
        file = max(files)
        print(file)
        path = os.path.join(path, f'{message.from_user.id}\documents\{file}')





        # exel_file.close()
    else:
        await message.answer("В доступе отказано!")
#
# def parser1(url)

# async def sheduled(path, days=0):
#     if "/documents" in path:
#         pass
#     else:
#         path = f'{path}/documents'
#     files = os.listdir(path)
#     exel_file = openpyxl.load_workbook(f'{path}/{files[0]}')
#     sheet = exel_file.active
#     days = sheet.cell(row=1, column=1).value
#     parser = sheet.cell(row=1, column=2).value
#     time = sheet.cell(row=1, column=3).value
#     print(time, days, parser)
#     for i in range(int(days)):
#         await schedule.every().day().at(time).do(sheduled(path))



if __name__ == "__main__":
    # need user_id + file to
    # loop = asyncio.get_event_loop()
    # # loop.create_task(sheduled("582897416"))

    executor.start_polling(dp, skip_updates=True)