diff options
Diffstat (limited to 'ishtar_common/models.py')
| -rw-r--r-- | ishtar_common/models.py | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/ishtar_common/models.py b/ishtar_common/models.py index a57e814f0..9f7261010 100644 --- a/ishtar_common/models.py +++ b/ishtar_common/models.py @@ -40,6 +40,7 @@ from django.core.validators import validate_slug from django.core.urlresolvers import reverse, NoReverseMatch from django.db.utils import DatabaseError from django.db.models import Q, Max, Count +from django.db.models.base import ModelBase from django.db.models.signals import post_save, pre_delete from django.utils.translation import ugettext_lazy as _, ugettext from django.utils.safestring import SafeUnicode, mark_safe @@ -993,6 +994,8 @@ def get_model_fields(model): options = model._meta for field in sorted(options.fields + options.many_to_many): fields[field.name] = field + if hasattr(model, 'get_extra_fields'): + fields.update(model.get_extra_fields()) return fields def import_class(full_path_classname): @@ -1075,6 +1078,8 @@ def get_associated_model(parent_model, keys): field = get_model_fields(OBJECT_CLS)[item] if hasattr(field, 'rel') and hasattr(field.rel, 'to'): model = field.rel.to + if type(field) == ModelBase: + model = field else: return get_associated_model(model, keys[1:]) return model @@ -1259,6 +1264,9 @@ TARGET_MODELS = [ ('archaeological_operations.models.OperationType', _(u"Operation type")), ('archaeological_operations.models.Period', _(u"Period")), ('archaeological_context_records.models.Unit', _(u"Unit")), + ('archaeological_finds.models.MaterialType', _(u"Material")), + ('archaeological_finds.models.ConservatoryState', _(u"Conservatory state")), + ('archaeological_finds.models.PreservationType', _(u"Preservation type")), ] TARGET_MODELS_KEYS = [tm[0] for tm in TARGET_MODELS] |