def get_search_context(self, element):
driver = self.get_driver()
if element is None:
return driver
from JDI.web.selenium.elements.composite.web_site import WebSite
try:
if isinstance(element, WebSite) or isinstance(element.get_parent(), WebSite):
return driver
except: pass
try:
if issubclass(element, WebSite) or issubclass(element.get_parent(), WebSite):
return driver
except: pass
from JDI.web.selenium.elements.base.base_element import BaseElement
from JDI.web.selenium.elements.base.element import Element
if (element is None or type(element) is BaseElement) or \
(element.get_parent() is None and element.avatar.frame_locator is None):
return self.get_driver().switch_to.default_content()
if type(element) is Element and element.avatar.has_web_element():
return element.get_web_element
locator = element.get_locator()
search_context = self.get_driver().switch_to.default_content() \
if WebDriverByUtils.contains_root(locator) \
else self.get_search_context(element.get_parent())
locator = WebDriverByUtils.trim_root(locator) \
if WebDriverByUtils.contains_root(locator) \
else locator
frame = element.avatar.frame_locator
if frame is not None:
self.switch_to_last_opened_window()
res = search_context.find_element(element.avatar.frame_locator[0], element.avatar.frame_locator[1])
driver.switch_to_frame(res)
return search_context.find_element(locator[0], locator[1]) if locator is not None else search_context
def __search_elements(self):
search_context = self.get_driver() \
if WebDriverByUtils.contains_root(self.by_locator) \
else self.get_search_context(self.element.parent)
locator = WebDriverByUtils.trim_root(self.by_locator) \
if WebDriverByUtils.contains_root(self.by_locator) \
else self.by_locator
if search_context is None:
search_context = self.get_driver()
return search_context.find_elements(locator[0], locator[1])
def __search_elements(self):
search_context = self.get_driver() \
if WebDriverByUtils.contains_root(self.by_locator) \
else self.get_search_context(self.element.parent)
locator = WebDriverByUtils.trim_root(self.by_locator) \
if WebDriverByUtils.contains_root(self.by_locator) \
else self.by_locator
if search_context is None:
search_context = self.get_driver()
return search_context.find_elements(locator[0], locator[1])
def get_search_context(self, element):
driver = self.get_driver()
if element is None:
return driver
from JDI.web.selenium.elements.composite.web_site import WebSite
try:
if isinstance(element, WebSite) or isinstance(
element.get_parent(), WebSite):
return driver
except:
pass
try:
if issubclass(element, WebSite) or issubclass(
element.get_parent(), WebSite):
return driver
except:
pass
from JDI.web.selenium.elements.base.base_element import BaseElement
from JDI.web.selenium.elements.base.element import Element
if (element is None or type(element) is BaseElement) or \
(element.get_parent() is None and element.avatar.frame_locator is None):
return self.get_driver().switch_to.default_content()
if type(element) is Element and element.avatar.has_web_element():
return element.get_web_element
locator = element.get_locator()
search_context = self.get_driver().switch_to.default_content() \
if WebDriverByUtils.contains_root(locator) \
else self.get_search_context(element.get_parent())
locator = WebDriverByUtils.trim_root(locator) \
if WebDriverByUtils.contains_root(locator) \
else locator
frame = element.avatar.frame_locator
if frame is not None:
self.switch_to_last_opened_window()
res = search_context.find_element(element.avatar.frame_locator[0],
element.avatar.frame_locator[1])
driver.switch_to_frame(res)
return search_context.find_element(
locator[0], locator[1]) if locator is not None else search_context