o =a/@sddlZddlmZddlmZddlmZddlmZddlmZddlm Z ej fd d Z ej fd d Z Gd ddej ejZejddddZejddddZejddddZGdddejZdS)N)types)util) coercions) expression) operators)rolescC |||S)zA synonym for the :meth:`.ARRAY.Comparator.any` method. This method is legacy and is here for backwards-compatibility. .. seealso:: :func:`_expression.any_` )anyotherarrexproperatorrZ/var/www/html/gps/gps/lib/python3.10/site-packages/sqlalchemy/dialects/postgresql/array.pyAny rcCr )zA synonym for the :meth:`.ARRAY.Comparator.all` method. This method is legacy and is here for backwards-compatibility. .. seealso:: :func:`_expression.all_` )allr rrrAll rrcsHeZdZdZdZdZfddZeddZd d d Z dd d Z Z 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` postgresqlc sdd|D}tt|j|i|dd|D|_|d|jr%|jdntj}t|trBt|j |j dur;|j dndd|_ dSt||_ dS) NcSsg|] }ttj|qSr)rexpectrExpressionElementRole).0crrr fsz"array.__init__..cSsg|]}|jqSr)type)rargrrrrlstype_r) dimensions) superr__init__ _type_tuplepopsqltypesNULLTYPE isinstanceARRAY item_typer!r)selfclauseskw main_type __class__rrr#es"   zarray.__init__cCs|fSNrr+rrr_select_iterable|zarray._select_iterableFNcs@|stjurtjd|jddStfdd|DS)NT)_compared_to_operatorr_compared_to_typeuniquecsg|] }j|ddqS)T)_assume_scalarr) _bind_param)rorr+rrrrs z%array._bind_param..)rgetitemr BindParameterrr)r+robjr8rrr;rr9s zarray._bind_paramcCs"|tjtjtjfvrt|S|Sr1)rany_opall_opr<rGrouping)r+againstrrr self_groups zarray.self_group)FNr1) __name__ __module__ __qualname____doc____visit_name__stringify_dialectr#propertyr3r9rC __classcell__rrr/rr.s1   rz@>T) precedence is_comparisonz<@z&&c@seZdZdZGdddejjZeZ dddZe dd Z e d d Z d d Z ddZ ejddZddZddZddZdS)r)aPostgreSQL 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`. .. seealso:: :class:`_types.ARRAY` - base array type :class:`_postgresql.array` - produces a literal array value. c@s(eZdZdZddZddZddZdS) zARRAY.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`. cK|jt|tjdS)zBoolean expression. Test if elements are a superset of the elements of the argument array expression.  result_type)operateCONTAINSr&Boolean)r+r kwargsrrrcontainszARRAY.Comparator.containscCrO)zBoolean expression. Test if elements are a proper subset of the elements of the argument array expression. rP)rR CONTAINED_BYr&rTr+r rrr contained_byszARRAY.Comparator.contained_bycCrO)zuBoolean expression. Test if array has elements in common with an argument array expression. rP)rROVERLAPr&rTrYrrroverlaprWzARRAY.Comparator.overlapN)rDrErFrGrVrZr\rrrr Comparators   r]FNcCs>t|tr tdt|tr|}||_||_||_||_dS)aPConstruct an ARRAY. E.g.:: Column('myarray', ARRAY(Integer)) Arguments are: :param item_type: The data type of items of this array. Note that dimensionality is irrelevant here, so multi-dimensional arrays like ``INTEGER[][]``, are constructed as ``ARRAY(Integer)``, not as ``ARRAY(ARRAY(Integer))`` or such. :param as_tuple=False: Specify whether return results should be converted to tuples from lists. DBAPIs such as psycopg2 return lists by default. When tuples are returned, the results are hashable. :param dimensions: if non-None, the ARRAY will assume a fixed number of dimensions. This will cause the DDL emitted for this ARRAY to include the exact number of bracket clauses ``[]``, and will also optimize the performance of the type overall. Note that PG arrays are always implicitly "non-dimensioned", meaning they can store any number of dimensions no matter how they were declared. :param zero_indexes=False: when True, index values will be converted between Python zero-based and PostgreSQL one-based indexes, e.g. a value of one will be added to all index values before passing to the database. .. versionadded:: 0.9.5 zUDo not nest ARRAY types; ARRAY(basetype) handles multi-dimensional arrays of basetypeN)r(r) ValueErrorrr*as_tupler! zero_indexes)r+r*r_r!r`rrrr#s &  zARRAY.__init__cCs|jSr1)r_r2rrrhashable%r4zARRAY.hashablecCstSr1)listr2rrr python_type)szARRAY.python_typecCs||kSr1r)r+xyrrrcompare_values-szARRAY.compare_valuescstdurt|}dksdur,|rt|dttfs,r(fdd|DS|Sfdd|DS)Nrrc3s|]}|VqdSr1rrrd)itemprocrr ?sz$ARRAY._proc_array..c3s0|]}|durdndVqdS)Nr) _proc_arrayrg collectiondimrhr+rrriCs )rbr(tuple)r+arrrhrmrlrrkrrj0s  zARRAY._proc_arraycCst|jtjo |jjSr1)r(r*r&Enum native_enumr2rrr_against_native_enumMszARRAY._against_native_enumcCs|Sr1r)r+ bindvaluerrrbind_expressionTszARRAY.bind_expressioncs$j||fdd}|S)Ncs|dur|S|jtSr1)rjr!rbvalue item_procr+rrprocess\s  z%ARRAY.bind_processor..process)r* dialect_implbind_processor)r+dialectryrrwrr{Ws zARRAY.bind_processorcsFj|||fdd}jr!|ddfdd}|S)Ncs*|dur|S|jjrtStSr1)rjr!r_rnrbrurwrrryksz'ARRAY.result_processor..processcSs$td|d}|r|dSgS)Nz^{(.*)}$r,)rematchgroupsplit)rvinnerrrrhandle_raw_stringysz1ARRAY.result_processor..handle_raw_stringcs*|dur|St|tjr|S|Sr1)r(r string_typesru)rsuper_rprrry}s )r*rzresult_processorrr)r+r|coltyperyr)rrxr+rrrfs   zARRAY.result_processor)FNF)rDrErFrGr&r)r]comparator_factoryr#rJrarcrfrjrmemoized_propertyrrrtr{rrrrrr)s"- 2    r))r~rr&rsqlrrrreqrr ClauseList ColumnElementr custom_oprSrXr[r)rrrrs      n