U x^'@s6ddlmZddlmZddlmZddlmZddlmZddl m Z d Z e j e j Ze jd ed d d Ze jd ed d d Ze jded d d Ze jded d d Ze jded d d Ze jded d d Ze jded d d ZGdddejjZeeejj<GdddejZeeej<eed<GdddeZeed<dS))absolute_import)colspecs) ischema_names)types)util) operators)JSONJSONBz->>T) precedenceZnatural_self_precedentZeager_groupingz#>>?z?&z?|z@>z<@c@seZdZddZddZdS) JSONPathTypecs||fdd}|S)Ncs>t|tjjstdd|D}dd|}r:|}|S)NcSsg|]}t|qSr text_type.0elemrrK/tmp/pip-install-dq5v43_d/SQLAlchemy/sqlalchemy/dialects/postgresql/json.py Msz@JSONPathType.bind_processor..process..{%s},  isinstancerZcollections_abcSequenceAssertionErrorjoinvaluetokensZ super_procrrprocessKs z,JSONPathType.bind_processor..process)Zstring_bind_processorselfdialectr"rr!rbind_processorHs  zJSONPathType.bind_processorcs||fdd}|S)Ncs>t|tjjstdd|D}dd|}r:|}|S)NcSsg|]}t|qSrrrrrrrZszCJSONPathType.literal_processor..process..rrrrr!rrr"Xs z/JSONPathType.literal_processor..process)Zstring_literal_processorr#rr!rliteral_processorUs  zJSONPathType.literal_processorN)__name__ __module__ __qualname__r&r'rrrrrGs rcsBeZdZdZeZdfdd ZGdddejj Z e Z Z S) r a Represent the PostgreSQL JSON type. This type is a specialization of the Core-level :class:`.types.JSON` type. Be sure to read the documentation for :class:`.types.JSON` for important tips regarding treatment of NULL values and ORM use. .. versionchanged:: 1.1 :class:`.postgresql.JSON` is now a PostgreSQL- specific specialization of the new :class:`.types.JSON` type. The operators provided by the PostgreSQL version of :class:`.JSON` include: * Index operations (the ``->`` operator):: data_table.c.data['some key'] data_table.c.data[5] * Index operations returning text (the ``->>`` operator):: data_table.c.data['some key'].astext == 'some value' Note that equivalent functionality is available via the :attr:`.JSON.Comparator.as_string` accessor. * Index operations with CAST (equivalent to ``CAST(col ->> ['some key'] AS )``):: data_table.c.data['some key'].astext.cast(Integer) == 5 Note that equivalent functionality is available via the :attr:`.JSON.Comparator.as_integer` and similar accessors. * Path index operations (the ``#>`` operator):: data_table.c.data[('key_1', 'key_2', 5, ..., 'key_n')] * Path index operations returning text (the ``#>>`` operator):: data_table.c.data[('key_1', 'key_2', 5, ..., 'key_n')].astext == 'some value' .. versionchanged:: 1.1 The :meth:`.ColumnElement.cast` operator on JSON objects now requires that the :attr:`.JSON.Comparator.astext` modifier be called explicitly, if the cast works only from a textual string. Index operations return an expression object whose type defaults to :class:`.JSON` by default, so that further JSON-oriented instructions may be called upon the result type. Custom serializers and deserializers are specified at the dialect level, that is using :func:`.create_engine`. The reason for this is that when using psycopg2, the DBAPI only allows serializers at the per-cursor or per-connection level. E.g.:: engine = create_engine("postgresql://scott:tiger@localhost/test", json_serializer=my_serialize_fn, json_deserializer=my_deserialize_fn ) When using the psycopg2 dialect, the json_deserializer is registered against the database using ``psycopg2.extras.register_default_json``. .. seealso:: :class:`.types.JSON` - Core level JSON type :class:`.JSONB` FNcs$tt|j|d|dk r ||_dS)a Construct a :class:`.JSON` type. :param none_as_null: if True, persist the value ``None`` as a SQL NULL value, not the JSON encoding of ``null``. Note that when this flag is False, the :func:`.null` construct can still be used to persist a NULL value:: from sqlalchemy import null conn.execute(table.insert(), data=null()) .. versionchanged:: 0.9.8 - Added ``none_as_null``, and :func:`.null` is now supported in order to persist a NULL value. .. seealso:: :attr:`.JSON.NULL` :param astext_type: the type to use for the :attr:`.JSON.Comparator.astext` accessor on indexed attributes. Defaults to :class:`.types.Text`. .. versionadded:: 1.1 ) none_as_nullN)superr __init__ astext_type)r$r+r. __class__rrr-sz JSON.__init__c@seZdZdZeddZdS)zJSON.Comparator0Define comparison operations for :class:`.JSON`.cCsPt|jjjtjjr0|jjjt |jj|jj dS|jjjt |jj|jj dSdS)a On an indexed expression, use the "astext" (e.g. "->>") conversion when rendered in SQL. E.g.:: select([data_table.c.data['some key'].astext]) .. seealso:: :meth:`.ColumnElement.cast` Z result_typeN) rexprrighttypesqltypesr rleftoperateJSONPATH_ASTEXTr.ASTEXT)r$rrrastextszJSON.Comparator.astextN)r(r)r*__doc__propertyr;rrrr Comparatorsr>)FN) r(r)r*r<r6ZTextr.r-r r>comparator_factory __classcell__rrr/rr fs Hr jsonc@s*eZdZdZdZGdddejZeZdS)r a_Represent the PostgreSQL JSONB type. The :class:`.JSONB` type stores arbitrary JSONB format data, e.g.:: data_table = Table('data_table', metadata, Column('id', Integer, primary_key=True), Column('data', JSONB) ) with engine.connect() as conn: conn.execute( data_table.insert(), data = {"key1": "value1", "key2": "value2"} ) The :class:`.JSONB` type includes all operations provided by :class:`.JSON`, including the same behaviors for indexing operations. It also adds additional operators specific to JSONB, including :meth:`.JSONB.Comparator.has_key`, :meth:`.JSONB.Comparator.has_all`, :meth:`.JSONB.Comparator.has_any`, :meth:`.JSONB.Comparator.contains`, and :meth:`.JSONB.Comparator.contained_by`. Like the :class:`.JSON` type, the :class:`.JSONB` type does not detect in-place changes when used with the ORM, unless the :mod:`sqlalchemy.ext.mutable` extension is used. Custom serializers and deserializers are shared with the :class:`.JSON` class, using the ``json_serializer`` and ``json_deserializer`` keyword arguments. These must be specified at the dialect level using :func:`.create_engine`. When using psycopg2, the serializers are associated with the jsonb type using ``psycopg2.extras.register_default_jsonb`` on a per-connection basis, in the same way that ``psycopg2.extras.register_default_json`` is used to register these handlers with the json type. .. versionadded:: 0.9.7 .. seealso:: :class:`.JSON` c@s8eZdZdZddZddZddZdd Zd d Zd S) zJSONB.Comparatorr1cCs|jt|tjdS)zvBoolean expression. Test for presence of a key. Note that the key may be a SQLA expression. r2)r8HAS_KEYr6Booleanr$otherrrrhas_key"szJSONB.Comparator.has_keycCs|jt|tjdS)zHBoolean expression. Test for presence of all keys in jsonb r2)r8HAS_ALLr6rCrDrrrhas_all(szJSONB.Comparator.has_allcCs|jt|tjdS)zGBoolean expression. Test for presence of any key in jsonb r2)r8HAS_ANYr6rCrDrrrhas_any-szJSONB.Comparator.has_anycKs|jt|tjdS)zBoolean expression. Test if keys (or array) are a superset of/contained the keys of the argument jsonb expression. r2)r8CONTAINSr6rC)r$rEkwargsrrrcontains2szJSONB.Comparator.containscCs|jt|tjdS)z|Boolean expression. Test if keys are a proper subset of the keys of the argument jsonb expression. r2)r8 CONTAINED_BYr6rCrDrrr contained_by8s zJSONB.Comparator.contained_byN) r(r)r*r<rFrHrJrMrOrrrrr>s r>N)r(r)r*r<Z__visit_name__r r>r?rrrrr s+!r ZjsonbN) __future__rbaserrrr6rZsqlr __all__Z _PRECEDENCEZjson_getitem_opZidx_precedenceZ custom_opr:r9rBrGrIrKrNr rr rrrrsr         R