diff options
Diffstat (limited to 'archaeological_files/wizards.py')
| -rw-r--r-- | archaeological_files/wizards.py | 24 |
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 |