Hide parcel field for UE QA - Add cultural atribution for sites

1 files changed, 15 insertions, 1 deletions

diff --git a/archaeological_operations/admin.py b/archaeological_operations/admin.py
index b6300cdf2..0aec94249 100644
--- a/archaeological_operations/admin.py
+++ b/archaeological_operations/admin.py
@@ -28,7 +28,8 @@ from django.contrib.gis.forms import PointField, MultiPolygonField, OSMWidget
 from django.utils.translation import ugettext_lazy as _
 
 from ishtar_common.apps import admin_site
-from ishtar_common.admin import HistorizedObjectAdmin, GeneralTypeAdmin
+from ishtar_common.admin import HistorizedObjectAdmin, GeneralTypeAdmin, \
+    HierarchicalTypeAdmin
 
 from . import models
 
@@ -206,6 +207,19 @@ class ParcelOwnerAdmin(HistorizedObjectAdmin):
 admin_site.register(models.ParcelOwner, ParcelOwnerAdmin)
 
 
+class CulturalAttributionTypeAdmin(HierarchicalTypeAdmin):
+    list_display = HierarchicalTypeAdmin.list_display
+    search_fields = ('label', 'parent__label', )
+    model = models.CulturalAttributionType
+    form = make_ajax_form(model, {
+        'parent': 'cultural_attribution_type',
+    })
+
+
+admin_site.register(models.CulturalAttributionType,
+                    CulturalAttributionTypeAdmin)
+
+
 general_models = [models.RemainType]
 for model in general_models:
     admin_site.register(model, GeneralTypeAdmin)