#1686 added back support for --raw output format

Author: allmightyspiff

SoftLayer/CLI/formatting.py

@@ -35,7 +35,7 @@ def format_output(data, fmt='table'):  # pylint: disable=R0911,R0912
 
     # responds to .prettytable()
     if hasattr(data, 'prettytable') and fmt in ('table', 'raw'):
-        return format_prettytable(data)
+        return format_prettytable(data, fmt)
 
     # responds to .to_python()
     if hasattr(data, 'to_python'):
@@ -68,12 +68,12 @@ def format_output(data, fmt='table'):  # pylint: disable=R0911,R0912
     return str(data)
 
 
-def format_prettytable(table):
+def format_prettytable(table, fmt='table'):
     """Converts SoftLayer.CLI.formatting.Table instance to a prettytable."""
     for i, row in enumerate(table.rows):
         for j, item in enumerate(row):
             table.rows[i][j] = format_output(item)
-    ptable = table.prettytable()
+    ptable = table.prettytable(fmt)
     return ptable
 
 
@@ -258,9 +258,12 @@ def to_python(self):
             items.append(dict(zip(self.columns, formatted_row)))
         return items
 
-    def prettytable(self):
+    def prettytable(self, fmt='table'):
         """Returns a RICH table instance."""
-        table = rTable(title=self.title, box=box.SQUARE, header_style="bright_cyan")
+        box_style = box.SQUARE
+        if fmt == 'raw':
+            box_style = None
+        table = rTable(title=self.title, box=box_style, header_style="bright_cyan")
         if self.sortby:
             try:
                 # https://docs.python.org/3/howto/sorting.html#key-functions

tests/CLI/helper_tests.py

@@ -396,6 +396,12 @@ def test_format_output_table_invalid_sort(self):
             formatting.format_output, t, 'table',
         )
 
+    def test_format_raw_table(self):
+        test_table = formatting.Table(['col1', 'col2'])
+        test_table.add_row(['row1', 'row2'])
+        pretty_table = formatting.format_output(test_table, 'raw')
+        self.assertIsNone(pretty_table.box)
+
 
 class TestTemplateArgs(testing.TestCase):