-
Notifications
You must be signed in to change notification settings - Fork 20
/
element.py
59 lines (51 loc) · 2.41 KB
/
element.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
from selenium.webdriver.support.ui import WebDriverWait
class BasePageElement(object):
def __set__(self, obj, value):
# fetch driver from caller
driver = obj.driver
# set element
WebDriverWait(driver, 100).until(
lambda driver: driver.find_element_by_id(self.locator))
# force focus on element before send_keys
driver.find_element_by_id(self.locator).click()
if value.startswith(':no-clear:'):
value = value.replace(':no-clear:', '')
else:
driver.find_element_by_id(self.locator).clear()
driver.find_element_by_id(self.locator).send_keys(value)
def __get__(self, obj, owner):
try:
driver = obj.driver
except AttributeError:
return
WebDriverWait(driver, 100).until(
lambda driver: driver.find_element_by_id(self.locator))
element = driver.find_element_by_id(self.locator)
return element.get_attribute("value")
class DropDownElement(object):
def __set__(self, obj, value):
driver = obj.driver
WebDriverWait(driver, 100).until(
lambda driver: driver.find_element_by_xpath("//select[@name='%s']/option" % self.locator))
if value.startswith(':value:'):
value = value.replace(':value:', '')
select = driver.find_element_by_xpath("//select[@name='%s']/option[normalize-space(@value)='%s']" % (self.locator, value))
elif value.startswith(':partial-text:'):
value = value.replace(':partial-text:', '')
select = driver.find_element_by_xpath("//select[@name='%s']/option[contains(normalize-space(text()),'%s')]" % (self.locator, value))
else:
select = driver.find_element_by_xpath("//select[@name='%s']//option[normalize-space(text())='%s']" % (self.locator, value))
select.click()
def __get__(self, obj, owner):
try:
driver = obj.driver
except AttributeError:
return
element = WebDriverWait(driver, 100).until(
lambda driver: driver.find_element_by_name(self.locator))
element = driver.find_element_by_name(self.locator)
element_value = element.get_attribute("value")
all_options = element.find_elements_by_tag_name("option")
for options in all_options:
if options.get_attribute("value") == element_value:
return str(options.text)