Changeset 9322 for main/waeup.kofa/trunk/src/waeup/kofa/browser/pages.py

Timestamp:

9 Oct 2012, 10:29:49 (12 years ago)

Author:

Henrik Bettermann

Message:

Send detailed information about file uploaded.

File:

• main/waeup.kofa/trunk/src/waeup/kofa/browser/pages.py (modified) (7 diffs)

main/waeup.kofa/trunk/src/waeup/kofa/browser/pages.py

@@ -67 +67 @@
 from waeup.kofa.utils.helpers import get_user_account
 from waeup.kofa.mandates.mandate import PasswordMandate
+from waeup.kofa.datacenter import DataCenterFile
 
 grok.context(IKofaObject)
@@ -154 +155 @@
         view.redirect(view.url(view.context, redirect))
     return
+
+def getPreviewTable(view, n):
+    """Get transposed table with n sample record.
+
+    The first column contains the headers.
+    """
+    if not view.reader:
+        return
+    header = view.getPreviewHeader()
+    num = 0
+    data = []
+    for line in view.reader:
+        if num > n-1:
+            break
+        num += 1
+        data.append(line)
+    result = []
+    for name in header:
+        result_line = []
+        result_line.append(name)
+        for d in data:
+            result_line.append(d[name])
+        result.append(result_line)
+    return result
 
 # Save function used for save methods in pages
@@ -838 +863 @@
     cancel_button =_(u'Cancel')
 
-    def _notifyImportManagers(self, normalized_filename):
+    def getPreviewHeader(self):
+        """Get the header fields of uploaded CSV file.
+        """
+        reader = csv.reader(open(self.fullpath, 'rb'))
+        return reader.next()
+
+    def getImporters(self):
+        importers = getAllUtilitiesRegisteredFor(IBatchProcessor)
+        importers = sorted(
+            [dict(title=x.name, name=x.util_name) for x in importers])
+        return importers
+
+    def _notifyImportManagers(self, normalized_filename, importer, import_mode):
         """Send email to Import Managers
         """
+        # Get information about file
+        self.fullpath = os.path.join(self.context.storage, normalized_filename)
+        uploadfile = DataCenterFile(self.fullpath)
+        self.reader = csv.DictReader(open(self.fullpath, 'rb'))
+        table = getPreviewTable(self, 3)
+        mail_table = ''
+        for line in table:
+            header = line[0]
+            data = str(line[1:]).strip('[').strip(']')
+            mail_table += '%s: %s\n' % (line[0], data)
+        # Collect all recipient addresses
         kofa_utils = getUtility(IKofaUtils)
         import_managers = get_users_with_role(
@@ -862 +910 @@
                       'waeup.kofa',
                       target_language=kofa_utils.PORTAL_LANGUAGE)
-            text = _('File ${a} uploaded.', mapping = {'a':normalized_filename})
+            text = _("""File: ${a}
+Processor: ${b}
+Import Mode: ${c}
+Datasets: ${d}
+
+${e}
+
+""", mapping = {'a':normalized_filename,
+                'b':importer,
+                'c':import_mode,
+                'd':uploadfile.lines - 1,
+                'e':mail_table})
             success = kofa_utils.sendContactForm(
                     fullname,email,
@@ -875 +934 @@
             return
 
-    def update(self, uploadfile=None, CANCEL=None, SUBMIT=None):
+    def update(self, uploadfile=None, import_mode=None,
+               importer=None, CANCEL=None, SUBMIT=None):
         if CANCEL is not None:
             self.redirect(self.url(self.context))
@@ -903 +963 @@
             ob_class = self.__implemented__.__name__.replace('waeup.kofa.','')
             logger.info('%s - uploaded: %s' % (ob_class, target))
-            self._notifyImportManagers(normalized_filename)
+            self._notifyImportManagers(
+                normalized_filename, importer, import_mode)
 
         except IOError:
@@ -1050 +1111 @@
 
     def getPreviewTable(self):
-        """Get transposed table with 3 sample records.
-
-        The first column contains the headers.
-        """
-        if not self.reader:
-            return
-        header = self.getPreviewHeader()
-        num = 0
-        data = []
-        for line in self.reader:
-            if num > 2:
-                break
-            num += 1
-            data.append(line)
-        result = []
-        for name in header:
-            result_line = []
-            result_line.append(name)
-            for d in data:
-                result_line.append(d[name])
-            result.append(result_line)
-        return result
+        return getPreviewTable(self, 3)
 
     def getImporters(self):