{"id":336,"date":"2017-12-18T15:43:20","date_gmt":"2017-12-18T20:43:20","guid":{"rendered":"https:\/\/arthurpemberton.com\/?p=336"},"modified":"2018-12-26T22:38:51","modified_gmt":"2018-12-27T03:38:51","slug":"full-page-screenshots-with-python-and-selenium","status":"publish","type":"post","link":"https:\/\/arthurpemberton.com\/2017\/12\/full-page-screenshots-with-python-and-selenium","title":{"rendered":"Full page screenshots with Python and Selenium"},"content":{"rendered":"

Currently, none of the major Selenium drivers (browsers) support the ability to easily take a screenshot of an entire web page. The following function takes multiple screenshots through the viewport and scrolls between screenshots, then stitches the resulting images into a single PNG.<\/p>\n

<\/p>\n

\r\ndef save_fullpage_screenshot(driver, url, output_path, tmp_prefix='selenium_screenshot', tmp_suffix='.png'):\r\n\t"""\r\n\tCreates a full page screenshot using a selenium driver by scrolling and taking multiple screenshots,\r\n\tand stitching them into a single image.\r\n\t"""\r\n\r\n\t# get the page\r\n\tdriver.get(url)\r\n\r\n\t# get dimensions\r\n\twindow_height = driver.execute_script('return window.innerHeight')\r\n\tscroll_height = driver.execute_script('return document.body.parentNode.scrollHeight')\r\n\tnum = int( math.ceil( float(scroll_height) \/ float(window_height) ) )\r\n\r\n\t# get temp files\r\n\ttempfiles = []\r\n\tfor i in xrange( num ):\r\n\t\tfd,path = tempfile.mkstemp(prefix='{0}-{1:02}-'.format(tmp_prefix, i+1), suffix=tmp_suffix)\r\n\t\tos.close(fd)\r\n\t\ttempfiles.append(path)\r\n\t\tpass\r\n\r\n\ttry:\r\n\t\t# take screenshots\r\n\t\tfor i,path in enumerate(tempfiles):\r\n\t\t\tif i > 0:\r\n\t\t\t\tdriver.execute_script( 'window.scrollBy(%d,%d)' % (0, window_height) )\r\n\t\t\t\r\n\t\t\tdriver.save_screenshot(path)\r\n\t\t\tpass\r\n\t\t\r\n\t\t# stitch images together\r\n\t\tstiched = None\r\n\t\tfor i,path in enumerate(tempfiles):\r\n\t\t\timg = Image.open(path)\r\n\t\t\t\r\n\t\t\tw, h = img.size\r\n\t\t\ty = i * window_height\r\n\t\t\t\r\n\t\t\tif i == ( len(tempfiles) - 1 ):\r\n\t\t\t\timg = img.crop((0, h-(scroll_height % h), w, h))\r\n\t\t\t\tw, h = img.size\r\n\t\t\t\tpass\r\n\t\t\t\r\n\t\t\tif stiched is None:\r\n\t\t\t\tstiched = Image.new('RGB', (w, scroll_height))\r\n\t\t\t\r\n\t\t\tstiched.paste(img, (\r\n\t\t\t\t0, # x0\r\n\t\t\t\ty, # y0\r\n\t\t\t\tw, # x1\r\n\t\t\t\ty + h # y1\r\n\t\t\t))\r\n\t\t\tpass\r\n\t\tstiched.save(output_path)\r\n\tfinally:\r\n\t\t# cleanup\r\n\t\tfor path in tempfiles:\r\n\t\t\tif os.path.isfile(path):\r\n\t\t\t\tos.remove(path)\r\n\t\tpass\r\n\r\n\treturn output_path\r\n<\/pre>\n
The following libraries are required:<\/p>\n