Implement feature that allows to export tabular data suited to a… (#437)

2 files changed, 27 insertions, 1 deletions

diff --git a/src/tablib/formats/__init__.py b/src/tablib/formats/__init__.py
index 5c46008..543f682 100644
--- a/src/tablib/formats/__init__.py
+++ b/src/tablib/formats/__init__.py
@@ -12,6 +12,10 @@ from ._json import JSONFormat
 from ._tsv import TSVFormat
 
 uninstalled_format_messages = {
+    'cli': (
+        "The 'cli' format is not available. You may want to install the tabulate "
+        "package (or `pip install tablib[cli]`)."
+    ),
     'df': (
         "The 'df' format is not available. You may want to install the pandas "
         "package (or `pip install tablib[pandas]`)."
@@ -88,7 +92,7 @@ class Registry:
     def register(self, key, format_or_path):
         from tablib.core import Databook, Dataset
 
-        # Create Databook.<format> read or read/write properties 
+        # Create Databook.<format> read or read/write properties
         setattr(Databook, key, ImportExportBookDescriptor(key, format_or_path))
 
         # Create Dataset.<format> read or read/write properties,
@@ -124,6 +128,8 @@ class Registry:
         if find_spec('pandas'):
             self.register('df', 'tablib.formats._df.DataFrameFormat')
         self.register('rst', 'tablib.formats._rst.ReSTFormat')
+        if find_spec('tabulate'):
+            self.register('cli', 'tablib.formats._cli.CLIFormat')
 
     def formats(self):
         for key, frm in self._formats.items():
diff --git a/src/tablib/formats/_cli.py b/src/tablib/formats/_cli.py
new file mode 100644
index 0000000..0297ae8
--- /dev/null
+++ b/src/tablib/formats/_cli.py
@@ -0,0 +1,20 @@
+"""Tablib - Command-line Interface table export support.
+
+   Generates a representation for CLI from the dataset.
+   Wrapper for tabulate library.
+"""
+from tabulate import tabulate as Tabulate
+
+
+class CLIFormat:
+    """ Class responsible to export to CLI Format """
+    title = 'cli'
+    DEFAULT_FMT = 'plain'
+
+    @classmethod
+    def export_set(cls, dataset, **kwargs):
+        """Returns CLI representation of a Dataset."""
+        if dataset.headers:
+            kwargs.setdefault('headers', dataset.headers)
+        kwargs.setdefault('tablefmt', cls.DEFAULT_FMT)
+        return Tabulate(dataset, **kwargs)