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)