src/web/iuse.py

   1 #!/usr/bin/python
   2
   3 import datetime
   4 import logging
   5 import os
   6 import random
   7 import tornado.web
   8
   9 from . import base
  10
  11 class Image(base.BaseHandler):
  12         def write_error(self, status_code, **kwargs):
  13                 """
  14                         Select a random image from the errors directory
  15                         and return the content.
  16                 """
  17                 self.set_expires(3600)
  18
  19                 self.set_header("Content-Type", "image/png")
  20
  21                 template_path = self.application.settings.get("template_path", "")
  22                 template_path = os.path.join(template_path, "i-use", "errors")
  23
  24                 images = os.listdir(template_path)
  25                 if images:
  26                         image = random.choice(images)
  27                         image = os.path.join(template_path, image)
  28
  29                         imgdata = None
  30                         with open(image, "rb") as f:
  31                                 imgdata = f.read()
  32
  33                         self.finish(imgdata)
  34
  35         def get(self, profile_id, image_id):
  36                 image = None
  37
  38                 when = self.get_argument_date("when", None)
  39
  40                 logging.info("Rendering new image for profile: %s" % profile_id)
  41
  42                 profile = self.fireinfo.get_profile_with_data(profile_id, when=when)
  43                 if not profile:
  44                         raise tornado.web.HTTPError(404, "Profile '%s' was not found." % profile_id)
  45
  46                 image_cls = self.iuse.get_imagetype(image_id)
  47                 if not image_cls:
  48                         raise tornado.web.HTTPError(404, "Image class is unknown: %s" % image_id)
  49
  50                 # Render the image
  51                 image = image_cls(self.backend, self, profile).to_string()
  52
  53                 # Cache generate images for 3 hours
  54                 self.set_expires(3600 * 3)
  55
  56                 self.set_header("Content-Type", "image/png")
  57                 self.write(image)