1- import settings
1+ from selenium .common .exceptions import (
2+ NoSuchElementException ,
3+ StaleElementReferenceException ,
4+ TimeoutException ,
5+ )
6+ from selenium .webdriver .support import expected_conditions as EC
7+ from selenium .webdriver .support .ui import WebDriverWait
28
9+ import settings
310from base import expected_conditions as ec
4- from selenium .webdriver .support .ui import WebDriverWait
5- from selenium .webdriver .support import expected_conditions as EC
6- from selenium .common .exceptions import StaleElementReferenceException , TimeoutException , NoSuchElementException
11+
712
813class WebElementWrapper :
914 """A wrapper for selenium's WebElement. Supports all WebElement attributes
@@ -13,6 +18,7 @@ class WebElementWrapper:
1318 :param str attribute_name: The attribute name of the locator in its containing class.
1419 :param locator: An object of the type Locator.
1520 """
21+
1622 def __init__ (self , driver , attribute_name , locator ):
1723 self .driver = driver
1824 self .locator = locator
@@ -90,8 +96,7 @@ def click_expecting_popup(self, timeout=settings.TIMEOUT):
9096 self .click ()
9197
9298 try :
93- WebDriverWait (self .driver , timeout ).until (
94- EC .number_of_windows_to_be (2 ))
99+ WebDriverWait (self .driver , timeout ).until (EC .number_of_windows_to_be (2 ))
95100 except TimeoutException :
96101 raise ValueError ('No new window was opened.' )
97102 self .driver .close ()
@@ -106,13 +111,15 @@ def send_keys_deliberately(self, keys):
106111 for k in keys :
107112 self .element .send_keys (k )
108113
114+
109115class BaseLocator :
110116 """Abstract base class from which all Locator classes inherit.
111117
112118 Includes the method of how to locate the element (a subclass of selenium By),
113119 a string that actually identifies the element, and a timeout for how long to wait
114120 when searching for the element.
115121 """
122+
116123 def __init__ (self , selector , path , timeout = settings .TIMEOUT ):
117124 self .selector = selector
118125 self .path = path
@@ -135,6 +142,7 @@ class Locator(BaseLocator):
135142 most notably `get_web_element`. You may end up waiting longer than your timeout because some
136143 methods use more than one Wait.
137144 """
145+
138146 def get_web_element (self , driver , attribute_name ):
139147 """
140148 Check if element is on page and visible before returning the selenium
@@ -150,39 +158,54 @@ def get_web_element(self, driver, attribute_name):
150158 WebDriverWait (driver , self .timeout ).until (
151159 EC .presence_of_element_located (self .location )
152160 )
153- except (TimeoutException , StaleElementReferenceException ):
154- raise ValueError ('Element {} not present on page. {}' .format (
155- attribute_name , driver .current_url )) from None
161+ except (TimeoutException , StaleElementReferenceException ):
162+ raise ValueError (
163+ 'Element {} not present on page. {}' .format (
164+ attribute_name , driver .current_url
165+ )
166+ ) from None
156167
157168 try :
158169 WebDriverWait (driver , self .timeout ).until (
159170 EC .visibility_of_element_located (self .location )
160171 )
161- except (TimeoutException , StaleElementReferenceException ):
162- raise ValueError ('Element {} not visible before timeout. {}' .format (
163- attribute_name , driver .current_url )) from None
172+ except (TimeoutException , StaleElementReferenceException ):
173+ raise ValueError (
174+ 'Element {} not visible before timeout. {}' .format (
175+ attribute_name , driver .current_url
176+ )
177+ ) from None
164178
165179 try :
166180 WebDriverWait (driver , self .timeout ).until (
167181 EC .element_to_be_clickable (self .location )
168182 )
169- except (TimeoutException , StaleElementReferenceException ):
170- raise ValueError ('Element {} not clickable before timeout. {}' .format (
171- attribute_name , driver .current_url )) from None
183+ except (TimeoutException , StaleElementReferenceException ):
184+ raise ValueError (
185+ 'Element {} not clickable before timeout. {}' .format (
186+ attribute_name , driver .current_url
187+ )
188+ ) from None
172189
173190 if 'href' in attribute_name :
174191 try :
175192 WebDriverWait (driver , self .timeout ).until (
176193 ec .link_has_href (self .location )
177194 )
178- except (TimeoutException , StaleElementReferenceException ):
179- raise ValueError ('Element {} on page but does not have a href. {}' .format (
180- attribute_name , driver .current_url )) from None
195+ except (TimeoutException , StaleElementReferenceException ):
196+ raise ValueError (
197+ 'Element {} on page but does not have a href. {}' .format (
198+ attribute_name , driver .current_url
199+ )
200+ ) from None
181201 try :
182202 return driver .find_element (self .selector , self .path )
183203 except NoSuchElementException :
184- raise ValueError ('Element {} was present, but now is gone. {}' .format (
185- attribute_name , driver .current_url )) from None
204+ raise ValueError (
205+ 'Element {} was present, but now is gone. {}' .format (
206+ attribute_name , driver .current_url
207+ )
208+ ) from None
186209
187210 def get_element (self , driver , attribute_name ):
188211 return WebElementWrapper (driver , attribute_name , self )
@@ -195,6 +218,7 @@ class GroupLocator(BaseLocator):
195218 :param str path: Identifying string that is shared between the elements
196219 you are attempting to locate.
197220 """
221+
198222 def get_web_elements (self , driver ):
199223 return driver .find_elements (self .selector , self .path )
200224
@@ -214,7 +238,10 @@ class ComponentLocator(Locator):
214238 :param component_class: A subclass of BaseElment.
215239 Note: Currently the parameters selector, path, and timeout don't do anything.
216240 """
217- def __init__ (self , component_class , selector = None , path = None , timeout = settings .TIMEOUT ):
241+
242+ def __init__ (
243+ self , component_class , selector = None , path = None , timeout = settings .TIMEOUT
244+ ):
218245 super ().__init__ (selector , path , timeout )
219246 self .component_class = component_class
220247
@@ -227,6 +254,7 @@ class BaseElement:
227254 Handles waffled pages, storage of the WebDriver, and returning WebElements when Locators are
228255 accessed.
229256 """
257+
230258 default_timeout = settings .TIMEOUT
231259
232260 def __new__ (cls , * args , ** kwargs ):
0 commit comments