add a csv formatter for list apps

2 files changed, 36 insertions, 1 deletions

diff --git a/cliff/formatters/commaseparated.py b/cliff/formatters/commaseparated.py
new file mode 100644
index 0000000..f061b59
--- /dev/null
+++ b/cliff/formatters/commaseparated.py
@@ -0,0 +1,33 @@
+"""Output formatters using csv format.
+"""
+
+import csv
+
+from .base import ListFormatter
+
+
+class CSVLister(ListFormatter):
+
+    QUOTE_MODES = {
+        'all': csv.QUOTE_ALL,
+        'minimal': csv.QUOTE_MINIMAL,
+        'nonnumeric': csv.QUOTE_NONNUMERIC,
+        'none': csv.QUOTE_NONE,
+        }
+
+    def add_argument_group(self, parser):
+        group = parser.add_argument_group('CSV Formatter')
+        group.add_argument(
+            '--quote',
+            choices=sorted(self.QUOTE_MODES.keys()),
+            dest='quote_mode',
+            default='nonnumeric',
+            help='when to include quotes, defaults to nonnumeric',
+            )
+
+    def emit_list(self, column_names, data, stdout, parsed_args):
+        writer = csv.writer(stdout, quoting=self.QUOTE_MODES[parsed_args.quote_mode])
+        writer.writerow(column_names)
+        for row in data:
+            writer.writerow(row)
+        return
diff --git a/cliff/lister.py b/cliff/lister.py
index 01f25ef..15ac83a 100644
--- a/cliff/lister.py
+++ b/cliff/lister.py
@@ -27,7 +27,9 @@ class Lister(Command):
             description='List output formatter options',
             )
         formatter_choices = sorted(self.formatters.keys())
-        formatter_default = formatter_choices[0]
+        formatter_default = 'table'
+        if formatter_default not in formatter_choices:
+            formatter_default = formatter_choices[0]
         formatter_group.add_argument(
             '-f', '--format',
             dest='formatter',