fireinfo: Fix creating images for half-empty profiles

[people/shoehn/ipfire.org.git] / webapp / backend / iuse.py

#!/usr/bin/python

from __future__ import division

import StringIO
import logging
import os.path

from PIL import Image, ImageDraw, ImageFont, PngImagePlugin

from misc import Object

image_types = []

class IUse(Object):
	def get_imagetype(self, id):
		id = int(id)

		for image_type in image_types:
			if image_type.id == id:
				return image_type


class ImageObject(Object):
	default_mode = "RGBA"
	default_size = 100, 100

	_filename = None
	_font = "Ubuntu-R.ttf"
	_font_size = 10

	def __init__(self, backend, request, profile):
		Object.__init__(self, backend)

		self.request = request
		self.profile = profile

		# Create new image
		if self.filename and os.path.exists(self.filename):
			self.open(self.filename)
		else:
			self._image = Image.new(self.default_mode, self.default_size)

		self.draw()

	def open(self, filename):
		logging.debug("Opening image as a template: %s" % filename)

		image = Image.open(filename)
		self._image = image.convert(self.default_mode)

	def save(self, filename):
		self._image.save(filename, "PNG", optimize=True)

	def to_string(self):
		f = StringIO.StringIO()

		self.save(f)

		return f.getvalue()

	@property
	def paint(self):
		if not hasattr(self, "_draw"):
			self._draw = ImageDraw.Draw(self._image)

		return self._draw

	def draw(self):
		raise NotImplementedError

	@property
	def font(self):
		if not hasattr(self, "__font"):
			fontfile = os.path.join(
				self.request.application.settings.get("template_path", ""),
				"i-use", "fonts", self._font
			)

			self.__font = ImageFont.truetype(fontfile, self._font_size, encoding="unic")

		return self.__font

	def draw_text(self, pos, text, **kwargs):
		if not kwargs.has_key("font"):
			kwargs["font"] = self.font

		return self.paint.text(pos, text, **kwargs)

	@property
	def filename(self):
		if not self._filename:
			return

		return os.path.join(
			self.request.application.settings.get("template_path", ""),
			"i-use", self._filename
		)

	@property
	def locale(self):
		return self.request.locale


#class Image1(ImageObject):
#	id = 0
#
#	default_size = 500, 50
#
#	def draw(self):
#		# Background
#		self.paint.rectangle(((0, 0), self.default_size), fill="#000000")
#
#		# Release information
#		self.paint.text((10, 10), "%s" % self.profile.release)
#
#		# Hardware information
#		hw = [
#			self.profile.cpu.model_string,
#			"Mem: %.1fG" % (self.profile.memory / 1024**2),
#		]
#
#		if self.profile.virtual:
#			virt = "V-%s" % self.profile.hypervisor.vendor
#			if self.profile.hypervisor.type == "para":
#				virt = "%s-PV" % virt
#			hw.append(virt)
#
#		if self.profile.cpu.capable_64bit:
#			hw.append("64")
#
#		self.paint.text((10, 30), "%s" % " | ".join(hw))
#
#
#IUse().add_imagetype(Image1)


class Image1(ImageObject):
	id = 0

	default_size = 500, 50

	_filename = "i-use-1.png"
	_font = "Ubuntu-R.ttf"
	_font_size = 14

	def draw(self):
		_ = self.locale.translate

		line1 = [_("%s on %s") % (self.profile.release_short, self.profile.arch),]
		line2 = []

		# Show the appliance model in the second line if available
		if self.profile.appliance:
			line2.append(self.profile.appliance)

		# Show the hypervisor vendor for virtual machines
		elif self.profile.virtual:
			if self.profile.hypervisor:
				line2.append(_("Virtualised on %s") % self.profile.hypervisor)
			else:
				line2.append(_("Running in a virtualised environment"))

		# Otherwise show some general hardware information of the machine
		else:
			if self.profile.processor:
				line2.append(self.profile.processor.friendly_string)

			line2.append(self.profile.friendly_memory)

		self.draw_text((225, 9), " | ".join(line1))
		self.draw_text((225, 27), "%s" % " - ".join(line2))


image_types.append(Image1)