Changeset 12240 for main/waeup.ikoba/trunk/src/waeup/ikoba/documents/fileviewlets.py

Timestamp:

15 Dec 2014, 07:44:01 (10 years ago)

Author:

Henrik Bettermann

Message:

Protect pdf download if document has not been published.

Add workflow transition 'retract'.

File:

• main/waeup.ikoba/trunk/src/waeup/ikoba/documents/fileviewlets.py (modified) (3 diffs)

main/waeup.ikoba/trunk/src/waeup/ikoba/documents/fileviewlets.py

@@ -19 +19 @@
 import grok
 from waeup.ikoba.interfaces import MessageFactory as _
-
+from waeup.ikoba.interfaces import PUBLISHED
+from waeup.ikoba.browser.layout import UtilityView
 from waeup.ikoba.browser.fileviewlets import (
     FileDisplay, FileUpload, Image)
@@ -55 +56 @@
 
 
-class PDFScanImage(Image):
+class PDFScanImage(Image, UtilityView):
     """Scan document.
     """
@@ -64 +65 @@
     download_name = u'sample.pdf'
 
+    def render(self):
+        if self.context.state != PUBLISHED:
+            self.flash(_('The document requested has not yet been published.'),
+                type="warning")
+            self.redirect(self.url(self.context))
+            return
+        return super(PDFScanImage, self).render()
+