
    +h<)                        d dl mZ ddlmZ ddlmZ ddlmZ dZej                  ej                     Z ej                  dedd	      Z ej                  d
edd	      Z ej                  dedd	      Z ej                  dedd	      Z ej                  dedd	      Z ej                  dedd	      Z ej                  dedd	      Z G d dej(                  j*                        Z G d dej(                        Z G d de      Zy)    )absolute_import   )types)util)	operators)JSONJSONBz->>T)
precedencenatural_self_precedenteager_groupingz#>>?z?&z?|z@>z<@c                       e Zd Zd Zd Zy)JSONPathTypec                 4    | j                  |      fd}|S )Nc                     t        | t        j                  j                        sJ | D cg c]  }t        j                  |       }}ddj                  |      z  } r |       } | S c c}w Nz{%s}z, 
isinstancer   collections_abcSequence	text_typejoinvalueelemtokens
super_procs      V/var/www/html/venv/lib/python3.12/site-packages/sqlalchemy/dialects/postgresql/json.pyprocessz,JSONPathType.bind_processor.<locals>.processI   e    eT%9%9%B%BCCC7<=tdnnT*=F=dii/0E"5)L	 >   A*)string_bind_processorselfdialectr   r   s      @r   bind_processorzJSONPathType.bind_processorF   s    //8
	     c                 4    | j                  |      fd}|S )Nc                     t        | t        j                  j                        sJ | D cg c]  }t        j                  |       }}ddj                  |      z  } r |       } | S c c}w r   r   r   s      r   r   z/JSONPathType.literal_processor.<locals>.processV   r    r!   )string_literal_processorr#   s      @r   literal_processorzJSONPathType.literal_processorS   s    227;
	 r'   N)__name__
__module____qualname__r&   r+    r'   r   r   r   E   s    r'   r   c                        e 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'

    .. versionchanged:: 1.1  The :meth:`_expression.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:`_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://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`

    .. versionchanged:: 1.1 :class:`_postgresql.JSON` is now a PostgreSQL-
       specific specialization of the new :class:`_types.JSON` type.

    c                 B    t         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())

         .. versionchanged:: 0.9.8 - Added ``none_as_null``, and :func:`.null`
            is now supported in order to persist a NULL value.

         .. 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`.

         .. versionadded:: 1.1

        )none_as_nullN)superr   __init__astext_type)r$   r2   r5   	__class__s      r   r4   zJSON.__init__   s*    2 	dD""="*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operateJSONPATH_ASTEXTr5   ASTEXT)r$   s    r   astextzJSON.Comparator.astext   s     $))//..0J0JKyy~~--#IIOO $		 5 5 .   yy~~--DIIOO9N9N .  r'   N)r,   r-   r.   __doc__propertyrE   r/   r'   r   
Comparatorr8      s    D		 
	r'   rH   )FN)r,   r-   r.   rF   r@   Textr5   r4   r   rH   comparator_factory__classcell__)r6   s   @r   r   r   a   s<    Qf  (--/K+: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`,
    and :meth:`.JSONB.Comparator.contained_by`.

    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.

    .. versionadded:: 0.9.7

    .. seealso::

        :class:`_types.JSON`

    c                   .    e Zd ZdZd Zd Zd Zd Zd Zy)JSONB.Comparatorr9   c                 N    | j                  t        |t        j                        S )zvBoolean expression.  Test for presence of a key.  Note that the
            key may be a SQLA expression.
            r;   )rB   HAS_KEYr@   Booleanr$   others     r   has_keyzJSONB.Comparator.has_key(  s     <<H<L<L<MMr'   c                 N    | j                  t        |t        j                        S )z;Boolean expression.  Test for presence of all keys in jsonbr;   )rB   HAS_ALLr@   rQ   rR   s     r   has_allzJSONB.Comparator.has_all.      <<H<L<L<MMr'   c                 N    | j                  t        |t        j                        S )z:Boolean expression.  Test for presence of any key in jsonbr;   )rB   HAS_ANYr@   rQ   rR   s     r   has_anyzJSONB.Comparator.has_any2  rX   r'   c                 N    | j                  t        |t        j                        S )zBoolean 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.
            r;   )rB   CONTAINSr@   rQ   )r$   rS   kwargss      r   containszJSONB.Comparator.contains6  s     <<%X=M=M<NNr'   c                 N    | j                  t        |t        j                        S )z|Boolean expression.  Test if keys are a proper subset of the
            keys of the argument jsonb expression.
            r;   )rB   CONTAINED_BYr@   rQ   rR   s     r   contained_byzJSONB.Comparator.contained_by?  s'     <<e1A1A    r'   N)	r,   r-   r.   rF   rT   rW   r[   r_   rb   r/   r'   r   rH   rN   %  s"    D	N	N	N	O	r'   rH   N)r,   r-   r.   rF   __visit_name__r   rH   rJ   r/   r'   r   r	   r	      s(    -^ N T__  D $r'   r	   N)
__future__r    r   r@   r   sqlr   __all___PRECEDENCEjson_getitem_opidx_precedence	custom_oprD   rC   rP   rV   rZ   r]   ra   r   r   r	   r/   r'   r   <module>rl      sE   ' !   &&y'@'@A					
 &)%%		 )

	 )

	 )

	 9	 #y""	8==-- 8O$8== O$dT$D T$r'   