Forums

from pyvirtualdisplay import Display
from selenium import webdriver
import time
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions
from selenium.webdriver.common.by import By

with Display():
    # we can now start Firefox and it will run inside the virtual display

    profile = webdriver.FirefoxProfile()
    profile.set_preference('browser.download.folderList', 2) # custom location
    profile.set_preference('browser.download.manager.showWhenStarting', False)
    profile.set_preference('browser.download.dir', '/home/mmc777/dsp/SELENIUM')
    profile.set_preference('browser.helperApps.neverAsk.saveToDisk', 'text/csv')
    driver = webdriver.Firefox(profile)
    # put the rest of our selenium code in a try/finally
    # to make sure we always clean up at the end
    try:

        #driver.set_window_size(1120, 550)

        user_name = '******'
        password = '*******'
        driver.get("https://track.cruxsystems.com/login")
        #print(driver.page_source)
        driver.implicitly_wait(30)

        #driver.find_element_by_id('tab_ct').click()
        driver.find_elements_by_xpath("//input[@type = 'email']")[0].send_keys(user_name)
        driver.find_elements_by_xpath("//input[@type = 'password']")[0].send_keys(password)
        driver.find_elements_by_xpath("//input[@type = 'submit']")[0].click()
        #driver.explicitly_wait(30)
        #info = driver.find_element_by_id("ref_info_premain")

        driver.implicitly_wait(30)
        driver.find_elements_by_xpath("//button[@uib-tooltip='Containers']")[0].click()
        time.sleep(5)
        driver.find_elements_by_xpath("//button[@uib-tooltip='Select all']")[0].click()
        WebDriverWait(driver, 50).until(
            expected_conditions.element_to_be_clickable(
                (By.XPATH, '//button[@uib-tooltip="Download"]')))
        driver.find_elements_by_xpath("//button[@uib-tooltip='Download']")[0].click()
        #result = driver.find_elements_by_xpath("//div[@class='flex-33']")[0]
        #print(result.text)

    finally:
        driver.quit()
        print('finished')