spooning
commented
9 years ago Originally submitted to Google Code by Andreas.EbbertKarroum on 10 Feb 2010
I improved the en-/decoding bits. would be nice, if this is included in the upcoming release:
def _absnorm(self, path):
return os.path.normpath(os.path.abspath(path.replace('/', os.sep)))
def _write_to_file(self, path, content, mode):
path = self._absnorm(path)
parent = os.path.dirname(path)
if not os.path.exists(parent):
os.makedirs(parent)
f = open(path, mode+'b')
f.write(content)
f.close()
return path
def capture_remote_screenshot(self, path=None):
"""Captures a screenshot and returns it as a string
Given path must be relative to Robot Framework output directory,
otherwise the embedded image is not shown in the log file. If path is
not given, file with name similar to 'selenium-image-x.png' is created
directly under the output directory.
"""
# configure path
if path and os.path.isabs(path):
raise RuntimeError("Given path must be relative to Robot outpudir")
if not path:
path = self._namegen.next()
outdir = NAMESPACES.current.variables['${outputdir}']
fullpath = os.path.join(outdir, path)
if not os.path.exists(os.path.split(fullpath)[0]):
os.makedirs(os.path.split(fullpath)[0])
# retrieve remote screenshot
self._info("Retrieving Screenshot")
screenshot =self._selenium.capture_entire_page_screenshot_to_string("background=#CCFFDD") screenshot=base64.b64decode(screenshot)
# save screenshot
self._info("Saving screenshot to file '%s'" % fullpath)
self._write_to_file(fullpath, screenshot, 'w')
self._html('<td></td></tr><tr><td colspan="3"><a href="%s">'
'<img src="%s" width="700px"></a></td></tr>' % (path, path))
In order to use the screenshot as String feature from Selenium, which allows for taking screenshots on a remote system, I added this to init.py. I'm sure it can be done better, but I hope you get the idea :) One of the problems is that the String that holds the binary image, is encoded in the default encoding of the platform on which the screenshot is taken. Since this can vary, we have to provide a parameter for that:
self._selenium.capture_entire_page_screenshot_to_string("background=#CCFFDD") screenshot=base64.b64decode(screenshot) try: screenshot = screenshot.decode(codec) except UnicodeDecodeError: raise RuntimeError("Got error when decoding Screenshot-String with '%s'" % codec)
path))