관리-도구
편집 파일: json.cpython-38.pyc
U -?�f�+ � @ s� d dl mZ d dl m Z d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlm Z d d lm Z d d lm Z d d lm Z d d lm Z d dlmZ d dlmZ dZG dd� dejj�ZG dd� de�ZG dd� dej�ZG dd� de�ZdS )� )�ARRAY)�array)�ASTEXT)� CONTAINED_BY)�CONTAINS)� DELETE_PATH)�HAS_ALL)�HAS_ANY)�HAS_KEY)�JSONPATH_ASTEXT)� PATH_EXISTS)� PATH_MATCH� )�types)�cast)�JSON�JSONBc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� JSONPathTypec s � fdd�}|S )Nc s< t | t�r| S | r(dd�tt| �� } nd} � r8� | �} | S )Nz{%s}z, z{})� isinstance�str�join�map)�value�� super_proc� �S/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/dialects/postgresql/json.py�process s z(JSONPathType._processor.<locals>.processr )�self�dialectr r r r r � _processor s zJSONPathType._processorc C s | � || �|��S �N)r Zstring_bind_processor�r r r r r �bind_processor/ s zJSONPathType.bind_processorc C s | � || �|��S r! )r Zstring_literal_processorr" r r r �literal_processor2 s z JSONPathType.literal_processorN)�__name__� __module__� __qualname__r r# r$ r r r r r s r c @ s e Zd ZdZd ZdS )�JSONPATHad JSON Path Type. This is usually required to cast literal values to json path when using json search like function, such as ``jsonb_path_query_array`` or ``jsonb_path_exists``:: stmt = sa.select( sa.func.jsonb_path_query_array( table.c.jsonb_col, cast("$.address.id", JSONPATH) ) ) N)r% r& r' �__doc__�__visit_name__r r r r r( 6 s r( c sB e Zd ZdZe�� Zd� fdd� ZG dd� dejj �Z e Z � Z S ) r a Represent the PostgreSQL JSON type. :class:`_postgresql.JSON` is used automatically whenever the base :class:`_types.JSON` datatype is used against a PostgreSQL backend, however base :class:`_types.JSON` datatype does not provide Python accessors for PostgreSQL-specific comparison methods such as :meth:`_postgresql.JSON.Comparator.astext`; additionally, to use PostgreSQL ``JSONB``, the :class:`_postgresql.JSONB` datatype should be used explicitly. .. seealso:: :class:`_types.JSON` - main documentation for the generic cross-platform JSON datatype. The operators provided by the PostgreSQL version of :class:`_types.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 <type>)``):: 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' Index operations return an expression object whose type defaults to :class:`_types.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:`_sa.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+psycopg2://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:`_postgresql.JSONB` FNc s t � j|d� |dk r || _dS )a� Construct a :class:`_types.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()}) .. seealso:: :attr:`_types.JSON.NULL` :param astext_type: the type to use for the :attr:`.JSON.Comparator.astext` accessor on indexed attributes. Defaults to :class:`_types.Text`. )� none_as_nullN)�super�__init__� astext_type)r r+ r. �� __class__r r r- � s z JSON.__init__c @ s e Zd ZdZedd� �ZdS )zJSON.Comparator�6Define comparison operations for :class:`_types.JSON`.c C sP t | jjjtjj�r0| jjjt | jj| jj d�S | jjjt | jj| jj d�S dS )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:`_expression.ColumnElement.cast` �Z result_typeN) r �expr�right�type�sqltypesr r �left�operater r. r )r r r r �astext� s � �zJSON.Comparator.astextN)r% r& r' r) �propertyr9 r r r r � Comparator� s r; )FN) r% r& r' r) r6 �Textr. r- r r; �comparator_factory� __classcell__r r r/ r r H s J r c @ s* e Zd ZdZd ZG dd� dej�ZeZdS )r a Represent the PostgreSQL JSONB type. The :class:`_postgresql.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:`_postgresql.JSONB` type includes all operations provided by :class:`_types.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`, :meth:`.JSONB.Comparator.contained_by`, :meth:`.JSONB.Comparator.delete_path`, :meth:`.JSONB.Comparator.path_exists` and :meth:`.JSONB.Comparator.path_match`. Like the :class:`_types.JSON` type, the :class:`_postgresql.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:`_types.JSON` class, using the ``json_serializer`` and ``json_deserializer`` keyword arguments. These must be specified at the dialect level using :func:`_sa.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. .. seealso:: :class:`_types.JSON` c @ sP e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dS )zJSONB.Comparatorr1 c C s | j t|tjd�S )zvBoolean expression. Test for presence of a key. Note that the key may be a SQLA expression. r2 )r8 r r6 �Boolean�r �otherr r r �has_key s zJSONB.Comparator.has_keyc C s | j t|tjd�S )z;Boolean expression. Test for presence of all keys in jsonbr2 )r8 r r6 r? r@ r r r �has_all s zJSONB.Comparator.has_allc C s | j t|tjd�S )z:Boolean expression. Test for presence of any key in jsonbr2 )r8 r r6 r? r@ r r r �has_any s zJSONB.Comparator.has_anyc K s | j t|tjd�S )z�Boolean expression. Test if keys (or array) are a superset of/contained the keys of the argument jsonb expression. kwargs may be ignored by this operator but are required for API conformance. r2 )r8 r r6 r? )r rA �kwargsr r r �contains s zJSONB.Comparator.containsc C s | j t|tjd�S )z|Boolean expression. Test if keys are a proper subset of the keys of the argument jsonb expression. r2 )r8 r r6 r? r@ r r r � contained_by s �z JSONB.Comparator.contained_byc C s2 t |t�st|�}t|ttj��}| jt|td�S )a JSONB expression. Deletes field or array element specified in the argument array. The input may be a list of strings that will be coerced to an ``ARRAY`` or an instance of :meth:`_postgres.array`. .. versionadded:: 2.0 r2 ) r � _pg_arrayr r r6 ZTEXTr8 r r )r r Z right_sider r r � delete_path! s z JSONB.Comparator.delete_pathc C s | j t|tjd�S )z�Boolean expression. Test for presence of item given by the argument JSONPath expression. .. versionadded:: 2.0 r2 )r8 r r6 r? r@ r r r � path_exists/ s �z JSONB.Comparator.path_existsc C s | j t|tjd�S )z�Boolean expression. Test if JSONPath predicate given by the argument JSONPath expression matches. Only the first item of the result is taken into account. .. versionadded:: 2.0 r2 )r8 r r6 r? r@ r r r � path_match9 s �zJSONB.Comparator.path_matchN) r% r& r' r) rB rC rD rF rG rI rJ rK r r r r r; � s r; N)r% r& r' r) r* r r; r= r r r r r � s 0Fr N)r r rH � operatorsr r r r r r r r r r � r r6 �sqlr �__all__r r r( r r r r r �<module> s&