def __init__(self, browser, selectors, *args, **kwargs): self.selectors_msg = '''Selectors %s didn't match any elements on the page (%s) and 'therefore the attempt to call method on it failed.''' % (selectors, browser.url) # Try taking screenshot from issue take_screenshot(browser, nameformat='${browser}-error.${ext}') super(ElementNotFoundError, self).__init__(self.selectors_msg, *args, **kwargs)
def take_screenshot(self, nameformat=None): return take_screenshot(self._browser, nameformat)