Encode/decode binary values for PyGreSQL driver

sqlobject · May 16, 2017 · 4c812a1 · 4c812a1
1 parent 7570514
commit 4c812a1
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git a/docs/News.rst b/docs/News.rst
@@ -13,6 +13,8 @@ SQLObject 3.4.0 (master)
 * Python 2.6 is no longer supported. The minimal supported version is
   Python 2.7.
 
+* Encode/decode binary values for PyGreSQL driver.
+
 SQLObject 3.3.0
 ===============
 

diff --git a/sqlobject/col.py b/sqlobject/col.py
@@ -1770,6 +1770,9 @@ def to_python(self, value, state):
         if isinstance(value, str):
             if not PY2 and dbName == "mysql":
                 value = value.encode('ascii', errors='surrogateescape')
+            if dbName == "postgres" and connection.driver == 'pygresql':
+                from pg import unescape_bytea
+                value = unescape_bytea(value)
             if dbName == "sqlite":
                 if not PY2:
                     value = bytes(value, 'ascii')

diff --git a/sqlobject/postgres/pgconnection.py b/sqlobject/postgres/pgconnection.py
@@ -85,6 +85,10 @@ def __init__(self, dsn=None, host=None, port=None, db=None,
             # Register a converter for psycopg Binary type.
             registerConverter(type(self.module.Binary('')),
                               PsycoBinaryConverter)
+        elif driver == 'pygresql':
+            from pg import escape_bytea
+            self.createBinary = \
+                lambda value, escape_bytea=escape_bytea: escape_bytea(value)
         elif type(self.module.Binary) in (
                 type, type(PostgresBinaryConverter)) and \
                 type(self.module.Binary(b'')) not in (bytes, unicode_type):