o
    g\-                     @   s   d dl mZ d dl m Z d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlm	Z	 d d	lm
Z
 d d
lmZ d dlmZ d dlmZ ddlmZ ddlmZ dZG dd dejjZG dd deZG dd dejZG dd deZdS )   )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 Zd Zdd Zdd Zdd ZdS )JSONPathTypec                    s    fdd}|S )Nc                    s<   t | tr| S | rddtt|  } nd}  r | } | S )Nz{%s}z, z{})
isinstancestrjoinmap)value
super_proc e/var/www/html/ecg_monitoring/venv/lib/python3.10/site-packages/sqlalchemy/dialects/postgresql/json.pyprocess   s   
z(JSONPathType._processor.<locals>.processr   )selfdialectr   r   r   r   r   
_processor   s   zJSONPathType._processorc                 C      |  || |S N)r    string_bind_processorr   r   r   r   r   bind_processor/      zJSONPathType.bind_processorc                 C   r!   r"   )r    string_literal_processorr$   r   r   r   literal_processor2   r&   zJSONPathType.literal_processorN)__name__
__module____qualname__r    r%   r(   r   r   r   r   r      s    r   c                   @   s   e Zd ZdZd ZdS )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                       sF   e Zd ZdZdZe Zd	 fdd	ZG dd dej	j
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`

    TFNc                    s$   t  j|d |dur|| _dS dS )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/   r2   	__class__r   r   r1      s   
zJSON.__init__c                   @   s   e Zd ZdZedd ZdS )zJSON.Comparator6Define comparison operations for :class:`_types.JSON`.c                 C   sL   t | jjjtjjr| jjjt	| jj| jj
dS | jjjt| jj| jj
dS )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`

            result_type)r   exprrighttypesqltypesr   r   leftoperater   r2   r   )r   r   r   r   astext   s   zJSON.Comparator.astextN)r)   r*   r+   r-   propertyr>   r   r   r   r   
Comparator   s    r@   )FN)r)   r*   r+   r-   render_bind_castr;   Textr2   r1   r   r@   comparator_factory__classcell__r   r   r3   r   r   H   s    Mr   c                   @   s*   e Zd ZdZd ZG dd dejZeZdS )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 Zd ZdZdd Zdd Zdd Zdd	 Zd
d Zdd Z	dd Z
dd ZdS )zJSONB.Comparatorr5   c                 C      | j t|tjdS )zBoolean expression.  Test for presence of a key (equivalent of
            the ``?`` operator).  Note that the key may be a SQLA expression.
            r6   )r=   r
   r;   Booleanr   otherr   r   r   has_key     zJSONB.Comparator.has_keyc                 C   rE   )zxBoolean expression.  Test for presence of all keys in jsonb
            (equivalent of the ``?&`` operator)
            r6   )r=   r   r;   rF   rG   r   r   r   has_all  rJ   zJSONB.Comparator.has_allc                 C   rE   )zwBoolean expression.  Test for presence of any key in jsonb
            (equivalent of the ``?|`` operator)
            r6   )r=   r	   r;   rF   rG   r   r   r   has_any  rJ   zJSONB.Comparator.has_anyc                 K   rE   )a"  Boolean expression.  Test if keys (or array) are a superset
            of/contained the keys of the argument jsonb expression
            (equivalent of the ``@>`` operator).

            kwargs may be ignored by this operator but are required for API
            conformance.
            r6   )r=   r   r;   rF   )r   rH   kwargsr   r   r   contains  s   zJSONB.Comparator.containsc                 C   rE   )zBoolean expression.  Test if keys are a proper subset of the
            keys of the argument jsonb expression
            (equivalent of the ``<@`` operator).
            r6   )r=   r   r;   rF   rG   r   r   r   contained_by$  s   zJSONB.Comparator.contained_byc                 C   s2   t |ts	t|}t|ttj}| jt|tdS )a=  JSONB expression. Deletes field or array element specified in
            the argument array (equivalent of the ``#-`` operator).

            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
            r6   )	r   	_pg_arrayr   r   r;   TEXTr=   r   r   )r   r   
right_sider   r   r   delete_path-  s   
	zJSONB.Comparator.delete_pathc                 C   rE   )zBoolean expression. Test for presence of item given by the
            argument JSONPath expression (equivalent of the ``@?`` operator).

            .. versionadded:: 2.0
            r6   )r=   r   r;   rF   rG   r   r   r   path_exists;  s   zJSONB.Comparator.path_existsc                 C   rE   )a  Boolean expression. Test if JSONPath predicate given by the
            argument JSONPath expression matches
            (equivalent of the ``@@`` operator).

            Only the first item of the result is taken into account.

            .. versionadded:: 2.0
            r6   )r=   r   r;   rF   rG   r   r   r   
path_matchE  s   	zJSONB.Comparator.path_matchN)r)   r*   r+   r-   rI   rK   rL   rN   rO   rS   rT   rU   r   r   r   r   r@     s    
	
r@   N)r)   r*   r+   r-   r.   r   r@   rC   r   r   r   r   r      s
    1Mr   N)r   r   rP   	operatorsr   r   r   r   r   r	   r
   r   r   r    r   r;   sqlr   __all__r   r   r,   r   r   r   r   r   <module>   s(   	 
