Changeset 9209 for main/waeup.kofa/branches/uli-zc-async/src/waeup/kofa/utils/batching.py

Timestamp:

20 Sep 2012, 08:49:37 (12 years ago)

Author:

uli

Message:

Merge changes from update branch (includes trunk changes until r9107).

Location:

main/waeup.kofa/branches/uli-zc-async

Files:

• . (modified) (1 prop)
• src/waeup/kofa/utils/batching.py (modified) (4 diffs)

main/waeup.kofa/branches/uli-zc-async/src/waeup/kofa/utils/batching.py

@@ -304 +304 @@
         finished_writer = csv.DictWriter(open(finished_path, 'wb'),
                                          finished_headers)
+        os.chmod(finished_path, 0664)
         finished_writer.writerow(dict([(x,x) for x in finished_headers]))
 
@@ -398 +399 @@
                 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.
@@ -506 +518 @@
 
     The resulting CSV file will be stored in a new temporary directory
-    (using :func:`tempfile.mkdtemp`).
+    (using :func:`tempfile.mkdtemp`). It will be named after the
+    exporter used with `.csv` filename extension.
 
     Returns the path to the created CSV file.
@@ -516 +529 @@
     exporter = getUtility(ICSVExporter, name=exporter_name)
     output_dir = tempfile.mkdtemp()
-    output_path = os.path.join(output_dir, 'export.csv')
+    filename = '%s.csv' % exporter_name
+    output_path = os.path.join(output_dir, filename)
     exporter.export_all(site, filepath=output_path)
     return output_path