utils: Wrap CSV in quotes

Signed-off-by: Stephen Finucane <[email protected]>
Closes: #46

Author: stephenfin

git_pw/utils.py

@@ -3,7 +3,9 @@
 """
 
 from __future__ import print_function
+
 import codecs
+import csv
 import os
 import subprocess
 import sys
@@ -12,6 +14,11 @@
 from tabulate import tabulate
 
 
+if sys.version_info < (3, 0):
+    from io import BytesIO  # noqa
+else:
+    from io import StringIO  # noqa
+
 if sys.version_info < (3, 0):
     _text = unicode  # noqa
 else:
@@ -63,12 +70,12 @@ def _tabulate(output, headers, fmt):
     elif fmt == 'simple':
         return tabulate(output, headers, tablefmt='simple')
     elif fmt == 'csv':
-        result = []
-        result.append(','.join(headers))
+        result = StringIO()
+        writer = csv.writer(result, quoting=csv.QUOTE_ALL)
+        writer.writerow(headers)
         for item in output:
-            result.append(
-                ','.join(_text(x if x is not None else '') for x in item))
-        return '\n'.join(result)
+            writer.writerow(_text(x if x is not None else '') for x in item)
+        return result.getvalue()
 
     print('pw.format must be one of: table, simple, csv')
     sys.exit(1)

releasenotes/notes/issue-46-50933643cd5c8db0.yaml

@@ -0,0 +1,4 @@
+---
+upgrade:
+  - |
+    CSV-formatted output is now quoted by default.