1 files changed, 120 insertions, 1 deletions
diff --git a/archaeological_finds/views.py b/archaeological_finds/views.py
index 1b08e9853..35c7d5976 100644
--- a/archaeological_finds/views.py
+++ b/archaeological_finds/views.py
@@ -28,16 +28,21 @@ from django.utils.translation import ugettext_lazy as _
 from django.views.generic import TemplateView
 from django.views.generic.edit import CreateView, FormView
 
+from ishtar_common.models import IshtarUser
+from archaeological_operations.models import AdministrativeAct
+
 from ishtar_common.forms import FinalForm
 from ishtar_common.forms_common import SourceForm, AuthorFormset, \
     SourceDeletionForm
-from ishtar_common.models import IshtarUser
+from archaeological_operations.forms import FinalAdministrativeActDeleteForm
 from archaeological_context_records.forms \
     import RecordFormSelection as RecordFormSelectionTable
 
 from ishtar_common.views import get_item, show_item, revert_item, \
     get_autocomplete_generic, IshtarMixin, LoginRequiredMixin
+
 from ishtar_common.wizards import SearchWizard
+from archaeological_operations.wizards import AdministrativeActDeletionWizard
 
 from wizards import *
 from forms import *
@@ -55,10 +60,20 @@ get_find_for_treatment = get_item(
 show_treatment = show_item(models.Treatment, 'treatment')
 get_treatment = get_item(models.Treatment, 'get_treatement', 'treatment')
 
+get_administrativeacttreatment = get_item(
+    AdministrativeAct, 'get_administrativeacttreatment',
+    'administrativeacttreatment',
+    base_request={"treatment__pk__isnull": False})
+
 show_treatmentfile = show_item(models.TreatmentFile, 'treatmentfile')
 get_treatmentfile = get_item(models.TreatmentFile, 'get_treatementfile',
                              'treatmentfile')
 
+get_administrativeacttreatmentfile = get_item(
+    AdministrativeAct, 'get_administrativeacttreatmentfile',
+    'administrativeacttreatmentfile',
+    base_request={"treatment_file__pk__isnull": False})
+
 
 def autocomplete_treatmentfile(request):
     if not request.user.has_perm('ishtar_common.view_treatment',
@@ -375,6 +390,59 @@ treatment_deletion_wizard = TreatmentDeletionWizard.as_view([
     label=_(u"Treatment deletion"),
     url_name='treatment_deletion',)
 
+treatment_administrativeact_search_wizard = \
+    SearchWizard.as_view([
+        ('selec-treatment_admacttreatment_search',
+         AdministrativeActTreatmentFormSelection)],
+        label=_(u"Treatment: search administrative act"),
+        url_name='treatment_admacttreatment_search',)
+
+treatment_administrativeact_wizard = \
+    TreatmentAdministrativeActWizard.as_view([
+        ('selec-treatment_admacttreatment', TreatmentFormSelection),
+        ('administrativeact-treatment_admacttreatment',
+         AdministrativeActTreatmentForm),
+        ('final-treatment_admacttreatment', FinalForm)],
+        label=_(u"Treatment: new administrative act"),
+        url_name='treatment_admacttreatment',)
+
+treatment_administrativeact_modification_wizard = \
+    TreatmentEditAdministrativeActWizard.as_view([
+        ('selec-treatment_admacttreatment_modification',
+         AdministrativeActTreatmentFormSelection),
+        ('administrativeact-treatment_admacttreatment_modification',
+         AdministrativeActTreatmentModifForm),
+        ('final-treatment_admacttreatment_modification', FinalForm)],
+        label=_(u"Treatment: administrative act modification"),
+        url_name='treatment_admacttreatment_modification',)
+
+treatment_admacttreatment_deletion_wizard = \
+    AdministrativeActDeletionWizard.as_view([
+        ('selec-treatment_admacttreatment_deletion',
+         AdministrativeActTreatmentFormSelection),
+        ('final-treatment_admacttreatment_deletion',
+         FinalAdministrativeActDeleteForm)],
+        label=_(u"Treatment: administrative act deletion"),
+        url_name='treatment_admacttreatment_deletion',)
+
+
+def treatment_administrativeacttreatment_modify(request, pk):
+    treatment_administrativeact_modification_wizard(request)
+    TreatmentEditAdministrativeActWizard.session_set_value(
+        request,
+        'selec-treatment_admacttreatment_modification',
+        'pk', pk, reset=True)
+    return redirect(
+        reverse(
+            'treatment_admacttreatment_modification',
+            kwargs={
+                'step':
+                    'administrativeact-treatment_admacttreatment_modification'
+            }))
+
+
+# treatment file
+
 treatmentfile_search_wizard = SearchWizard.as_view([
     ('general-treatmentfile_search', TreatmentFileFormSelection)],
     label=_(u"Treatment file search"),
@@ -404,6 +472,57 @@ treatmentfile_deletion_wizard = TreatmentFileDeletionWizard.as_view([
     label=_(u"Treatment file deletion"),
     url_name='treatmentfile_deletion',)
 
+treatmentfile_admacttreatmentfile_search_wizard = \
+    SearchWizard.as_view([
+        ('selec-treatmentfle_admacttreatmentfle_search',
+        AdministrativeActTreatmentFileFormSelection)],
+        label=_(u"Treatment file: search administrative act"),
+        url_name='treatmentfle_admacttreatmentfle_search',)
+
+
+treatmentfile_admacttreatmentfile_wizard = \
+    TreatmentFileAdministrativeActWizard.as_view([
+        ('selec-treatmentfle_admacttreatmentfle', TreatmentFileFormSelection),
+        ('admact-treatmentfle_admacttreatmentfle',
+         AdministrativeActTreatmentFileForm),
+        ('final-treatmentfle_admacttreatmentfle', FinalForm)],
+        label=_(u"Treatment file: new administrative act"),
+        url_name='treatmentfle_admacttreatmentfle',)
+
+treatmentfile_admacttreatmentfile_modification_wizard = \
+    TreatmentFileEditAdministrativeActWizard.as_view([
+        ('selec-treatmentfle_admacttreatmentfle_modification',
+         AdministrativeActTreatmentFileFormSelection),
+        ('admact-treatmentfle_admacttreatmentfle_modification',
+         AdministrativeActTreatmentFileModifForm),
+        ('final-treatmentfle_admacttreatmentfle_modification', FinalForm)],
+        label=_(u"Treatment file: administrative act modification"),
+        url_name='treatmentfle_admacttreatmentfle_modification',)
+
+treatmentfile_admacttreatmentfile_deletion_wizard = \
+    AdministrativeActDeletionWizard.as_view([
+        ('selec-treatmentfle_admacttreatmentfle_deletion',
+         AdministrativeActTreatmentFileFormSelection),
+        ('final-treatmentfle_admacttreatmentfle_deletion',
+         FinalAdministrativeActDeleteForm)],
+        label=_(u"Treatment file: administrative act deletion"),
+        url_name='treatmentfle_admacttreatmentfle_deletion',)
+
+
+def treatmentfile_administrativeacttreatmentfile_modify(request, pk):
+    treatmentfile_admacttreatmentfile_modification_wizard(request)
+    TreatmentFileEditAdministrativeActWizard.session_set_value(
+        request,
+        'selec-treatmentfle_admacttreatmentfle_modification',
+        'pk', pk, reset=True)
+    return redirect(
+        reverse(
+            'treatmentfle_admacttreatmentfle_modification',
+            kwargs={
+                'step':
+                    'admact-treatmentfle_admacttreatmentfle_modification'
+            }))
+
 """
 treatment_source_creation_wizard = TreatmentSourceWizard.as_view([
     ('selec-treatment_source_creation', SourceTreatmentFormSelection),