summaryrefslogtreecommitdiff
path: root/clifftablib/formatters.py
diff options
context:
space:
mode:
Diffstat (limited to 'clifftablib/formatters.py')
-rw-r--r--clifftablib/formatters.py49
1 files changed, 49 insertions, 0 deletions
diff --git a/clifftablib/formatters.py b/clifftablib/formatters.py
new file mode 100644
index 0000000..d7e5ca0
--- /dev/null
+++ b/clifftablib/formatters.py
@@ -0,0 +1,49 @@
+"""Output formatters using tablib.
+"""
+
+from cliff.formatters.base import ListFormatter, SingleFormatter
+
+import tablib
+
+
+class TablibFormatterBase(ListFormatter, SingleFormatter):
+
+ def add_argument_group(self, parser):
+ return
+
+ def emit_list(self, column_names, data, stdout, parsed_args):
+ dataset = tablib.Dataset(headers=column_names)
+ for row in data:
+ dataset.append(row)
+ stdout.write(self._format_dataset(dataset))
+ stdout.write('\n')
+ return
+
+ def emit_one(self, column_names, data, stdout, parsed_args):
+ dataset = tablib.Dataset(headers=('Field', 'Value'))
+ for name, value in zip(column_names, data):
+ dataset.append((name, value))
+ stdout.write(self._format_dataset(dataset))
+ stdout.write('\n')
+ return
+
+
+class YamlFormatter(TablibFormatterBase):
+ """YAML output"""
+
+ def _format_dataset(self, dataset):
+ return dataset.yaml
+
+
+class HtmlFormatter(TablibFormatterBase):
+ """HTML output"""
+
+ def _format_dataset(self, dataset):
+ return dataset.html
+
+
+class JsonFormatter(TablibFormatterBase):
+ """JSON output"""
+
+ def _format_dataset(self, dataset):
+ return dataset.json