Force psycopg to decode strings received from database to unicode.

Unicode data caused queries to output nothing and exit with an UnicodeDecodeError.
dbcli · Jan 6, 2015 · 4aaa117 · 4aaa117
1 parent f3f8c92
commit 4aaa117
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/pgcli/pgexecute.py b/pgcli/pgexecute.py
@@ -1,9 +1,15 @@
 import logging
 import psycopg2
+import psycopg2.extensions
 from .packages import pgspecial
 
 _logger = logging.getLogger(__name__)
 
+# Cast all database input to unicode automatically.
+# See http://initd.org/psycopg/docs/usage.html#unicode-handling for more info.
+psycopg2.extensions.register_type(psycopg2.extensions.UNICODE)
+psycopg2.extensions.register_type(psycopg2.extensions.UNICODEARRAY)
+
 def _parse_dsn(dsn, default_user, default_password, default_host,
         default_port):
     """