Changeset 9b911db for asadb/util/db_form_utils.py

Timestamp:

Aug 9, 2011, 3:01:19 AM (14 years ago)

Author:

Alex Dehnert <adehnert@…>

Branches:

master, space-access, stable, stage, test-hooks

Children:

d8290de

Parents:

628e06c

git-author:

Alex Dehnert <adehnert@…> (08/09/11 03:01:19)

git-committer:

Alex Dehnert <adehnert@…> (08/09/11 03:01:19)

Message:

Make StaticWidget? work with Select fields too

File:

• asadb/util/db_form_utils.py (modified) (4 diffs)

asadb/util/db_form_utils.py

@@ -1 @@
-from django.forms.widgets import Widget, HiddenInput
+from django.forms.widgets import Widget, Select, HiddenInput
 from django.utils.safestring import mark_safe
 from django.utils.html import conditional_escape
+from itertools import chain
 
 class StaticWidget(Widget):
@@ -13 +14 @@
         self.value = kwargs['value']
         del kwargs['value']
+        if 'choices' in kwargs:
+            if kwargs['choices'] is not None:
+                self.choices = kwargs['choices']
+            else:
+                self.choices = ()
+            del kwargs['choices']
+        else:
+            self.choices = ()
         super(StaticWidget, self).__init__(*args, **kwargs)
         self.inner = HiddenInput()
@@ -19 +28 @@
         return self.value
 
-    def render(self, name, value, attrs=None, ):
+    def render(self, name, value, attrs=None, choices=(), ):
         if value is None:
             value = ''
+        label = value
+        the_choices = chain(self.choices, choices)
+        for choice_value, choice_label in the_choices:
+            if choice_value == value:
+                label = choice_label
         # We have this inner hidden widget because that allows us to nicely
         # handle a transition from StaticWidget to another widget type
@@ -27 +41 @@
         # authenticated)
         hidden_render = self.inner.render(name, value, attrs=attrs)
-        return mark_safe(conditional_escape(value) + hidden_render)
+        return mark_safe(conditional_escape(label) + hidden_render)
+
+    @classmethod
+    def replace_widget(cls, formfield, value):
+        choices = None
+        choices = getattr(formfield.widget, 'choices', ())
+        formfield.widget = cls(value=value, choices=choices)