Manage treatment / treatment files sources

1 files changed, 17 insertions, 1 deletions

diff --git a/archaeological_finds/wizards.py b/archaeological_finds/wizards.py
index f5f43f1e6..c9d492329 100644
--- a/archaeological_finds/wizards.py
+++ b/archaeological_finds/wizards.py
@@ -190,7 +190,9 @@ class TreatmentFileAdministrativeActWizard(
                 admin = AdministrativeAct.objects.get(pk=admin_id)
                 if not admin.operation:
                     return
-                return ((_(u"Operation"), unicode(admin.operation)),)
+                return (
+                    (_(u"Operation"), unicode(admin.operation)),
+                )
             except AdministrativeAct.DoesNotExist:
                 return
         return
@@ -217,3 +219,17 @@ class FindSourceDeletionWizard(DeletionWizard):
 
 class TreatmentSourceWizard(SourceWizard):
     model = models.TreatmentSource
+
+
+class TreatmentSourceDeletionWizard(DeletionWizard):
+    model = models.TreatmentSource
+    fields = ['treatment', 'title', 'source_type', 'authors']
+
+
+class TreatmentFileSourceWizard(SourceWizard):
+    model = models.TreatmentFileSource
+
+
+class TreatmentFileSourceDeletionWizard(DeletionWizard):
+    model = models.TreatmentFileSource
+    fields = ['treatment_file', 'title', 'source_type', 'authors']