Merge branch 'master' of git://github.com/lapilofu/philo

[philo.git] / admin / widgets.py

diff --git a/admin/widgets.py b/admin/widgets.py
index a1f5bae..7a47c63 100644 (file)
--- a/admin/widgets.py
+++ b/admin/widgets.py
@@ -1,5 +1,10 @@
 from django import forms
 from django.conf import settings
+from django.contrib.admin.widgets import FilteredSelectMultiple
+from django.utils.translation import ugettext as _
+from django.utils.safestring import mark_safe
+from django.utils.text import truncate_words
+from django.utils.html import escape
 
 
 class ModelLookupWidget(forms.TextInput):
@@ -26,4 +31,35 @@ class ModelLookupWidget(forms.TextInput):
                                output += '&nbsp;<strong>%s</strong>' % escape(truncate_words(value_object, 14))
                        except value_class.DoesNotExist:
                                pass
-               return mark_safe(output)
\ No newline at end of file
+               return mark_safe(output)
+
+
+class TagFilteredSelectMultiple(FilteredSelectMultiple):
+       """
+       A SelectMultiple with a JavaScript filter interface.
+
+       Note that the resulting JavaScript assumes that the jsi18n
+       catalog has been loaded in the page
+       """
+       class Media:
+               js = (settings.ADMIN_MEDIA_PREFIX + "js/core.js",
+                         settings.ADMIN_MEDIA_PREFIX + "js/SelectBox.js",
+                         settings.ADMIN_MEDIA_PREFIX + "js/SelectFilter2.js")
+               
+               if 'staticmedia' in settings.INSTALLED_APPS:
+                       import staticmedia
+                       js += (staticmedia.url('admin/js/TagCreation.js'),)
+               else:
+                       js += (settings.ADMIN_MEDIA_PREFIX + "js/TagCreation.js",)
+
+       def render(self, name, value, attrs=None, choices=()):
+               if attrs is None: attrs = {}
+               attrs['class'] = 'selectfilter'
+               if self.is_stacked: attrs['class'] += 'stacked'
+               output = [super(FilteredSelectMultiple, self).render(name, value, attrs, choices)]
+               output.append(u'<script type="text/javascript">addEvent(window, "load", function(e) {')
+               # TODO: "id_" is hard-coded here. This should instead use the correct
+               # API to determine the ID dynamically.
+               output.append(u'SelectFilter.init("id_%s", "%s", %s, "%s"); tagCreation.init("id_%s"); });</script>\n' % \
+                       (name, self.verbose_name.replace('"', '\\"'), int(self.is_stacked), settings.ADMIN_MEDIA_PREFIX, name))
+               return mark_safe(u''.join(output))
\ No newline at end of file