web: Rename base module

[ipfire.org.git] / src / web / handlers_iuse.py

#!/usr/bin/python

import datetime
import logging
import os
import random
import tornado.web

from . import base

class IUseImage(base.BaseHandler):
	expires = 3600 * 3

	def write_error(self, status_code, **kwargs):
		"""
			Select a random image from the errors directory
			and return the content.
		"""
		self.set_expires(3600)

		self.set_header("Content-Type", "image/png")

		template_path = self.application.settings.get("template_path", "")
		template_path = os.path.join(template_path, "i-use", "errors")

		images = os.listdir(template_path)
		if images:
			image = random.choice(images)
			image = os.path.join(template_path, image)

			imgdata = None
			with open(image, "rb") as f:
				imgdata = f.read()

			self.finish(imgdata)

	def get(self, profile_id, image_id):
		image = None

		when = self.get_argument_date("when", None)

		logging.info("Rendering new image for profile: %s" % profile_id)

		profile = self.fireinfo.get_profile_with_data(profile_id, when=when)
		if not profile:
			raise tornado.web.HTTPError(404, "Profile '%s' was not found." % profile_id)

		image_cls = self.iuse.get_imagetype(image_id)
		if not image_cls:
			raise tornado.web.HTTPError(404, "Image class is unknown: %s" % image_id)

		# Render the image
		image = image_cls(self.backend, self, profile).to_string()

		# Cache generate images for 3 hours
		self.set_expires(3600 * 3)

		self.set_header("Content-Type", "image/png")
		self.write(image)