from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import Select
from selenium.common.exceptions import NoSuchElementException
from datetime import datetime
import unittest, time
import xlsxwriter

sites = (
("http://192.168.0.12/images.html", 2),
)

def main():
    results = []
    for site in sites:
        base_url = site[0]
        for i in range(0, site[1]):
            driver = webdriver.Chrome()

            driver.get(base_url)
            loadEventEnd = driver.execute_script( "return ( window.performance.timing.loadEventEnd - window.performance.timing.navigationStart )")
            domComplete = driver.execute_script( "return ( window.performance.timing.domComplete - window.performance.timing.navigationStart )")
            domLoading = driver.execute_script( "return ( window.performance.timing.domLoading - window.performance.timing.navigationStart )")
            domContentLoadedEventEnd = driver.execute_script( "return ( window.performance.timing.domContentLoadedEventEnd - window.performance.timing.navigationStart )")
            results.append((base_url, loadEventEnd, domComplete, domLoading, domContentLoadedEventEnd))
            driver.quit()

    # Create a workbook and add a worksheet.
    workbook = xlsxwriter.Workbook('test' + datetime.now().isoformat().replace('.', '-').replace(':', '-') + '.xlsx')
    worksheet = workbook.add_worksheet()


    # Write some data headers.
    worksheet.write('A1', 'Site')
    worksheet.write('B1', 'loadEventEnd')
    worksheet.write('C1', 'domComplete')
    worksheet.write('D1', 'domLoading')
    worksheet.write('E1', 'domContentLoadedEventEnd')


    # Start from the first cell below the headers.
    domContentLoadedEventEnd = 4
    domLoading = 3
    domComplete = 2
    loadEventEnd = 1
    site = 0
    row = 1

    worksheet.write(row, site, "Url")
    worksheet.write(row, loadEventEnd, "loadEventEnd")
    worksheet.write(row, domComplete, "domComplete")
    worksheet.write(row, domLoading, "domLoading")
    worksheet.write(row, domContentLoadedEventEnd, "domContentLoadedEventEnd")

    for url, res1, res2, res3, res4 in (results):
        worksheet.write(row, site, url)
        worksheet.write(row, loadEventEnd, res1)
        worksheet.write(row, domComplete, res2)
        worksheet.write(row, domLoading, res3)
        worksheet.write(row, domContentLoadedEventEnd, res4)
        row += 1

    workbook.close()


if __name__ == "__main__":
    main()