webapp/handlers_iuse.py

   1 #!/usr/bin/python
   2
   3 import logging
   4 import os
   5 import random
   6 import tornado.web
   7
   8 from handlers_base import *
   9 import backend
  10
  11 class IUseImage(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_header("Content-Type", "image/png")
  18
  19                 template_path = self.application.settings.get("template_path", "")
  20                 template_path = os.path.join(template_path, "i-use", "errors")
  21
  22                 images = os.listdir(template_path)
  23                 if images:
  24                         image = random.choice(images)
  25                         image = os.path.join(template_path, image)
  26
  27                         imgdata = None
  28                         with open(image, "rb") as f:
  29                                 imgdata = f.read()
  30
  31                         self.finish(imgdata)
  32
  33         def get(self, profile_id, image_id):
  34                 image = None
  35
  36                 when = self.get_argument_date("when", None)
  37
  38                 # Try to get the image from memcache. If we have a cache miss we
  39                 # build a new one.
  40                 mem_id = "iuse-%s-%s-%s" % (profile_id, image_id, self.locale.code)
  41                 if when:
  42                         mem_id += "-%s" % when.isoformat()
  43
  44                 cache = self.get_argument("cache", "true")
  45                 if cache == "true":
  46                         image = self.memcached.get(mem_id)
  47
  48                 if image:
  49                         logging.debug("Got image from cache for profile: %s" % profile_id)
  50                 else:
  51                         logging.info("Rendering new image for profile: %s" % profile_id)
  52
  53                         image_cls = self.iuse.get_imagetype(image_id)
  54                         if not image_cls:
  55                                 raise tornado.web.HTTPError(404, "Image class is unknown: %s" % image_id)
  56
  57                         profile = self.fireinfo.get_profile(profile_id, when=when)
  58                         if not profile:
  59                                 raise tornado.web.HTTPError(404, "Profile '%s' was not found." % profile_id)
  60
  61                         # Render the image
  62                         image = image_cls(self.backend, self, profile).to_string()
  63
  64                         # Save the image to the memcache for 60 minutes
  65                         self.memcached.set(mem_id, image, 3600)
  66
  67                 self.set_header("Content-Type", "image/png")
  68                 self.write(image)
  69