
    g\-                     D   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ej(                  j*                        Z G d de      Z G d dej(                        Z G d de      Zy)   )ARRAY)array)ASTEXT)CONTAINED_BY)CONTAINS)DELETE_PATH)HAS_ALL)HAS_ANY)HAS_KEY)JSONPATH_ASTEXT)PATH_EXISTS)
PATH_MATCH   )types)cast)JSONJSONBc                       e Zd Zd Zd Zd Zy)JSONPathTypec                     fd}|S )Nc                     t        | t              r| S | r#ddj                  t        t        |             z  } nd} r |       } | S )Nz{%s}z, z{})
isinstancestrjoinmap)value
super_procs    b/var/www/html/AdoubleTech/venv/lib/python3.12/site-packages/sqlalchemy/dialects/postgresql/json.pyprocessz(JSONPathType._processor.<locals>.process   sI    %%  $))CUO"<="5)L     )selfdialectr   r   s     ` r   
_processorzJSONPathType._processor   s    	 r    c                 D    | j                  || j                  |            S N)r$   string_bind_processorr"   r#   s     r   bind_processorzJSONPathType.bind_processor/   s    w(B(B7(KLLr    c                 D    | j                  || j                  |            S r&   )r$   string_literal_processorr(   s     r   literal_processorzJSONPathType.literal_processor2   s    w(E(Eg(NOOr    N)__name__
__module____qualname__r$   r)   r,   r!   r    r   r   r      s    $MPr    r   c                       e Zd ZdZd Zy)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   r1   r1   6   s      Nr    r1   c                        e Zd ZdZdZ ej                         Zd fd	Z G d dej                  j                        Z
e
Z x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`

    Tc                 :    t         |   |       ||| _        yy)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"   r6   r9   	__class__s      r   r8   zJSON.__init__   s(    * 	l3"*D #r    c                        e Zd ZdZed        Zy)JSON.Comparator6Define comparison operations for :class:`_types.JSON`.c                    t        | j                  j                  j                  t        j
                  j                        rT| j                  j                  j                  t        | j                  j                  | j                  j                        S | j                  j                  j                  t        | j                  j                  | j                  j                        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   r9   r   )r"   s    r   astextzJSON.Comparator.astext   s     $))//..0J0JKyy~~--#IIOO $		 5 5 .   yy~~--DIIOO9N9N .  r    N)r-   r.   r/   r2   propertyrG   r!   r    r   
Comparatorr<      s    D		 
	r    rI   )FN)r-   r.   r/   r2   render_bind_castrD   Textr9   r8   r   rI   comparator_factory__classcell__)r:   s   @r   r   r   H   sB    KZ (--/K+2X]]-- 8 $r    r   c                   B    e Zd ZdZd Z G d dej                        ZeZy)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                   @    e Zd ZdZd Zd Zd Zd Zd Zd Z	d Z
d	 Zy
)JSONB.Comparatorr=   c                 N    | j                  t        |t        j                        S )zBoolean expression.  Test for presence of a key (equivalent of
            the ``?`` operator).  Note that the key may be a SQLA expression.
            r?   )rF   r   rD   Booleanr"   others     r   has_keyzJSONB.Comparator.has_key       <<H<L<L<MMr    c                 N    | j                  t        |t        j                        S )zxBoolean expression.  Test for presence of all keys in jsonb
            (equivalent of the ``?&`` operator)
            r?   )rF   r	   rD   rR   rS   s     r   has_allzJSONB.Comparator.has_all  rV   r    c                 N    | j                  t        |t        j                        S )zwBoolean expression.  Test for presence of any key in jsonb
            (equivalent of the ``?|`` operator)
            r?   )rF   r
   rD   rR   rS   s     r   has_anyzJSONB.Comparator.has_any  rV   r    c                 N    | j                  t        |t        j                        S )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.
            r?   )rF   r   rD   rR   )r"   rT   kwargss      r   containszJSONB.Comparator.contains  s     <<%X=M=M<NNr    c                 N    | j                  t        |t        j                        S )zBoolean expression.  Test if keys are a proper subset of the
            keys of the argument jsonb expression
            (equivalent of the ``<@`` operator).
            r?   )rF   r   rD   rR   rS   s     r   contained_byzJSONB.Comparator.contained_by$  s'    
 <<e1A1A    r    c                     t        |t              st        |      }t        |t        t        j
                              }| j                  t        |t              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
            r?   )	r   	_pg_arrayr   r   rD   TEXTrF   r   r   )r"   r   
right_sides      r   delete_pathzJSONB.Comparator.delete_path-  sB     eY/!%(eU8==%9:J<<ZU<KKr    c                 N    | j                  t        |t        j                        S )zBoolean expression. Test for presence of item given by the
            argument JSONPath expression (equivalent of the ``@?`` operator).

            .. versionadded:: 2.0
            r?   )rF   r   rD   rR   rS   s     r   path_existszJSONB.Comparator.path_exists;  s'     <<U0@0@    r    c                 N    | j                  t        |t        j                        S )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
            r?   )rF   r   rD   rR   rS   s     r   
path_matchzJSONB.Comparator.path_matchE  s'     <<Ex/?/?    r    N)r-   r.   r/   r2   rU   rX   rZ   r]   r_   rd   rf   rh   r!   r    r   rI   rP     s2    D	N	N	N	O		L		r    rI   N)r-   r.   r/   r2   r3   r   rI   rL   r!   r    r   r   r      s*    /b NKT__ KZ $r    r   N)r   r   ra   	operatorsr   r   r   r   r	   r
   r   r   r   r    r   rD   sqlr   __all__r   r   r1   r   r!   r    r   <module>rm      s{     %  #  "    & " ! ! 
P8==-- P4 |  $F$8== F$RA$D A$r    