Manage treatment / treatment files sources

1 files changed, 80 insertions, 2 deletions

diff --git a/archaeological_finds/views.py b/archaeological_finds/views.py
index 35c7d5976..2b5aef79c 100644
--- a/archaeological_finds/views.py
+++ b/archaeological_finds/views.py
@@ -523,7 +523,17 @@ def treatmentfile_administrativeacttreatmentfile_modify(request, pk):
                     'admact-treatmentfle_admacttreatmentfle_modification'
             }))
 
-"""
+# sources
+
+show_treatmentsource = show_item(models.TreatmentSource, 'treatmentsource')
+get_treatmentsource = get_item(models.TreatmentSource, 'get_treatmentsource',
+                               'treatmentsource')
+
+treatment_source_search_wizard = SearchWizard.as_view([
+    ('selec-treatment_source_search', TreatmentSourceFormSelection)],
+    label=_(u"Treatment: source search"),
+    url_name='treatment_source_search',)
+
 treatment_source_creation_wizard = TreatmentSourceWizard.as_view([
     ('selec-treatment_source_creation', SourceTreatmentFormSelection),
     ('source-treatment_source_creation', SourceForm),
@@ -531,4 +541,72 @@ treatment_source_creation_wizard = TreatmentSourceWizard.as_view([
     ('final-treatment_source_creation', FinalForm)],
     url_name='treatment_source_creation',)
 
-"""
+treatment_source_modification_wizard = TreatmentSourceWizard.as_view([
+    ('selec-treatment_source_modification', TreatmentSourceFormSelection),
+    ('source-treatment_source_modification', SourceForm),
+    ('authors-treatment_source_modification', AuthorFormset),
+    ('final-treatment_source_modification', FinalForm)],
+    label=_(u"Treatment: source modification"),
+    url_name='treatment_source_modification',)
+
+
+def treatment_source_modify(request, pk):
+    treatment_source_modification_wizard(request)
+    TreatmentSourceWizard.session_set_value(
+        request, 'selec-treatment_source_modification', 'pk', pk, reset=True)
+    return redirect(reverse(
+        'treatment_source_modification',
+        kwargs={'step': 'source-treatment_source_modification'}))
+
+treatment_source_deletion_wizard = TreatmentSourceDeletionWizard.as_view([
+    ('selec-treatment_source_deletion', TreatmentSourceFormSelection),
+    ('final-treatment_source_deletion', SourceDeletionForm)],
+    label=_(u"Treatment: source deletion"),
+    url_name='treatment_source_deletion',)
+
+# treatment file sources
+
+show_treatmentfilesource = show_item(models.TreatmentFileSource,
+                                     'treatmentfilesource')
+get_treatmentfilesource = get_item(
+    models.TreatmentFileSource, 'get_treatmentfilesource',
+    'treatmentfilesource')
+
+treatmentfile_source_search_wizard = SearchWizard.as_view([
+    ('selec-treatmentfile_source_search', TreatmentFileSourceFormSelection)],
+    label=_(u"Treatment file: source search"),
+    url_name='treatmentfile_source_search',)
+
+treatmentfile_source_creation_wizard = TreatmentFileSourceWizard.as_view([
+    ('selec-treatmentfile_source_creation', SourceTreatmentFileFormSelection),
+    ('source-treatmentfile_source_creation', SourceForm),
+    ('authors-treatmentfile_source_creation', AuthorFormset),
+    ('final-treatmentfile_source_creation', FinalForm)],
+    url_name='treatmentfile_source_creation',)
+
+treatmentfile_source_modification_wizard = TreatmentFileSourceWizard.as_view([
+    ('selec-treatmentfile_source_modification',
+     TreatmentFileSourceFormSelection),
+    ('source-treatmentfile_source_modification', SourceForm),
+    ('authors-treatmentfile_source_modification', AuthorFormset),
+    ('final-treatmentfile_source_modification', FinalForm)],
+    label=_(u"Treatment file: source modification"),
+    url_name='treatmentfile_source_modification',)
+
+
+def treatmentfile_source_modify(request, pk):
+    treatmentfile_source_modification_wizard(request)
+    TreatmentFileSourceWizard.session_set_value(
+        request, 'selec-treatmentfile_source_modification', 'pk', pk,
+        reset=True)
+    return redirect(reverse(
+        'treatmentfile_source_modification',
+        kwargs={'step': 'source-treatmentfile_source_modification'}))
+
+treatmentfile_source_deletion_wizard = \
+    TreatmentFileSourceDeletionWizard.as_view([
+        ('selec-treatmentfile_source_deletion',
+         TreatmentFileSourceFormSelection),
+        ('final-treatmentfile_source_deletion', SourceDeletionForm)],
+        label=_(u"Treatment file: source deletion"),
+        url_name='treatmentfile_source_deletion',)