Cockroachdb sqlalchemy dialect doesn't fully support JSONB

I’m testing recently added JSONB.

Setup:

python3.6
psycopg2==2.7.4
SQLAlchemy==1.2.5
cockroachdb v2.0.1

I was able to save a sample data into the database. This is how it looks in a dump file:

{
    "id" : "2a4b4578-1a5d-455b-9317-c90c5f3afe15",
    "enabled" : true,
    "language" : "en-us",
    "ui_metadata" : "{\"origin\": \"111\", \"ui\": null, \"version\": null}",
}

I’m now loading data with sqlalchemy session.query.all() Error:

    return json_deserializer(value)
  File "/usr/local/lib/python3.6/json/__init__.py", line 348, in loads
    'not {!r}'.format(s.__class__.__name__))
TypeError: the JSON object must be str, bytes or bytearray, not 'dict'

https://bitbucket.org/zzzeek/sqlalchemy/src/7541d9b258e4d0c115d6160592ac7e17f8fd50cc/lib/sqlalchemy/sql/sqltypes.py?at=master&fileviewer=file-view-default#sqltypes.py-2145:2155 https://github.com/cockroachdb/cockroachdb-python/blob/master/cockroachdb/sqlalchemy/dialect.py

This line fails:

json_deserializer = dialect._json_deserializer or json.loads

Because there is no _json_deserializer and it tries to call json.loads on a python dict.

[bdarnell]

I’ve just uploaded a new version to pypi.

[desprit]

@bdarnell

I’m not using JSON 😃

from sqlalchemy.dialects.postgresql import JSONB

class myClass(Base):
    my_field = Column(JSONB, nullable=False, default=lambda: {})

I’m using dbeaver as a gui tool and it says that data type of that column is jsonb. So when my code throws TypeError: the JSON object must be str, bytes or bytearray, not 'dict', it somehow detects JSON instead of JSONB.