Changeset 12161 for main/waeup.ikoba/trunk/src/waeup/ikoba/customers/documents.py

Timestamp:

7 Dec 2014, 10:07:29 (10 years ago)

Author:

Henrik Bettermann

Message:

Define connected_files and getMD5 for all kind of customer documents.

File:

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

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

@@ -21 +21 @@
 import os
 import grok
+from hashlib import md5
 from zope.component import queryUtility, getUtility
 from zope.component.interfaces import IFactory
@@ -66 +67 @@
     grok.baseclass()
 
+    filenames = ()
+
     @property
     def customer(self):
@@ -105 +108 @@
             return
 
+    @property
+    def connected_files(self):
+        store = getUtility(IExtFileStore)
+        files = []
+        try:
+            for filename in self.filenames:
+                attrname = filename.replace('.','_')
+                file = store.getFileByContext(self, attr=filename)
+                files.append((attrname, file))
+        except AttributeError:
+            # In unit tests we don't have a customer to
+            # determine the file path.
+            return
+        return files
+
+    def setMD5(self):
+        """Set md5 checksum attribute for all files connected to this document.
+        """
+        for file in self.connected_files:
+            attrname = '%s_md5' % file[0]
+            checksum = md5(file[1].read()).hexdigest()
+            setattr(self, attrname, checksum)
+        return
+
 
 class CustomerSampleDocument(CustomerDocumentBase):
     """This is a sample customer document.
     """
+
+    filenames = ('sample',)
 
 CustomerSampleDocument = attrs_to_fields(CustomerSampleDocument)