Python/Django - Take screenshots of pages that require authentication with Selenium

screenshot_example.py

#this would be in your view.py

from pyvirtualdisplay import Display
from selenium import webdriver

def take_screenshot(request):
	HOST = request.get_host()

	display = Display(visible=0, size=(1920,1440))
	display.start()

	browser = webdriver.Firefox()

	#before we can add a cookie, we need to access the page first
	browser.get(HOST)
	browser.add_cookie({ 'name': 'sessionid', 'value': request.COOKIES['sessionid'], 'path': '/', 'domain': HOST })

	#assuming your user is a SuperAdmin
	browser.get(HOST+'/admin/')
	browser.save_screenshot(MEDIA_ROOT+'/test.png')

	browser.quit()
	display.stop()