Document upload and metadata input working

apps/common/forms.py

@@ -0,0 +1,109 @@
+from django import forms 
+from django.utils.safestring import mark_safe
+from django.utils.translation import ugettext as _
+from django.db import models
+import types
+
+
+def return_attrib(obj, attrib, arguments=None):
+    try:
+        result = reduce(getattr, attrib.split("."), obj)
+        if isinstance(result, types.MethodType):
+            if arguments:
+                return result(**arguments)
+            else:
+                return result()
+        else:
+            return result
+    except Exception, err:
+        if settings.DEBUG:
+            return "Attribute error: %s; %s" % (attrib, err)
+        else:
+            pass
+
+
+class DetailSelectMultiple(forms.widgets.SelectMultiple):
+    def __init__(self, queryset=None, *args, **kwargs):
+        self.queryset=queryset
+        super(DetailSelectMultiple, self).__init__(*args, **kwargs)
+            
+    def render(self, name, value, attrs=None, choices=()):
+        if value is None: value = ''
+        #final_attrs = self.build_attrs(attrs, name=name)
+        output = u'<ul class="list">'
+        options = None
+        if value:
+            if getattr(value, '__iter__', None):
+                options = [(index, string) for index, string in self.choices if index in value]
+            else:
+                options = [(index, string) for index, string in self.choices if index == value]
+        else:
+            if self.choices:
+                if self.choices[0] != (u'', u'---------') and value != []:
+                    options = [(index, string) for index, string in self.choices]
+                
+        if options:
+            for index, string in options:
+                if self.queryset:
+                    try:
+                        output += u'<li><a href="%s">%s</a></li>' % (self.queryset.get(pk=index).get_absolute_url(), string)
+                    except AttributeError:
+                        output += u'<li>%s</li>' % (string)
+                else:
+                 output += u'<li>%s</li>' % string
+        else:
+            output += u'<li>%s</li>' % _(u"None")
+        return mark_safe(output + u'</ul>\n')
+
+
+class DetailForm(forms.ModelForm):
+    def __init__(self, extra_fields=None, *args, **kwargs):
+        super(DetailForm, self).__init__(*args, **kwargs)
+        if extra_fields:
+            for extra_field in extra_fields:
+                result = return_attrib(self.instance, extra_field['field'])
+                label = 'label' in extra_field and extra_field['label'] or None
+                #TODO: Add others result types <=> Field types
+                if isinstance(result, models.query.QuerySet):
+                    self.fields[extra_field['field']]=forms.ModelMultipleChoiceField(queryset=result, label=label)
+
+        for field_name, field in self.fields.items():
+            if isinstance(field.widget, forms.widgets.SelectMultiple):
+                self.fields[field_name].widget = DetailSelectMultiple(
+                    choices=field.widget.choices,
+                    attrs=field.widget.attrs,
+                    queryset=getattr(field, 'queryset', None),
+                )
+                self.fields[field_name].help_text=''
+            elif isinstance(field.widget, forms.widgets.Select):
+                self.fields[field_name].widget = DetailSelectMultiple(
+                    choices=field.widget.choices,
+                    attrs=field.widget.attrs,
+                    queryset=getattr(field, 'queryset', None),
+                )
+                self.fields[field_name].help_text=''
+
+
+class GenericConfirmForm(forms.Form):
+    pass
+
+
+class GenericAssignRemoveForm(forms.Form):
+    left_list = forms.ModelMultipleChoiceField(required=False, queryset=None)
+    right_list = forms.ModelMultipleChoiceField(required=False, queryset=None)
+    def __init__(self, left_list_qryset=None, right_list_qryset=None, left_filter=None, *args, **kwargs):
+        super(GenericAssignRemoveForm, self).__init__(*args, **kwargs)
+        if left_filter:
+            self.fields['left_list'].queryset = left_list_qryset.filter(*left_filter)
+        else:
+            self.fields['left_list'].queryset = left_list_qryset
+
+        self.fields['right_list'].queryset = right_list_qryset
+
+        
+class FilterForm(forms.Form):
+    def __init__(self, list_filters, *args, **kwargs):
+        super(FilterForm, self).__init__(*args, **kwargs)
+        for list_filter in list_filters:
+            label = list_filter.get('title', list_filter['name'])
+            self.fields[list_filter['name']] = forms.ModelChoiceField(queryset=list_filter['queryset'], label=label[0].upper() + label[1:], required=False)