Grids: manage multiple values in the same column

1 files changed, 37 insertions, 25 deletions

diff --git a/ishtar_common/widgets.py b/ishtar_common/widgets.py
index 509907034..2aa9b12f6 100644
--- a/ishtar_common/widgets.py
+++ b/ishtar_common/widgets.py
@@ -525,34 +525,46 @@ class JQueryJqGrid(forms.RadioSelect):
         self.source_full = source_full
 
     def get_cols(self):
-        col_names, extra_cols = [], []
-        for field_name in getattr(self.associated_model, self.table_cols):
-            field = self.associated_model
-            if type(field_name) in (list, tuple):
-                field_name = field_name[0]
+        jq_col_names, extra_cols = [], []
+        col_labels = {}
+        if hasattr(self.associated_model, self.table_cols + '_LBL'):
+            col_labels = getattr(self.associated_model,
+                                 self.table_cols + '_LBL')
+        for col_names in getattr(self.associated_model, self.table_cols):
             field_verbose_names = []
-            keys = field_name.split('.')
-            field_verbose_name = ""
-            for key in keys:
-                if hasattr(field, 'rel') and field.rel:
-                    field = field.rel.to
-                try:
-                    field = field._meta.get_field(key)
-                    field_verbose_name = field.verbose_name
-                    field_name = field.name
-                except (fields.FieldDoesNotExist, AttributeError):
-                    if hasattr(field, key + '_lbl'):
-                        field_name = key
-                        field_verbose_name = getattr(field, key + '_lbl')
-                    else:
-                        continue
-            field_verbose_names.append(unicode(field_verbose_name))
-            col_names.append(u'"%s"' % settings.JOINT.join(
-                [f for f in field_verbose_names if f]))
+            field_verbose_name, field_name = "", ""
+            if type(col_names) not in (list, tuple):
+                col_names = [col_names]
+            for col_name in col_names:
+                field = self.associated_model
+                keys = col_name.split('.')
+                f_name = ''
+                for key in keys:
+                    if hasattr(field, 'rel') and field.rel:
+                        field = field.rel.to
+                    try:
+                        field = field._meta.get_field(key)
+                        field_verbose_name = field.verbose_name
+                        f_name = field.name
+                    except (fields.FieldDoesNotExist, AttributeError):
+                        if hasattr(field, key + '_lbl'):
+                            f_name = key
+                            field_verbose_name = getattr(field, key + '_lbl')
+                        else:
+                            continue
+                if field_name:
+                    field_name += "__"
+                field_name += f_name
+                field_verbose_names.append(unicode(field_verbose_name))
+            if field_name in col_labels:
+                jq_col_names.append(u'"%s"' % unicode(col_labels[field_name]))
+            else:
+                jq_col_names.append(u'"%s"' % settings.JOINT.join(
+                    [f for f in field_verbose_names if f]))
             extra_cols.append(self.COL_TPL % {'idx': field_name})
-        col_names = col_names and ", ".join(col_names) or ""
+        jq_col_names = jq_col_names and ", ".join(jq_col_names) or ""
         extra_cols = extra_cols and ", ".join(extra_cols) or ""
-        return col_names, extra_cols
+        return jq_col_names, extra_cols
 
     def render(self, name, value=None, attrs=None):
         t = loader.get_template('blocks/form_snippet.html')