HTML2Image

HTML2Image is a lightweight Python package that acts as a wrapper around the headless mode of existing web browsers to generate images from URLs and from HTML+CSS strings or files.

This package has been tested on Windows, Ubuntu (desktop and server) and MacOS. It is currently in a work in progress stage. If you encounter any problem or difficulties while using it, feel free to open an issue on the GitHub page of this project. Feedback is also welcome!

Principle

Most web browsers have a Headless Mode, which is a way to run them without displaying any graphical interface. Headless mode is mainly used for automated testings but also comes in handy if you want to take screenshots of web pages that are exact replicas of what you would see on your screen if you were using the browser yourself.

However, for the sake of taking screenshots, headless mode is not very convenient to use. HTML2Image aims to hide the inconveniences of the browsers' headless modes while adding useful features such as allowing to create an image from as little as a string.

For more information about headless modes :

• (Chrome) https://developers.google.com/web/updates/2017/04/headless-chrome
• (Firefox) https://developer.mozilla.org/en-US/docs/Mozilla/Firefox/Headless_mode

Installation

HTML2Image is published on PyPI and can be installed through pip:

pip install --upgrade html2image

In addition to this package, at least one of the following browsers must be installed on your machine :

• Google Chrome (Windows, MacOS)
• Chromium Brower (Linux)

Usage

First, import the package and instantiate it

from html2image import Html2Image
hti = Html2Image()

hti = Html2Image(size=(500, 200))

hti.size = (500, 200)

Then take a screenshot

The screenshot method is the basis of this package, most of the time, you won't need to use anything else. It can take screenshots of a lot of things :

• URLs via the url parameter;
• HTML and CSS files via the html_file and css_file parameters;
• HTML and CSS strings via the html_str and css_str parameters;
• and "other" types of files via the other_file parameter (try it with .svg files!).

And you can also (optional):

• Change the size of the screenshots using the size parameter;
• Save the screenshots as a specific name using the save_as parameter.

N.B. : The screenshot method returns a list containing the path(s) of the screenshot(s) it took.

A few examples

• URL to image

hti.screenshot(url='https://www.python.org', save_as='python_org.png')

• HTML & CSS strings to image

html = """<h1> An interesting title </h1> This page will be red"""
css = "body {background: red;}"

hti.screenshot(html_str=html, css_str=css, save_as='red_page.png')

• HTML & CSS files to image

hti.screenshot(
    html_file='blue_page.html', css_file='blue_background.css',
    save_as='blue_page.png'
)

• Other files to image

hti.screenshot(other_file='star.svg')

• Change the screenshots' size

hti.screenshot(other_file='star.svg', size=(500, 500))

Click to show all the images generated with all the code above

• Change the directory to which the screenshots are saved

hti = Html2Image(output_path='my_screenshot_folder')

OR

hti.output_path = 'my_screenshot_folder'

N.B. : the output path will be changed for all future screenshots.

Use lists in place of any parameters while using the screenshot method

• Screenshot multiple objects using only one filename, or one filename per file:

# create three files from one filename
hti.screenshot(html_str=['A', 'B', 'C'], save_as='ABC.png')
# outputs ABC_0.png, ABC_1.png, ABC_2.png

# create three files from from different filenames
hti.screenshot(html_str=['A', 'B', 'C'], save_as=['A.png', 'B.png', 'C.png'])
# outputs A.png, B.png, C.png

• Take multiple screenshots with the same size

# take four screenshots with a resolution of 100*50
hti.screenshot(
    html_str=['A', 'B', 'C', 'D']
    size=(100, 50)
)

• Take multiple screenshots with different sizes

# take four screenshots with different resolutions from three given sizes
hti.screenshot(
    html_str=['A', 'B', 'C', 'D'],
    size=[(100, 50), (100, 100), (50, 50)]
)
# respectively 100*50, 100*100, 50*50, 50*50
# if not enough sizes are given, the last size in the list will be repeated

• Apply CSS string(s) to multiple HTML string(s)

# screenshot two html strings and apply css strings on both
hti.screenshot(
    html_str=['A', 'B'],
    css_str='body {background: red;}'
)

# screenshot two html strings and apply multiple css strings on both
hti.screenshot(
    html_str=['A', 'B'],
    css_str=['body {background: red;}', 'body {font-size: 50px;}']
)

# screenshot one html string and apply multiple css strings on it
hti.screenshot(
    html_str='A',
    css_str=['body {background: red;}', 'body {font-size: 50px;}']
)

• Retrieve the path of the generated file(s)
  The screenshot method returns a list containing the path(s) of the screenshot(s):

paths = hti.screenshot(
    html_str=['A', 'B', 'C'],
    save_as="letters.png",
)

print(paths)
# >>> ['D:\\myFiles\\letters_0.png', 'D:\\myFiles\\letters_1.png', 'D:\\myFiles\\letters_2.png']

Using the CLI

HTML2image comes with a Command Line Interface which you can use to generate screenshots from files and urls on the go.

The CLI is a work in progress and may be subject to changes. You can call it by typing hti or html2image into a terminal.

Testing

Only basic testing is available at the moment. To run tests, run PyTest at the root of the project:

python -m pytest

TODO List

• A nice CLI (Currently in a WIP state)
  • A better way to name the CLI's outputed files ?
• Support of other browsers, such as Firefox
• More extensive doc + comments
• PDF generation?
• Testing on push/PR with GitHub Actions
• Use threads or multiprocessing to speed up screenshot taking