Add relation between operations

1 files changed, 21 insertions, 1 deletions

diff --git a/archaeological_operations/admin.py b/archaeological_operations/admin.py
index 3374bcef7..42f63fdcd 100644
--- a/archaeological_operations/admin.py
+++ b/archaeological_operations/admin.py
@@ -36,6 +36,7 @@ if FILES_AVAILABLE:
 
     admin.site.register(models.AdministrativeAct, AdministrativeActAdmin)
 
+
 class PeriodAdmin(admin.ModelAdmin):
     list_display = ('label', 'start_date', 'end_date', 'parent', 'available',
                     'order')
@@ -44,6 +45,7 @@ class PeriodAdmin(admin.ModelAdmin):
 
 admin.site.register(models.Period, PeriodAdmin)
 
+
 class ArchaeologicalSiteAdmin(HistorizedObjectAdmin):
     list_display = ('name', 'reference')
     search_fields = ('name', 'reference')
@@ -51,6 +53,7 @@ class ArchaeologicalSiteAdmin(HistorizedObjectAdmin):
 
 admin.site.register(models.ArchaeologicalSite, ArchaeologicalSiteAdmin)
 
+
 class OperationAdmin(HistorizedObjectAdmin):
     list_display = ['year', 'operation_code', 'start_date',
                     'excavation_end_date', 'end_date',
@@ -64,6 +67,7 @@ class OperationAdmin(HistorizedObjectAdmin):
 
 admin.site.register(models.Operation, OperationAdmin)
 
+
 class OperationSourceAdmin(admin.ModelAdmin):
     list_display = ('operation', 'title', 'source_type',)
     list_filter = ('source_type',)
@@ -72,8 +76,9 @@ class OperationSourceAdmin(admin.ModelAdmin):
 
 admin.site.register(models.OperationSource, OperationSourceAdmin)
 
+
 class ParcelAdmin(HistorizedObjectAdmin):
-    list_display = ['section', 'parcel_number', 'operation',]
+    list_display = ['section', 'parcel_number', 'operation']
     if FILES_AVAILABLE:
         list_display.append('associated_file')
     search_fields = ('operation__name',)
@@ -81,12 +86,27 @@ class ParcelAdmin(HistorizedObjectAdmin):
 
 admin.site.register(models.Parcel, ParcelAdmin)
 
+
 class OperationTypeAdmin(GeneralTypeAdmin):
     list_display = GeneralTypeAdmin.list_display + ['order', 'preventive']
     model = models.OperationType
 
 admin.site.register(models.OperationType, OperationTypeAdmin)
 
+
+class RecordRelationsAdmin(admin.ModelAdmin):
+    list_display = ('left_record', 'relation_type', 'right_record')
+    list_filter = ('relation_type',)
+
+admin.site.register(models.RecordRelations, RecordRelationsAdmin)
+
+
+class RelationTypeAdmin(admin.ModelAdmin):
+    list_display = ('label', 'txt_idx', 'available', 'symmetrical',
+                    'inverse_relation')
+
+admin.site.register(models.RelationType, RelationTypeAdmin)
+
 general_models = [models.RemainType, models.ActType]
 for model in general_models:
     admin.site.register(model, GeneralTypeAdmin)