mygpo/web/templatetags/devices.py

   1 from django import template
   2 from django.utils.safestring import mark_safe
   3 from django.utils.translation import ugettext
   4 from django.core.urlresolvers import reverse
   5
   6 _ = ugettext
   7
   8 from mygpo.api.models import DEVICE_TYPES
   9 from mygpo.web.views.device import show
  10
  11 register = template.Library()
  12
  13 # Create a dictionary of device_type -> caption mappings
  14 DEVICE_TYPES_DICT = dict(DEVICE_TYPES)
  15
  16 # This dictionary maps device types to their icon files
  17 DEVICE_TYPE_ICONS = {
  18         'desktop': 'computer.png',
  19         'laptop': 'stock_notebook.png',
  20         'mobile': 'stock_cell-phone.png',
  21         'server': 'server.png',
  22         'other': 'audio-x-generic.png',
  23 }
  24
  25 @register.filter
  26 def device_type(device):
  27     return DEVICE_TYPES_DICT.get(device.type, _('Unknown'))
  28
  29 @register.filter
  30 def device_icon(device, size=16):
  31     icon = DEVICE_TYPE_ICONS.get(device.type, None)
  32     caption = DEVICE_TYPES_DICT.get(device.type, None)
  33
  34     if icon is not None and caption is not None:
  35         caption = ugettext(caption)
  36         html = ('<img src="/media/%(size)dx%(size)d/%(icon)s" '+
  37                 'alt="%(caption)s" class="device_icon"/>') % locals()
  38         return mark_safe(html)
  39
  40     return ''
  41
  42 @register.filter
  43 def device_list(devices):
  44     return mark_safe('<br/>'.join([ '<a href="%s">%s&nbsp;%s</a>' % (reverse(show, args=[d.id]), device_icon(d), d.name.replace(' ', '&nbsp;')) for d in devices]))
  45