o [hc@sjddlmZddlmZddlmZddlmZddlmZGdddejZ Gd d d eZ d d Z d S))ARRAY)elements) expression) functions)ColumnCollectionConstraintc@sHeZdZdZdZddZd ddZddZej fd d Z e d d Z dS)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:: 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:: SELECT string_agg(a, ',' ORDER BY a) FROM table; .. versionadded:: 1.1 .. versionchanged:: 1.2.13 - the ORDER BY argument may be multiple terms .. seealso:: :class:`_functions.array_agg` cGsXt||_t|}|dkrtd|dkr t|d|_dStj|dtji|_dS)Nz)at least one ORDER BY element is requiredr_literal_as_text)rZ_literal_as_bindstargetlen TypeErrororder_byZ ClauseList)selfr rZ_lobr/home/ubuntu/experiments/live_experiments/Pythonexperiments/Otree/venv/lib/python3.10/site-packages/sqlalchemy/dialects/postgresql/ext.py__init__4s  zaggregate_order_by.__init__NcCs|SNr)rZagainstrrr self_groupAszaggregate_order_by.self_groupcKs |j|jfSrr r)rkwargsrrr get_childrenDs zaggregate_order_by.get_childrencKs,||jfi||_||jfi||_dSrr)rclonekwrrr_copy_internalsGsz"aggregate_order_by._copy_internalscCs|jj|jjSr)r _from_objectsrrrrrrKsz aggregate_order_by._from_objectsr) __name__ __module__ __qualname____doc____visit_name__rrrrZ_clonerpropertyrrrrrrs" rc@s6eZdZdZdZdZedddddZd d Z dS) ExcludeConstraintzA table-level EXCLUDE constraint. Defines an EXCLUDE constraint as described in the `postgres documentation`__. __ http://www.postgresql.org/docs/9.0/static/sql-createtable.html#SQL-CREATETABLE-EXCLUDE Zexclude_constraintNwherez:class:`.ExcludeConstraint`z$:paramref:`.ExcludeConstraint.where`cOsg}g}i|_t|\}}t|||D]0\\}}} } } | dur&|| |dur-|jn| } | dur8| |j| <t|}||| | fq||_tj |g|R| d| d| dd| dd|_ | d} | durxtj | d d |_ | d i|_dS) a2 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'} ) ) :param \*elements: A sequence of two tuples of the form ``(column, operator)`` where "column" is a SQL expression element or a raw SQL string, most typically a :class:`_schema.Column` object, and "operator" is a string containing the operator to use. 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. :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'usingZgistr$T)Zallow_coercion_to_textops) operatorszipZ"_extract_col_expression_collectionappendr%rZ_literal_as_columnZ _render_exprsrrgetr(r r$r))rrrcolumnsZ render_exprsZ expressionsr*exprcolumnZstrnameZ add_elementoperatorr%r$rrrr^s>W      zExcludeConstraint.__init__c sLfddjD}j|jjjjjd}|j j|S)Ncsg|] }|j|fqSr)r*).0colrrr sz*ExcludeConstraint.copy..)r%r&r'r$r() r.keys __class__r%r&r'r$r(dispatch_update)rrrcrrrcopyszExcludeConstraint.copy) rrrr r!r$rZ_document_text_coercionrr:rrrrr#Ps   zr#cOst|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. .. versionadded:: 1.1 Z_default_array_type)rrfunc array_agg)argrrrrr<s r<N) arrayrsqlrrrZ sql.schemarZ ColumnElementrr#r<rrrrs     A