Model management of sources

1 files changed, 30 insertions, 6 deletions

diff --git a/ishtar/furnitures/admin.py b/ishtar/furnitures/admin.py
index 8a47ea429..3a93f1852 100644
--- a/ishtar/furnitures/admin.py
+++ b/ishtar/furnitures/admin.py
@@ -80,6 +80,14 @@ class OperationAdmin(HistorizedObjectAdmin):
 
 admin.site.register(models.Operation, OperationAdmin)
 
+class OperationSourceAdmin(admin.ModelAdmin):
+    list_display = ('operation', 'title', 'source_type',)
+    list_filter = ('source_type',)
+    search_fields = ('title', 'operation__name')
+    model = models.OperationSource
+
+admin.site.register(models.OperationSource, OperationSourceAdmin)
+
 class ParcelAdmin(HistorizedObjectAdmin):
     list_display = ('section', 'parcel_number', 'operation', 'associated_file')
     search_fields = ('operation__name',)
@@ -110,13 +118,13 @@ class ContextRecordAdmin(HistorizedObjectAdmin):
 
 admin.site.register(models.ContextRecord, ContextRecordAdmin)
 
-class SourceAdmin(admin.ModelAdmin):
-    list_display = ('title', 'source_type',)
+class ContextRecordSourceAdmin(admin.ModelAdmin):
+    list_display = ('context_record', 'title', 'source_type',)
     list_filter = ('source_type',)
-    search_fields = ('title',)
-    model = models.Source
+    search_fields = ('title', )
+    model = models.ContextRecordSource
 
-admin.site.register(models.Source, SourceAdmin)
+admin.site.register(models.ContextRecordSource, ContextRecordSourceAdmin)
 
 class BaseItemAdmin(HistorizedObjectAdmin):
     list_display = ('label', 'context_record', 'is_isolated')
@@ -134,6 +142,14 @@ class ItemAdmin(HistorizedObjectAdmin):
 
 admin.site.register(models.Item, ItemAdmin)
 
+class ItemSourceAdmin(admin.ModelAdmin):
+    list_display = ('item', 'title', 'source_type',)
+    list_filter = ('source_type',)
+    search_fields = ('title', )
+    model = models.ItemSource
+
+admin.site.register(models.ItemSource, ItemSourceAdmin)
+
 class WarehouseAdmin(HistorizedObjectAdmin):
     list_display = ('name', 'warehouse_type', 'town')
     list_filter = ('warehouse_type',)
@@ -176,7 +192,7 @@ class TownAdmin(admin.ModelAdmin):
 admin.site.register(models.Town, TownAdmin)
 
 class AuthorAdmin(admin.ModelAdmin):
-    list_display = ['person', 'source', 'author_type']
+    list_display = ['person', 'author_type']
     list_filter = ("author_type",)
     model = models.Author
 
@@ -196,6 +212,14 @@ class TreatmentAdmin(HistorizedObjectAdmin):
 
 admin.site.register(models.Treatment, TreatmentAdmin)
 
+class TreatmentSourceAdmin(admin.ModelAdmin):
+    list_display = ('treatment', 'title', 'source_type',)
+    list_filter = ('source_type',)
+    search_fields = ('title',)
+    model = models.TreatmentSource
+
+admin.site.register(models.TreatmentSource, TreatmentSourceAdmin)
+
 basic_models = [models.PersonType, models.IshtarUser, models.FileType,
                 models.OperationType, models.DatingType, models.DatingQuality,
                 models.SourceType, models.MaterialType, models.ParcelOwner,