Python: Selenium with PhantomJs to capture Screenshots

This article is deprecated as PhantomJs has been deprecated from Selenium as driver.

PhantomJS is a headless WebKit script-able with a JavaScript API. It has fast and native support for various web standards: DOM handling, CSS selector, JSON, Canvas, and SVG. And Selenium is a portable software-testing framework for web applications. Selenium provides a record/playback tool for authoring tests without the need to learn a test scripting language (Selenium IDE)

Using the combination of selenium and PhantomJs can give you a way to capture screenshots and use it in your choices.

Installation of PhantomJs

For that let’s install PhantomJs in your computer. For Ubuntu/Debian platform, you can use like this:

>sudo apt-get install phantomjs

Or,

>sudo apt-get update
>sudo apt-get install build-essential chrpath libssl-dev libxft-dev
>sudo apt-get install libfreetype6 libfreetype6-dev
>sudo apt-get install libfontconfig1 libfontconfig1-dev
>cd ~
>wget https://bitbucket.org/ariya/phantomjs/downloads/phantomjs-2.1.1-linux-x86_64.tar.bz2
>sudo tar xvjf phantomjs-2.1.1-linux-x86_64.tar.bz2
>sudo mv phantomjs-2.1.1-linux-x86_64 /usr/local/share
>sudo ln -sf /usr/local/share/phantomjs-2.1.1-linux-x86_64/bin/phantomjs /usr/local/bin

Or use npm:

>npm install phantomjs2

Installation of Selenium

Install selenium by using pip install selenium

Let’s go to coding:

from selenium import webdriver  # Import selenium web driver

driver = webdriver.PhantomJs()  # PhantomJs webdriver

driver.get('https://google.com')

driver.save_screenshot('google.jpg')

Done! your image will be saved as google.jpg.

Using with Django

Suppose you want to store the image in Django ImageFile, you can use the following code:

screenshot = driver.screenshot_as_base64()  # binary image

my_model = MyModel()
my_model.image_field = ContentFile(screenshot, 'google.jpg')
MyModel.save()

That’s all for now. Cheers!!

Last updated: Apr 05, 2024