From d568719c187a7418f56f17e789af98a51674a0b3 Mon Sep 17 00:00:00 2001 From: Roberto Rosario Date: Wed, 16 Mar 2011 04:50:36 -0400 Subject: [PATCH] Get attributes for object or parent of object --- apps/common/templatetags/attribute_tags.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/apps/common/templatetags/attribute_tags.py b/apps/common/templatetags/attribute_tags.py index 2fafff9a97..cea6f1a49c 100644 --- a/apps/common/templatetags/attribute_tags.py +++ b/apps/common/templatetags/attribute_tags.py @@ -12,5 +12,9 @@ def object_property(value, arg): return return_attrib(value, arg) @register.filter -def get_model_list_columns(value): - return model_list_columns.get(type(value), []) +def get_model_list_columns(obj): + for key, value in model_list_columns.items(): + if isinstance(obj, key): + return value + + return []