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
# !/usr/bin/env python # -*- coding=utf-8 -*- from time import sleep from page.page import BasePage from util.browser import Chrome class InvoicePage(BasePage): # page = __file__ menu = '' subject = '' def do_something(self): pass if __name__ == '__main__': d = Chrome.normal() # d = Chrome.headless() p = InvoicePage(d) p.load() p.do_somthing() sleep(5) d.quit()