philo/admin/widgets.py

   1 from django import forms
   2 from django.conf import settings
   3 from django.contrib.admin.widgets import url_params_from_lookup_dict
   4 from django.utils.html import escape
   5 from django.utils.safestring import mark_safe
   6 from django.utils.text import truncate_words
   7 from django.utils.translation import ugettext as _
   8
   9
  10 class ModelLookupWidget(forms.TextInput):
  11         # is_hidden = False
  12
  13         def __init__(self, content_type, attrs=None, limit_choices_to=None):
  14                 self.content_type = content_type
  15                 self.limit_choices_to = limit_choices_to
  16                 super(ModelLookupWidget, self).__init__(attrs)
  17
  18         def render(self, name, value, attrs=None):
  19                 related_url = '../../../%s/%s/' % (self.content_type.app_label, self.content_type.model)
  20                 params = url_params_from_lookup_dict(self.limit_choices_to)
  21                 if params:
  22                         url = u'?' + u'&amp;'.join([u'%s=%s' % (k, v) for k, v in params.items()])
  23                 else:
  24                         url = u''
  25                 if attrs is None:
  26                         attrs = {}
  27                 if "class" not in attrs:
  28                         attrs['class'] = 'vForeignKeyRawIdAdminField'
  29                 output = [super(ModelLookupWidget, self).render(name, value, attrs)]
  30                 output.append('<a href="%s%s" class="related-lookup" id="lookup_id_%s" onclick="return showRelatedObjectLookupPopup(this);">' % (related_url, url, name))
  31                 output.append('<img src="%simg/admin/selector-search.gif" width="16" height="16" alt="%s" />' % (settings.ADMIN_MEDIA_PREFIX, _('Lookup')))
  32                 output.append('</a>')
  33                 if value:
  34                         value_class = self.content_type.model_class()
  35                         try:
  36                                 value_object = value_class.objects.get(pk=value)
  37                                 output.append('&nbsp;<strong>%s</strong>' % escape(truncate_words(value_object, 14)))
  38                         except value_class.DoesNotExist:
  39                                 pass
  40                 return mark_safe(u''.join(output))