Changeset 9032

Timestamp:

21 Jul 2012, 08:16:03 (12 years ago)

Author:

Henrik Bettermann

Message:

Implement SkeletonDownloadView? for downloading csv skeleton files.

Compose export filenames of portal acronym and the exporter name.

Location:

main/waeup.kofa/trunk/src/waeup/kofa

Files:

• browser/pages.py (modified) (2 diffs)
• browser/templates/datacenteruploadpage.pt (modified) (2 diffs)
• browser/tests/test_browser.py (modified) (2 diffs)
• utils/batching.py (modified) (1 diff)

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

@@ -914 +914 @@
         return open(fullpath, 'rb').read()
 
+class SkeletonDownloadView(UtilityView, grok.View):
+    grok.context(IDataCenter)
+    grok.name('skeleton')
+    grok.require('waeup.manageDataCenter')
+
+    def update(self, processorname=None):
+        self.processorname = self.request.form['name']
+        self.filename = ('%s_000.csv' %
+            self.processorname.replace('processor','import'))
+        return
+
+    def render(self):
+        #ob_class = self.__implemented__.__name__.replace('waeup.kofa.','')
+        #self.context.logger.info(
+        #    '%s - skeleton downloaded: %s' % (ob_class, self.filename))
+        self.response.setHeader(
+            'Content-Type', 'text/csv; charset=UTF-8')
+        self.response.setHeader(
+            'Content-Disposition:', 'attachment; filename="%s' % self.filename)
+        processor = getUtility(IBatchProcessor, name=self.processorname)
+        csv_data = processor.get_csv_skeleton()
+        return csv_data
+
 class DatacenterImportStep1(KofaPage):
     """Manual import step 1: choose file
@@ -1375 +1398 @@
         self.response.setHeader(
             'Content-Type', 'text/csv; charset=UTF-8')
+        acronym = grok.getSite()['configuration'].acronym.replace(' ','')
+        filename = "%s%s.csv" % (acronym, exporter.title.title().replace(' ',''))
+        self.response.setHeader(
+            'Content-Disposition:', 'attachment; filename="%s' % filename)
         return csv_data
 

main/waeup.kofa/trunk/src/waeup/kofa/browser/templates/datacenteruploadpage.pt

@@ -19 +19 @@
       <th i18n:translate="">Processor</th>
       <th i18n:translate="">Available Fields</th>
-      <th i18n:translate="">CSV Skeleton File (not yet)</th>
+      <th i18n:translate="">CSV Skeleton File</th>
     </tr>
   </thead>
@@ -31 +31 @@
     </td>
     <td>
-      <span tal:replace="importer/name"></span>.csv
+      <a tal:attributes="href python: 'skeleton?name=' + importer['name']"
+         tal:content="python: importer['name'].replace('processor', 'import') + '_000.csv'">
+         FILENAME</a>
     </td>
   </tr>

main/waeup.kofa/trunk/src/waeup/kofa/browser/tests/test_browser.py

@@ -160 +160 @@
         self.assertEqual(self.browser.headers['Content-Type'],
                          'text/csv; charset=UTF-8')
+        self.assertTrue ('WAeUP.KofaFaculties.csv' in
+            self.browser.headers['content-disposition'])
         self.assertEqual(self.browser.contents,
             'code,title,title_prefix,users_with_local_roles\r\n'
@@ -169 +171 @@
                         'exported: faculties' in logcontent)
         return
+
+    def test_skeleton_download(self):
+        self.browser.addHeader('Authorization', 'Basic mgr:mgrpw')
+        self.browser.open(self.datacenter_path)
+        self.browser.getLink("Upload data").click()
+        self.assertTrue(
+            '<a href="skeleton?name=facultyprocessor">facultyimport_000.csv</a>'
+            in self.browser.contents)
+        self.browser.getLink('facultyimport_000.csv').click()
+        self.assertEqual(self.browser.headers['Status'], '200 Ok')
+        self.assertEqual(self.browser.headers['Content-Type'],
+                         'text/csv; charset=UTF-8')
+        self.assertEqual(self.browser.contents,
+            'code,local_roles,title,title_prefix\r\n')
+        return

main/waeup.kofa/trunk/src/waeup/kofa/utils/batching.py

@@ -394 +394 @@
                 os.path.abspath(finished_path), failed_path)
 
+    def get_csv_skeleton(self):
+        """Export CSV file only with a header of available fields.
+
+        A raw string with CSV data should be returned.
+        """
+        outfile = StringIO()
+        writer = csv.DictWriter(outfile, self.available_fields)
+        writer.writerow(dict(zip(self.available_fields, self.available_fields))) # header
+        outfile.seek(0)
+        return outfile.read()
+
 class ExporterBase(object):
     """A base for exporters.