+hu7~ddlZddlmZddlmZddlmZddlmZddlmZddlm Z ejfd Z ejfd Z Gd d ejejZej d ddZej dddZej dddZGddej(ZdZy)N)types)util) coercions) expression) operators)rolesc&|j||S)zjA synonym for the ARRAY-level :meth:`.ARRAY.Comparator.any` method. See that method for details. )anyotherarrexproperators W/var/www/html/venv/lib/python3.12/site-packages/sqlalchemy/dialects/postgresql/array.pyAnyr ;;uh ''c&|j||S)zjA synonym for the ARRAY-level :meth:`.ARRAY.Comparator.all` method. See that method for details. )allr s rAllrrrcNeZdZdZdZdZdZfdZedZ ddZ d dZ xZ S) arrayaA PostgreSQL ARRAY literal. This is used to produce ARRAY literals in SQL expressions, e.g.:: from sqlalchemy.dialects.postgresql import array from sqlalchemy.dialects import postgresql from sqlalchemy import select, func stmt = select(array([1,2]) + array([3,4,5])) print(stmt.compile(dialect=postgresql.dialect())) Produces the SQL:: SELECT ARRAY[%(param_1)s, %(param_2)s] || ARRAY[%(param_3)s, %(param_4)s, %(param_5)s]) AS anon_1 An instance of :class:`.array` will always have the datatype :class:`_types.ARRAY`. The "inner" type of the array is inferred from the values present, unless the ``type_`` keyword argument is passed:: array(['foo', 'bar'], type_=CHAR) Multidimensional arrays are produced by nesting :class:`.array` constructs. The dimensionality of the final :class:`_types.ARRAY` type is calculated by recursively adding the dimensions of the inner :class:`_types.ARRAY` type:: stmt = select( array([ array([1, 2]), array([3, 4]), array([column('q'), column('x')]) ]) ) print(stmt.compile(dialect=postgresql.dialect())) Produces:: SELECT ARRAY[ARRAY[%(param_1)s, %(param_2)s], ARRAY[%(param_3)s, %(param_4)s], ARRAY[q, x]] AS anon_1 .. versionadded:: 1.3.6 added support for multidimensional array literals .. seealso:: :class:`_postgresql.ARRAY` postgresqlTc |Dcgc]&}tjtj|(}}t t ||i||Dcgc]}|jc}|_|jd|jr|jdntj}t|tr8t|j|j|jdznd|_yt||_ycc}wcc}w)Ntype_r) dimensions)rexpectr ExpressionElementRolesuperr__init__type _type_tuplepopsqltypesNULLTYPE isinstanceARRAY item_typer)selfclauseskwcarg main_type __class__s rr"zarray.__init__\sFM ABI  U88! <   eT#W33078CHH8FF #'#3#3D  Q 9J9J i '##''3%//!3 DIi(DI) 9s +C9 C>c|fSNr+s r_select_iterablezarray._select_iterabless wrc |s|tjur%tjd||||jdSt |Dcgc]}|j ||d|c}Scc}w)NT)_compared_to_operatorr_compared_to_typeunique)_assume_scalarr)rgetitemr BindParameterr#r _bind_param)r+robjr;ros rr>zarray._bind_paramws X):)::++&."&))   ! $$ !D% sA&c|tjtjtjfvrt j |S|Sr3)rany_opall_opr<rGrouping)r+againsts r self_groupzarray.self_groups9 y'')9)99;L;LM M&&t, ,Kr)FNr3) __name__ __module__ __qualname____doc____visit_name__stringify_dialect inherit_cacher"propertyr6r>rF __classcell__)r1s@rrr$s>/bN$M).*rrz@>T) precedence is_comparisonz<@z&&ceZdZdZGddej j ZeZ ddZe dZ e dZ dZ d Z ejd Zd Zd Zd Zy)r)a PostgreSQL ARRAY type. .. versionchanged:: 1.1 The :class:`_postgresql.ARRAY` type is now a subclass of the core :class:`_types.ARRAY` type. The :class:`_postgresql.ARRAY` type is constructed in the same way as the core :class:`_types.ARRAY` type; a member type is required, and a number of dimensions is recommended if the type is to be used for more than one dimension:: from sqlalchemy.dialects import postgresql mytable = Table("mytable", metadata, Column("data", postgresql.ARRAY(Integer, dimensions=2)) ) The :class:`_postgresql.ARRAY` type provides all operations defined on the core :class:`_types.ARRAY` type, including support for "dimensions", indexed access, and simple matching such as :meth:`.types.ARRAY.Comparator.any` and :meth:`.types.ARRAY.Comparator.all`. :class:`_postgresql.ARRAY` class also provides PostgreSQL-specific methods for containment operations, including :meth:`.postgresql.ARRAY.Comparator.contains` :meth:`.postgresql.ARRAY.Comparator.contained_by`, and :meth:`.postgresql.ARRAY.Comparator.overlap`, e.g.:: mytable.c.data.contains([1, 2]) The :class:`_postgresql.ARRAY` type may not be supported on all PostgreSQL DBAPIs; it is currently known to work on psycopg2 only. Additionally, the :class:`_postgresql.ARRAY` type does not work directly in conjunction with the :class:`.ENUM` type. For a workaround, see the special type at :ref:`postgresql_array_of_enum`. .. container:: topic **Detecting Changes in ARRAY columns when using the ORM** The :class:`_postgresql.ARRAY` type, when used with the SQLAlchemy ORM, does not detect in-place mutations to the array. In order to detect these, the :mod:`sqlalchemy.ext.mutable` extension must be used, using the :class:`.MutableList` class:: from sqlalchemy.dialects.postgresql import ARRAY from sqlalchemy.ext.mutable import MutableList class SomeOrmClass(Base): # ... data = Column(MutableList.as_mutable(ARRAY(Integer))) This extension will allow "in-place" changes such to the array such as ``.append()`` to produce events which will be detected by the unit of work. Note that changes to elements **inside** the array, including subarrays that are mutated in place, are **not** detected. Alternatively, assigning a new array value to an ORM element that replaces the old one will always trigger a change event. .. seealso:: :class:`_types.ARRAY` - base array type :class:`_postgresql.array` - produces a literal array value. c"eZdZdZdZdZdZy)ARRAY.Comparatora*Define comparison operations for :class:`_types.ARRAY`. Note that these operations are in addition to those provided by the base :class:`.types.ARRAY.Comparator` class, including :meth:`.types.ARRAY.Comparator.any` and :meth:`.types.ARRAY.Comparator.all`. c N|jt|tjS)zBoolean expression. Test if elements are a superset of the elements of the argument array expression. kwargs may be ignored by this operator but are required for API conformance.  result_type)operateCONTAINSr&Boolean)r+r kwargss rcontainszARRAY.Comparator.containss<<%X=M=M<N NrcN|jt|tjS)zBoolean expression. Test if elements are a proper subset of the elements of the argument array expression. rW)rY CONTAINED_BYr&r[r+r s r contained_byzARRAY.Comparator.contained_bys'<<e1A1A  rcN|jt|tjS)zuBoolean expression. Test if array has elements in common with an argument array expression. rW)rYOVERLAPr&r[r`s roverlapzARRAY.Comparator.overlaps<<Hz$ARRAY._proc_array..Rs!;!(1+!;sc3TK|]}j|dz nd!ywNr) _proc_array)rurp collectiondimrvr+s rrwz$ARRAY._proc_array..Vs<   "C!GD s%()rmr(tuple)r+arrrvr|r{s` ```rrzzARRAY._proc_arrayCsp ;s)C 1H{ !#a&4-8!!;s!;;;!#& rczt|jtjxr|jjSr3)r(r*r&Enum native_enumr5s r_against_native_enumzARRAY._against_native_enum`s- t~~x}} 5 +** rc|Sr3r4)r+ bindvalues rbind_expressionzARRAY.bind_expressiongsrcjjj|j|fd}|S)NcP||Sj|jtSr3)rzrrmvalue item_procr+s rprocessz%ARRAY.bind_processor..processos-} ''9dootr)r* dialect_implbind_processor)r+dialectrrs` @rrzARRAY.bind_processorjs1NN//8GG   rcjj|j||fd}jr"|t j dfdfd}|S)Nc|||Sj|jjr tStSr3)rzrrhr}rmrs rrz'ARRAY.result_processor..process~sE} ''OO!]]E 15 rz^{(.*)}$cZj|jd}t|Sry)matchgroup_split_enum_values)rinnerpatterns rhandle_raw_stringz1ARRAY.result_processor..handle_raw_strings' e,2215)%00rch||St|tjr |S|Sr3)r(r string_types)rrsuper_rps rrz'ARRAY.result_processor..processsE= L !%):):;&e,r)r*rresult_processorrrecompile)r+rcoltyperrrrrs` @@@@rrzARRAY.result_processorys[NN//8II W    $ $Hjj-G 1 r)FNF)rGrHrIrJr&r)recomparator_factoryr"rNrkrnrrrzrmemoized_propertyrrrrr4rrr)r)sDL NX^^.. ND$HM0)d:    $rr)cfd|vr|r|jdSgS|jdd}|jdd}g}tjd|}d}|D]T}|dk(r| } |r"|j|jdd0|j tj d |V|S) N",z\"z _$ESC_QUOTE$_z\\\z(")Fz ([^\s,]+),?)splitreplacerappendextendfindall) array_stringtextresult on_quotes in_quotestoks rrrs ,*6|!!#&>B>    7D <<t $D F&II; #:% I  MM#++os; < MM"**^S9 : ; Mr)rrr&rsqlrrrr eqrr ClauseList ColumnElementr custom_oprZr_rcr)rr4rrrs !"+("+(lJ ! !:#;#;l^ 9  t F"y""4ATJ )  dq ECHNNCLr