def get_element_by_link_text(self, text): locator = (By.LINK_TEXT, text) try: WebDriverWait(self.b, 10).until(EC.visibility_of_element_located(locator)) except EC.NoSuchElementException as E: file_name = '%s_can_not_find_element_by_css_%s' %text function.save_screenshot(self.b, file_name) raise E return self.b.find_element_by_link_text(text)
def get_element_by_name(self, the_name): locator = (By.NAME, the_name) try: WebDriverWait(self.b, 10).until(EC.visibility_of_element_located(locator)) except EC.NoSuchElementException as E: filename = 'can_not_find_element_by_name_%s' % the_name function.save_screenshot(self.b, filename) raise E return self.b.find_element_by_name(the_name)
def get_element_by_css(self, css): locator = (By.CSS_SELECTOR, css) try: WebDriverWait(self.b, 10).until(EC.visibility_of_element_located(locator)) except EC.NoSuchElementException as E: filename = 'can_not_find_element_by_css_%s' %css function.save_screenshot(self.b, filename) raise E return self.b.find_element_by_css_selector(css)
def get_element_by_xpath(self, xpath): locator = (By.XPATH, xpath) print locator try: WebDriverWait(self.b, 10).until(EC.visibility_of_element_located(locator)) except EC.NoSuchElementException as E: file_name = '%s_can_not_find_element_by_css_%s' %xpath function.save_screenshot(self.b, file_name) raise E return self.b.find_element_by_xpath(xpath)