mxiIDlddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z dd lm Z d d l m Z d d l mZd dlmZd dlmZd dlmZd dlmZd dlmZd dlmZd dlmZd dlmZd dlmZe r0d dlmZd dlmZd dlm Z d dlm!Z!d dl"m#Z#d dl$m%Z%d dlm&Z&d dlm'Z'e de Z(Gd!d"ejBe(Z)Gd#d$eZ*d%Z+Gd&d'ejXe(Z-Gd(d)e-Z.Gd*d+e-Z/Gd,d-e-Z0Gd.d/e-Z1Gd0d1e-Z2Gd2d3e-Z3y4)5) annotations)Any)Iterable)List)Optional)overload)Tuple) TYPE_CHECKING)TypeVar)types)ARRAY) coercions)elements) expression) functions)roles)schema)ColumnCollectionConstraint)TEXT)InternalTraversal)_ColumnExpressionArgument)_DDLColumnArgument) ClauseElement) ColumnElement) OperatorType) FromClause)_CloneCallableType)_TraverseInternalsType_T)boundceZdZUdZdZdZdejfdejfdejfgZ de d<e ddZ e dd Z dd Z d dd Z dd Zej f ddZeddZy )aggregate_order_byaRepresent a PostgreSQL aggregate order by expression. E.g.:: from sqlalchemy.dialects.postgresql import aggregate_order_by expr = func.array_agg(aggregate_order_by(table.c.a, table.c.b.desc())) stmt = select(expr) would represent the expression: .. sourcecode:: sql SELECT array_agg(a ORDER BY b DESC) FROM table; Similarly:: expr = func.string_agg( table.c.a, aggregate_order_by(literal_column("','"), table.c.a) ) stmt = select(expr) Would represent: .. sourcecode:: sql SELECT string_agg(a, ',' ORDER BY a) FROM table; .. versionchanged:: 1.2.13 - the ORDER BY argument may be multiple terms .. seealso:: :class:`_functions.array_agg` postgresqltargettypeorder_byr _traverse_internalscyNselfr&r(s Y/home/htdocs/ttos/venv/lib/python3.12/site-packages/sqlalchemy/dialects/postgresql/ext.py__init__zaggregate_order_by.__init__Z  cyr+r,r-s r/r0zaggregate_order_by.__init__ar1r2cztjtj||_|jj |_t |}||dk(r td|dk(r-tjtj|d|_ytj|dtji|_y)Nrz)at least one ORDER BY element is requiredr _literal_as_text_role) rexpectrExpressionElementRoler&r'len TypeErrorr(r ClauseList)r.r&r(_lobs r/r0zaggregate_order_by.__init__hs &/%5%5  ' '&  KK$$ 8} 19GH H QY%,,++Xa[DM%//161L1LDMr2Nc|Sr+r,)r.againsts r/ self_groupzaggregate_order_by.self_groups  r2c 2|j|jfSr+r&r()r.kwargss r/ get_childrenzaggregate_order_by.get_childrens{{DMM))r2c d||jfi||_||jfi||_yr+r@)r.clonekws r/_copy_internalsz"aggregate_order_by._copy_internalss.DKK.2. dmm2r2 r2c\|jj|jjzSr+)r& _from_objectsr()r.s r/rHz aggregate_order_by._from_objectss!{{((4==+F+FFFr2)r&zColumnElement[_T]r(_ColumnExpressionArgument[Any])r&z_ColumnExpressionArgument[_T]r(rIr+)r=zOptional[OperatorType]returnr)rArrJzIterable[ClauseElement])rDrrErrJNone)rJzList[FromClause])__name__ __module__ __qualname____doc____visit_name__stringify_dialectrdp_clauseelementdp_typer)__annotations__rr0r>rBr_clonerFpropertyrHr,r2r/r$r$,s"H*N$ $556 "**+ &7783/  ! 2   - 2  -2015-  *+3//3'3AD3 3 GGr2r$c|eZdZdZdZdZdZdZejddd d d Z fd Z d d Z xZ S)ExcludeConstraintzA table-level EXCLUDE constraint. Defines an EXCLUDE constraint as described in the `PostgreSQL documentation`__. __ https://www.postgresql.org/docs/current/static/sql-createtable.html#SQL-CREATETABLE-EXCLUDE exclude_constraintNFr%wherez:class:`.ExcludeConstraint`z$:paramref:`.ExcludeConstraint.where`cg}g}i|_t|\}}ttjtj ||D]R\\}}} } } | |j | | |jn| } | | |j| <|j || | fT||_tj|g||jd|jd|jdd|jdd|_ |jd} | )tjtj| |_|jd i|_y) a Create an :class:`.ExcludeConstraint` object. E.g.:: const = ExcludeConstraint( (Column("period"), "&&"), (Column("group"), "="), where=(Column("group") != "some group"), ops={"group": "my_operator_class"}, ) The constraint is normally embedded into the :class:`_schema.Table` construct directly, or added later using :meth:`.append_constraint`:: some_table = Table( "some_table", metadata, Column("id", Integer, primary_key=True), Column("period", TSRANGE()), Column("group", String), ) some_table.append_constraint( ExcludeConstraint( (some_table.c.period, "&&"), (some_table.c.group, "="), where=some_table.c.group != "some group", name="some_table_excl_const", ops={"group": "my_operator_class"}, ) ) The exclude constraint defined in this example requires the ``btree_gist`` extension, that can be created using the command ``CREATE EXTENSION btree_gist;``. :param \*elements: A sequence of two tuples of the form ``(column, operator)`` where "column" is either a :class:`_schema.Column` object, or a SQL expression element (e.g. ``func.int8range(table.from, table.to)``) or the name of a column as string, and "operator" is a string containing the operator to use (e.g. `"&&"` or `"="`). In order to specify a column name when a :class:`_schema.Column` object is not available, while ensuring that any necessary quoting rules take effect, an ad-hoc :class:`_schema.Column` or :func:`_expression.column` object should be used. The ``column`` may also be a string SQL expression when passed as :func:`_expression.literal_column` or :func:`_expression.text` :param name: Optional, the in-database name of this constraint. :param deferrable: Optional bool. If set, emit DEFERRABLE or NOT DEFERRABLE when issuing DDL for this constraint. :param initially: Optional string. If set, emit INITIALLY when issuing DDL for this constraint. :param using: Optional string. If set, emit USING when issuing DDL for this constraint. Defaults to 'gist'. :param where: Optional SQL expression construct or literal SQL string. If set, emit WHERE when issuing DDL for this constraint. :param ops: Optional dictionary. Used to define operator classes for the elements; works the same way as that of the :ref:`postgresql_ops ` parameter specified to the :class:`_schema.Index` construct. .. versionadded:: 1.3.21 .. seealso:: :ref:`postgresql_operator_classes` - general description of how PostgreSQL operator classes are specified. Nname deferrable initially)r\r]r^usinggistrZops) operatorszipr expect_col_expression_collectionrDDLConstraintColumnRoleappendr\ _render_exprsrr0getr_r6StatementOptionRolerZra)r.rrEcolumns render_exprs expressionsrbexprcolumnstrname add_elementoperatorr\rZs r/r0zExcludeConstraint.__init__sQB !$h Y>A  6 6--{   ?  8 : 0T67K( &{+"("46;;'D'/t$   tX 6 7 8"*"++   vvl+ff[)  VVGV, w  "))%*C*CUKDJ66%$r2c t|||jDcgc]*\}}}t|ts|n|j |||f,c}}}|_ycc}}}wr+)super _set_parentrg isinstancestrc)r.tablerErmr\rq __class__s r/rtzExcludeConstraint._set_parent+sc E"*.););   %dH 'tS1uwwt}   s/Ac r|jDcgc])\}}}tj||j||f+}}}}|j||j |j |j|j|jd}|jj|j|Scc}}}w)N)r\r]r^rZr_) rgr_copy_expressionparentryr\r]r^rZr_dispatch_update)r. target_tablerErm_rqrrws r/_copyzExcludeConstraint._copy7s &*%7%7   "a''dkk<H    DNN nn****   4==)! s.B2)rzTuple[_DDLColumnArgument, str]rErrJrKr+)rLrMrNrOrPrZ inherit_cachecreate_drop_stringify_dialectr_document_text_coercionr0rtr __classcell__rys@r/rXrXso*N EM$0!%X%%%. A%7A%?BA% A%  A%F  r2rXcRt|d<tjj|i|S)zPostgreSQL-specific form of :class:`_functions.array_agg`, ensures return type is :class:`_postgresql.ARRAY` and not the plain :class:`_types.ARRAY`, unless an explicit ``type_`` is passed. _default_array_type)rrfunc array_agg)argrEs r/rrKs)!&B >> # #S /B //r2c"eZdZdZfdZxZS) _regconfig_fnTc t|}t|dkDrTtjtj |j dt|dd|tj}|g}ng}|Dcgc]4}tjtj |t|dd|6}}t|,||zi|ycc}w)Nr rr\)r\apply_propagate_attrstype_r\r) listr8rr6rr7popgetattrr REGCONFIGrsr0)r.argsrA initial_argrw addtl_argsrys r/r0z_regconfig_fn.__init__YsDz t9q=#**++ T640&*oo K'-KK    ++T640&*     ;3?? s59C)rLrMrNrr0rrs@r/rrVsM@@r2rc,eZdZdZdZej Zy) to_tsvectoraThe PostgreSQL ``to_tsvector`` SQL function. This function applies automatic casting of the REGCONFIG argument to use the :class:`_postgresql.REGCONFIG` datatype automatically, and applies a return type of :class:`_postgresql.TSVECTOR`. Assuming the PostgreSQL dialect has been imported, either by invoking ``from sqlalchemy.dialects import postgresql``, or by creating a PostgreSQL engine using ``create_engine("postgresql...")``, :class:`_postgresql.to_tsvector` will be used automatically when invoking ``sqlalchemy.func.to_tsvector()``, ensuring the correct argument and return type handlers are used at compile and execution time. .. versionadded:: 2.0.0rc1 TN)rLrMrNrOrr TSVECTORr'r,r2r/rrss"M >>Dr2rc,eZdZdZdZej Zy) to_tsqueryaThe PostgreSQL ``to_tsquery`` SQL function. This function applies automatic casting of the REGCONFIG argument to use the :class:`_postgresql.REGCONFIG` datatype automatically, and applies a return type of :class:`_postgresql.TSQUERY`. Assuming the PostgreSQL dialect has been imported, either by invoking ``from sqlalchemy.dialects import postgresql``, or by creating a PostgreSQL engine using ``create_engine("postgresql...")``, :class:`_postgresql.to_tsquery` will be used automatically when invoking ``sqlalchemy.func.to_tsquery()``, ensuring the correct argument and return type handlers are used at compile and execution time. .. versionadded:: 2.0.0rc1 TNrLrMrNrOrr TSQUERYr'r,r2r/rr"M ==Dr2rc,eZdZdZdZej Zy)plainto_tsqueryaThe PostgreSQL ``plainto_tsquery`` SQL function. This function applies automatic casting of the REGCONFIG argument to use the :class:`_postgresql.REGCONFIG` datatype automatically, and applies a return type of :class:`_postgresql.TSQUERY`. Assuming the PostgreSQL dialect has been imported, either by invoking ``from sqlalchemy.dialects import postgresql``, or by creating a PostgreSQL engine using ``create_engine("postgresql...")``, :class:`_postgresql.plainto_tsquery` will be used automatically when invoking ``sqlalchemy.func.plainto_tsquery()``, ensuring the correct argument and return type handlers are used at compile and execution time. .. versionadded:: 2.0.0rc1 TNrr,r2r/rrrr2rc,eZdZdZdZej Zy)phraseto_tsqueryaThe PostgreSQL ``phraseto_tsquery`` SQL function. This function applies automatic casting of the REGCONFIG argument to use the :class:`_postgresql.REGCONFIG` datatype automatically, and applies a return type of :class:`_postgresql.TSQUERY`. Assuming the PostgreSQL dialect has been imported, either by invoking ``from sqlalchemy.dialects import postgresql``, or by creating a PostgreSQL engine using ``create_engine("postgresql...")``, :class:`_postgresql.phraseto_tsquery` will be used automatically when invoking ``sqlalchemy.func.phraseto_tsquery()``, ensuring the correct argument and return type handlers are used at compile and execution time. .. versionadded:: 2.0.0rc1 TNrr,r2r/rrrr2rc,eZdZdZdZej Zy)websearch_to_tsqueryaThe PostgreSQL ``websearch_to_tsquery`` SQL function. This function applies automatic casting of the REGCONFIG argument to use the :class:`_postgresql.REGCONFIG` datatype automatically, and applies a return type of :class:`_postgresql.TSQUERY`. Assuming the PostgreSQL dialect has been imported, either by invoking ``from sqlalchemy.dialects import postgresql``, or by creating a PostgreSQL engine using ``create_engine("postgresql...")``, :class:`_postgresql.websearch_to_tsquery` will be used automatically when invoking ``sqlalchemy.func.websearch_to_tsquery()``, ensuring the correct argument and return type handlers are used at compile and execution time. .. versionadded:: 2.0.0rc1 TNrr,r2r/rrrr2rc*eZdZdZdZeZfdZxZS) ts_headlineaThe PostgreSQL ``ts_headline`` SQL function. This function applies automatic casting of the REGCONFIG argument to use the :class:`_postgresql.REGCONFIG` datatype automatically, and applies a return type of :class:`_types.TEXT`. Assuming the PostgreSQL dialect has been imported, either by invoking ``from sqlalchemy.dialects import postgresql``, or by creating a PostgreSQL engine using ``create_engine("postgresql...")``, :class:`_postgresql.ts_headline` will be used automatically when invoking ``sqlalchemy.func.ts_headline()``, ensuring the correct argument and return type handlers are used at compile and execution time. .. versionadded:: 2.0.0rc1 Tc 0t|}t|dkrd}nKt|dtjr,|dj j tjurd}nd}|rTtjtj|jd|t|ddtj}|g}ng}|Dcgc]4}tjtj|t|dd|6}}t!|D||zi|ycc}w) NFr Trr\)rr\rr)rr8rurrr'_type_affinityr rrr6rr7rrrrsr0)r.rrA has_regconfigrrwrrys r/r0zts_headline.__init__sDz t9q=!M tAw 6 6 7Q ++u}}<"M M #**++ &*T640oo K'-KK    ++T640&*     ;3?? s9D) rLrMrNrOrrr'r0rrs@r/rrs "M D&@&@r2rN)4 __future__rtypingrrrrrr r r r arrayrsqlrrrrrr sql.schemar sql.sqltypesr sql.visitorsr sql._typingrr sql.elementsrr sql.operatorsrsql.selectablerrr r!r$rXrGenericFunctionrrrrrrrr,r2r/rs# 4 -81---,26 TcG11"5cGLv2vr0@I--b1@:-,,m,},=,;@-;@r2