Changeset 12442 for main/waeup.ikoba/trunk

Timestamp:

11 Jan 2015, 16:53:38 (10 years ago)

Author:

Henrik Bettermann

Message:

Add handler for document removal. Files must be removed too.

Location:

main/waeup.ikoba/trunk/src/waeup/ikoba

Files:

• customers/documents.py (modified) (1 diff)
• documents/document.py (modified) (3 diffs)
• documents/tests/test_browser.py (modified) (2 diffs)

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

@@ -69 +69 @@
     local_roles = []
 
-    # Ikoba can store any number of files per Document object.
-    # However, we highly recommend to associate and store
-    # only one file per Document object. Thus the following 
-    # tuple should contain only a single filename string.
-    filenames = ()
-
     def __init__(self):
         super(CustomerDocumentBase, self).__init__()

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

@@ -40 +40 @@
     IPDFDocument, IHTMLDocument, IRESTDocument)
 
+
+@grok.subscribe(IDocument, grok.IObjectRemovedEvent)
+def handle_document_removed(document, event):
+    store = getUtility(IExtFileStore)
+    for filename in document.filenames:
+        store.deleteFileByContext(document, attr=filename)
+    return
+
 class Document(grok.Container):
     """This is a document.
@@ -48 +56 @@
 
     form_fields_interface = None
+
+    # Kofa can store any number of files per Document object.
+    # However, we highly recommend to associate and store
+    # only one file per Document object. Thus the following
+    # tuple should contain only a single filename string.
+    filenames = ()
 
     local_roles = [
@@ -132 +146 @@
     form_fields_interface = IPDFDocument
 
+    filenames = ('scan.pdf',)
+
 PDFDocument = attrs_to_fields(PDFDocument)
 

main/waeup.ikoba/trunk/src/waeup/ikoba/documents/tests/test_browser.py

@@ -150 +150 @@
             'href="http://localhost/app/documents/DOC1/sample.pdf">PDF File</a>'
             in self.browser.contents)
+        # The file can be found in the file system
+        file = getUtility(IExtFileStore).getFileByContext(
+            document, attr='sample.pdf')
+        file_content = file.read()
+        pdf.seek(0)
+        pdf_content = pdf.read()
+        self.assertEqual(file_content, pdf_content)
         # Browsing the link shows a real pdf only if the document
         # has been published
@@ -176 +183 @@
         self.browser.getControl("Remove selected", index=0).click()
         self.assertTrue('Successfully removed' in self.browser.contents)
+
+        # File has been removed too
+        file = getUtility(IExtFileStore).getFileByContext(
+            document, attr='sample.pdf')
+        self.assertTrue(file is None)
 
         # All actions are being logged