def take_screenshot(url): try: webdriver = selenium.webdriver.PhantomJS('vendor/phantomjs/bin/phantomjs') webdriver.get(url) webdriver.set_window_size(1280,800) imagedata = webdriver.get_screenshot_as_base64() webdriver.close() webdriver.quit() except Exception, e: raise
def _screenshot(self, webdriver, url, timeout): webdriver.get(url) webdriver.execute_async_script(self.wait_script) screenshot = webdriver.get_screenshot_as_base64() # strip off the data:img/png, part of the url if screenshot.startswith("data:image/png;base64,"): screenshot = screenshot.split(",", 1)[1] return screenshot
def take_screenshot(webdriver, file_name="sample.png"): """ @param webdriver: WebDriver handler. @type webdriver: WebDriver @param file_name: Name to label this screenshot. @type file_name: str """ if isinstance(webdriver, WebDriver): base64_data = webdriver.get_screenshot_as_base64() screenshot_data = base64.decodestring(base64_data) screenshot_file = open(file_name, "w") screenshot_file.write(screenshot_data) screenshot_file.close() else: webdriver.save_screenshot(filename)
def take_screenshot(webdriver, file_name = "sample.png"): """ @param webdriver: WebDriver handler. @type webdriver: WebDriver @param file_name: Name to label this screenshot. @type file_name: str """ if isinstance(webdriver, WebDriver): base64_data = webdriver.get_screenshot_as_base64() screenshot_data = base64.decodestring(base64_data) screenshot_file = open(file_name, "w") screenshot_file.write(screenshot_data) screenshot_file.close() else: webdriver.save_screenshot(filename)