def setUpClass(cls): logger.debug("setUp...") # self.driver = Chrome.normal() cls.driver = Chrome.headless() # cls.driver = webdriver.Firefox() cls.driver.implicitly_wait(10) cls.driver.maximize_window() page = LoginPage(cls.driver) page.login()
def fetch_links(): driver = Chrome.headless() page = LoginPage(driver) page.login() links = [] link_elements = driver.find_elements_by_xpath( '//a[not(contains(@href,"javascript"))]') for element in link_elements: link = element.get_attribute("href") if link: link = link.split('/')[-3:] link = [os.path.join(link[0], link[1]), link[2]] links.append(link) return links