Added a new template tag to override the painting of the admin rows

editor/templatetags/admin_tree_list.py

@@ -0,0 +1,127 @@
+from django.contrib.admin.templatetags.admin_list import result_headers
+from django.template import Library
+from django.core.exceptions import ObjectDoesNotExist
+from django.contrib.admin.util import lookup_field, display_for_field, label_for_field
+from django.utils.encoding import smart_unicode, force_unicode
+from django.utils.safestring import mark_safe
+from django.utils.html import escape, conditional_escape
+from django.db import models
+
+
+register = Library()
+
+def items_for_tree_result(cl, result, form):
+    """
+    Generates the actual list of data.
+    """
+    first = True
+    pk = cl.lookup_opts.pk.attname
+    for field_name in cl.list_display:
+        row_class = ''
+        try:
+            f, attr, value = lookup_field(field_name, result, cl.model_admin)
+        except (AttributeError, ObjectDoesNotExist):
+            result_repr = EMPTY_CHANGELIST_VALUE
+        else:
+            if f is None:
+                allow_tags = getattr(attr, 'allow_tags', False)
+                boolean = getattr(attr, 'boolean', False)
+                if boolean:
+                    allow_tags = True
+                    result_repr = _boolean_icon(value)
+                else:
+                    result_repr = smart_unicode(value)
+                # Strip HTML tags in the resulting text, except if the
+                # function has an "allow_tags" attribute set to True.
+                if not allow_tags:
+                    result_repr = escape(result_repr)
+                else:
+                    result_repr = mark_safe(result_repr)
+            else:
+                if value is None:
+                    result_repr = EMPTY_CHANGELIST_VALUE
+                if isinstance(f.rel, models.ManyToOneRel):
+                    result_repr = escape(getattr(result, f.name))
+                else:
+                    result_repr = display_for_field(value, f)
+                if isinstance(f, models.DateField) or isinstance(f, models.TimeField):
+                    row_class = ' class="nowrap"'
+            if first:
+                try:
+                    f, attr, value = lookup_field('action_checkbox', result, cl.model_admin)
+                    result_repr = mark_safe("%s%s" % (value, result_repr))
+                    if row_class:
+                        row_class = "%s%s" % (row_class[:-1],' disclosure"')
+                    else:
+                        row_class = ' class="disclosure"'
+                except (AttributeError, ObjectDoesNotExist):
+                    pass
+
+        if force_unicode(result_repr) == '':
+            result_repr = mark_safe(' ')
+        # If list_display_links not defined, add the link tag to the first field
+        if (first and not cl.list_display_links) or field_name in cl.list_display_links:
+            table_tag = 'td' #{True:'th', False:'td'}[first]
+            url = cl.url_for_result(result)
+            # Convert the pk to something that can be used in Javascript.
+            # Problem cases are long ints (23L) and non-ASCII strings.
+            if cl.to_field:
+                attr = str(cl.to_field)
+            else:
+                attr = pk
+            value = result.serializable_value(attr)
+            result_id = repr(force_unicode(value))[1:]
+            first = False
+            yield mark_safe(u'<%s%s><a href="%s"%s>%s</a></%s>' % \
+                (table_tag, row_class, url, (cl.is_popup and ' onclick="opener.dismissRelatedLookupPopup(window, %s); return false;"' % result_id or ''), conditional_escape(result_repr), table_tag))
+        else:
+            # By default the fields come from ModelAdmin.list_editable, but if we pull
+            # the fields out of the form instead of list_editable custom admins
+            # can provide fields on a per request basis
+            if form and field_name in form.fields:
+                bf = form[field_name]
+                result_repr = mark_safe(force_unicode(bf.errors) + force_unicode(bf))
+            else:
+                result_repr = conditional_escape(result_repr)
+            yield mark_safe(u'<td%s>%s</td>' % (row_class, result_repr))
+    if form and not form[cl.model._meta.pk.name].is_hidden:
+        yield mark_safe(u'<td>%s</td>' % force_unicode(form[cl.model._meta.pk.name]))
+
+class TreeList(list):
+    pass
+
+def tree_results(cl):
+    if cl.formset:
+        for res, form in zip(cl.result_list, cl.formset.forms):
+            result = TreeList(items_for_tree_result(cl, res, form))
+            if hasattr(res, 'pk'):
+                result.pk = res.pk
+                if res.parent:
+                    result.parent_pk = res.parent.pk
+                else:
+                    res.parent_pk = None
+            yield result
+    else:
+        for res in cl.result_list:
+            result = TreeList(items_for_tree_result(cl, res, None))
+            if hasattr(res, 'pk'):
+                result.pk = res.pk
+                if res.parent:
+                    result.parent_pk = res.parent.pk
+                else:
+                    res.parent_pk = None
+            yield result
+
+def result_tree_list(cl):
+    """
+    Displays the headers and data list together
+    """
+    import django
+    result = {'cl': cl,
+            'result_headers': list(result_headers(cl)),
+            'results': list(tree_results(cl))}
+    if django.VERSION[1] > 2:
+        from django.contrib.admin.templatetags.admin_list import result_hidden_fields
+        result['result_hidden_fields'] = list(result_hidden_fields(cl))
+    return result
+result_tree_list = register.inclusion_tag("admin/editor/tree_list_results.html")(result_tree_list)