Merge branch 'stable'

1 files changed, 8 insertions, 0 deletions

diff --git a/ishtar_common/models.py b/ishtar_common/models.py
index 2d723b60a..a21df0529 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
@@ -1006,6 +1007,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):
@@ -1088,6 +1091,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
@@ -1272,6 +1277,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]