Admin find: add dynamic fields

2 files changed, 16 insertions, 2 deletions

diff --git a/archaeological_finds/admin.py b/archaeological_finds/admin.py
index 45050eb2e..3aa64d1dd 100644
--- a/archaeological_finds/admin.py
+++ b/archaeological_finds/admin.py
@@ -54,6 +54,7 @@ class BaseFindAdmin(HistorizedObjectAdmin):
         'cache_short_id', 'cache_complete_id',
     ]
 
+
 admin_site.register(models.BaseFind, BaseFindAdmin)
 
 
@@ -66,7 +67,14 @@ class FindAdmin(HistorizedObjectAdmin):
     model = models.Find
     form = make_ajax_form(model, {
         'base_finds': 'base_find',
-        'container': 'container'
+        'container': 'container',
+        'container_ref': 'container',
+        'material_types': 'material_type',
+        'upstream_treatment': 'treatment',
+        'downstream_treatment': 'treatment',
+        'treatments': 'treatment',
+        'main_image': 'document',
+        'documents': 'document',
     })
     readonly_fields = HistorizedObjectAdmin.readonly_fields + [
         'datings', 'cached_label'
diff --git a/archaeological_finds/lookups.py b/archaeological_finds/lookups.py
index c1298fb83..df51f3824 100644
--- a/archaeological_finds/lookups.py
+++ b/archaeological_finds/lookups.py
@@ -1,6 +1,6 @@
 from ajax_select import register
 
-from ishtar_common.lookups import LookupChannel
+from ishtar_common.lookups import LookupChannel, TypeLookupChannel
 
 from django.db.models import Q
 from django.utils.encoding import force_text
@@ -84,3 +84,9 @@ class TreatmentFileLookup(LookupChannel):
 
     def format_item_display(self, item):
         return u"<span class='ajax-label'>%s</span>" % item.cached_label
+
+
+@register('material_type')
+class MaterialTypeLookup(TypeLookupChannel):
+    model = models.MaterialType
+