import tidy

options = dict(
		add_xml_decl=False,
		doctype='omit',
		indent=1,
		tidy_mark=0,
		hide_comments=False,
		wrap=100,
		new_blocklevel_tags='article,header,footer,section,nav',
		new_inline_tags='video,audio,canvas,ruby,rt,rp',
		input_xml=1,
		output_xhtml=1,
		char_encoding='utf8',
)

class PrettifyMiddleware(object):
	"""Prettify middleware"""

	def process_response(self, request, response):
		if response.__getitem__('Content-Type').split(';', 1)[0] == 'text/html':
			content = response.content
			content = str(tidy.parseString(content, **options))
			response.content = content
		return response