Add reminders for administrativ acts (refs #2060)

1 files changed, 23 insertions, 1 deletions

diff --git a/archaeological_files/wizards.py b/archaeological_files/wizards.py
index 7fb654b0b..4ad947641 100644
--- a/archaeological_files/wizards.py
+++ b/archaeological_files/wizards.py
@@ -1,6 +1,6 @@
 #!/usr/bin/env python
 # -*- coding: utf-8 -*-
-# Copyright (C) 2012-2013  Étienne Loks  <etienne.loks_AT_peacefrogsDOTnet>
+# Copyright (C) 2012-2014  Étienne Loks  <etienne.loks_AT_peacefrogsDOTnet>
 
 # This program is free software: you can redistribute it and/or modify
 # it under the terms of the GNU Affero General Public License as
@@ -155,6 +155,28 @@ class FileDeletionWizard(FileClosingWizard):
 class FileAdministrativeActWizard(OperationAdministrativeActWizard):
     model = models.File
 
+    def get_reminder(self):
+        form_key = 'selec-' + self.url_name
+        if self.url_name.endswith('_administrativeactfile'):
+            # modification and deletion are suffixed with '_modification'
+            # and '_deletion' so it is creation
+            file_id = self.session_get_value(form_key, "pk")
+            try:
+                return ((_(u"Archaelogical file"),
+                      unicode(models.File.objects.get(pk=file_id))),)
+            except models.File.DoesNotExist:
+                return
+        else:
+            admin_id = self.session_get_value(form_key, "pk")
+            try:
+                admin = AdministrativeAct.objects.get(pk=admin_id)
+                if not admin.associated_file:
+                    return
+                return ((_(u"Archaelogical file"),
+                         unicode(admin.associated_file)),)
+            except AdministrativeAct.DoesNotExist:
+                return
+
 class FileEditAdministrativeActWizard(FileAdministrativeActWizard):
     model = AdministrativeAct
     edit = True