ih*2ddlmZddlmZddlmZddlmZddlmZddlmZddlmZdd l m Z dd l m Z dd l m Z dd l mZdd l mZddl mZddl mZddl mZddl mZddl mZddl mZddl mZddlmZddlmZddlmZerddlmZddl m!Z!ddl"m#Z#ddl"m$Z$ddl"m%Z%dZ&Gdd ej'j(Z(Gd!d"e(Z)Gd#d$ej'Z'Gd%d&e'Z*d'S)() annotations)Any)Callable)List)Optional) TYPE_CHECKING)Union)ARRAY)array)ASTEXT) CONTAINED_BY)CONTAINS) DELETE_PATH)HAS_ALL)HAS_ANY)HAS_KEY)JSONPATH_ASTEXT) PATH_EXISTS) PATH_MATCH)types)cast)_T)Dialect) ColumnElement)_BindProcessorType)_LiteralProcessorType) TypeEngine)JSONJSONBc&eZdZd dZdd Zdd Zd S) JSONPathTypedialectr super_procOptional[Callable[[Any], Any]]returnCallable[[Any], Any]cdfd }|S)Nvaluerr'ct|tr|S|r,ddtt|z}nd}r |}|S)Nz{%s}z, z{}) isinstancestrjoinmap)r*r%s k/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/sqlalchemy/dialects/postgresql/json.pyprocessz(JSONPathType._processor..process/sg%%%   $))CUOO"<"<= *" 5))L)r*rr'r)selfr$r%r1s ` r0 _processorzJSONPathType._processor,s(      r2_BindProcessorType[Any]cT||||SN)r5string_bind_processorr4r$s r0bind_processorzJSONPathType.bind_processor@s$w(B(B7(K(KLLLr2_LiteralProcessorType[Any]cT||||Sr8)r5string_literal_processorr:s r0literal_processorzJSONPathType.literal_processorCs&w(E(Eg(N(NOOOr2N)r$rr%r&r'r()r$rr'r6)r$rr'r<)__name__ __module__ __qualname__r5r;r?r3r2r0r#r#+sZ(MMMMPPPPPPr2r#ceZdZdZdZdS)JSONPATHadJSON 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@rArB__doc____visit_name__r3r2r0rDrDIs   NNNr2rDceZdZUdZdZejZded< d dfd Z Gd d ej j e Z e Z xZS)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 )``):: 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` TzTypeEngine[str] astext_typeFN none_as_nullboolOptional[TypeEngine[str]]cbt|| ||_dSdS)aConstruct 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`. )rIN)super__init__rH)r4rIrH __class__s r0rNz JSON.__init__s<2 l333  "*D    # "r2c6eZdZUdZded<eddZdS) JSON.Comparator6Define comparison operations for :class:`_types.JSON`.r typer'ColumnElement[str]cJt|jjjtjjr;|jjt|jj|jj S|jjt|jj|jj S)aOn 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,exprrightrSsqltypesr r#leftoperaterrHr )r4s r0astextzJSON.Comparator.astexts$)/. 0JKK y~--#IO $ 5. y~--DIO9N.r2N)r'rT)r@rArBrE__annotations__propertyr]r3r2r0 ComparatorrQsCDD        r2r`)FN)rIrJrHrK)r@rArBrErender_bind_castrZTextrHr^rNr r`rcomparator_factory __classcell__)rOs@r0r r [sKKZ#08=??K2222#15+++++++:X]-b1<$r2r cLeZdZdZdZGddejeZeZdS)r!aRepresent 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^eZdZUdZded<ddZdd Zdd Zdd Zdd Z ddZ ddZ ddZ dS)JSONB.ComparatorrRr!rSotherrr'ColumnElement[bool]cP|t|tjS)zBoolean expression. Test for presence of a key (equivalent of the ``?`` operator). Note that the key may be a SQLA expression. rV)r\rrZBooleanr4rhs r0has_keyzJSONB.Comparator.has_key#<<H`` operator). kwargs may be ignored by this operator but are required for API conformance. rV)r\rrZrk)r4rhrss r0containszJSONB.Comparator.contains5s<<%X=M<NN Nr2cP|t|tjS)zBoolean expression. Test if keys are a proper subset of the keys of the argument jsonb expression (equivalent of the ``<@`` operator). rV)r\rrZrkrls r0 contained_byzJSONB.Comparator.contained_by?s) <<e1A  r2r Union[List[str], _pg_array[str]]ColumnElement[JSONB]ct|tst|}t|ttj}|t|tS)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 rV) r, _pg_arrayrr rZTEXTr\rr!)r4r right_sides r0 delete_pathzJSONB.Comparator.delete_pathHsTeY// )!%((eU8=%9%9::J<< ZU<KK Kr2cP|t|tjS)zBoolean expression. Test for presence of item given by the argument JSONPath expression (equivalent of the ``@?`` operator). .. versionadded:: 2.0 rV)r\rrZrkrls r0 path_existszJSONB.Comparator.path_existsXs) <<U0@  r2cP|t|tjS)aBoolean 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 rV)r\rrZrkrls r0 path_matchzJSONB.Comparator.path_matchbs)<<Ex/?  r2N)rhrr'ri)rhrrsrr'ri)r rxr'ry) r@rArBrEr^rmrprrrurwr~rrr3r2r0r`rgsDD  N N N N  N N N N  N N N N  O O O O     L L L L           r2r`N) r@rArBrErFr r`rrcr3r2r0r!r!sd//bNOOOOOT_R(OOOb$r2r!N)+ __future__rtypingrrrrrr r r r{ operatorsr rrrrrrrrrrrZsqlr sql._typingrengine.interfacesr sql.elementsr sql.type_apirrr__all__r r#rDr!r3r2r0rs#""""" %%%%%%######""""""&&&&&&""""""!!!!!!!!!!!!+,,,,,,------222222555555****** PPPPP8=-PPP<     |   $L$L$L$L$L$8=L$L$L$^E$E$E$E$E$DE$E$E$E$E$r2