import xlrd
import xlwt
import time
import datetime
from selenium import webdriver
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.support.select import Select

# Open a workbook
workbook = xlrd.open_workbook('items.xls')

# Loads only current sheets to memory
workbook = xlrd.open_workbook('items.xls', on_demand=True)

# Load a specific sheet by name
worksheet = workbook.sheet_by_name('Sheet1')

# Load a specific sheet by index
worksheet = workbook.sheet_by_index(0)

USERNAME = '58079'
PASSWORD = 'globe2013'
# HSN = 'hsn2371300'

driver = webdriver.Chrome(executable_path=r'../assets/chromedriver.exe')
# driver = webdriver.Firefox(executable_path=r'geckodriver.exe')
# driver = webdriver.Ie(executable_path=r'IEDriverServer.exe')

now = datetime.datetime.now().strftime('%d-%m-%Y')

# Initialize a workbook
book = xlwt.Workbook(encoding="utf-8")

# Add a sheet to the workbook
sheet1 = book.add_sheet("Results", cell_overwrite_ok=True)

# Write to the sheet of the workbook
sheet1.write(0, 0, "Tyre24.de [EUR]")
sheet1.write(0, 1, now)

# logowanie
driver.get('https://tyre24.alzura.com/de/de')
time.sleep(1)
driver.find_element_by_name('userid').send_keys(USERNAME)
driver.find_element_by_name('password').send_keys(PASSWORD)
driver.find_element_by_css_selector('input.widget-login-button-input').click()
time.sleep(1)
# logowanie
# szukanie
driver.get('https://tyre24.alzura.com/de/de/search/tyre/origin/profi#/?init=1&showTab=1&listView=1&rownum=30&minTyres=4')
time.sleep(2)
x = 0
y = 1
n = 1
print(datetime.datetime.now().strftime('%d-%m-%Y %H:%M:%S'))
print('ilosc pozycji:', worksheet.nrows)
for i in range(worksheet.nrows):
    HSN = worksheet.cell(x, 0).value
    try:
        driver.find_element_by_id('input-search-1').clear()
        driver.find_element_by_id('input-search-1').send_keys(HSN)
        driver.find_element_by_id('profi-search-button').click()
        time.sleep(2)
    except:
        driver.get('https://tyre24.alzura.com/de/de/search/tyre/origin/profi#/?init=1&showTab=1&listView=1&rownum=30&minTyres=4')
        time.sleep(2)
        driver.find_element_by_id('input-search-1').clear()
        driver.find_element_by_id('input-search-1').send_keys(HSN)
        driver.find_element_by_id('profi-search-button').click()
        time.sleep(2)
# cena
    try:
        table_price = driver.find_element_by_css_selector('table.si-price-table')
        hsn = HSN.replace('hsn', '')
        result = table_price.find_element_by_css_selector('span.ng-binding').text
        result = result.replace('.', ',')
        print(n, ':', hsn, result)
        sheet1.write(y, 0, hsn)
        sheet1.write(y, 1, result)
    except:
        hsn = HSN.replace('hsn', '')
        result = 'brak'
        print(n, ':', hsn, result)
        sheet1.write(y, 0, hsn)
        sheet1.write(y, 1, result)
    x = x + 1
    y = y + 1
    n = n + 1
# szukanie
# Save the workbook
book.save("results.xls")
driver.close()
print(datetime.datetime.now().strftime('%d-%m-%Y %H:%M:%S'))

# TODO: zapis co xls lub csv