1 files changed, 38 insertions, 0 deletions
diff --git a/litecli/encodingutils.py b/litecli/encodingutils.py
new file mode 100644
index 0000000..6caf14d
--- /dev/null
+++ b/litecli/encodingutils.py
@@ -0,0 +1,38 @@
+# -*- coding: utf-8 -*-
+from __future__ import unicode_literals
+
+from litecli.compat import PY2
+
+
+if PY2:
+    binary_type = str
+    string_types = basestring
+    text_type = unicode
+else:
+    binary_type = bytes
+    string_types = str
+    text_type = str
+
+
+def unicode2utf8(arg):
+    """Convert strings to UTF8-encoded bytes.
+
+    Only in Python 2. In Python 3 the args are expected as unicode.
+
+    """
+
+    if PY2 and isinstance(arg, text_type):
+        return arg.encode("utf-8")
+    return arg
+
+
+def utf8tounicode(arg):
+    """Convert UTF8-encoded bytes to strings.
+
+    Only in Python 2. In Python 3 the errors are returned as strings.
+
+    """
+
+    if PY2 and isinstance(arg, binary_type):
+        return arg.decode("utf-8")
+    return arg