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