B \@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:`.array_agg` cGsXt||_t|}|dkr&tdn.|dkr@t|d|_ntj|dtji|_dS)Nz)at least one ORDER BY element is requiredr_literal_as_text)rZ_literal_as_bindstargetlen TypeErrororder_byZ ClauseList)selfr rZ_lobrAlib/python3.7/site-packages/sqlalchemy/dialects/postgresql/ext.py__init__4s  zaggregate_order_by.__init__NcCs|S)Nr)rZagainstrrr self_groupAszaggregate_order_by.self_groupcKs |j|jfS)N)r r)rkwargsrrr get_childrenDszaggregate_order_by.get_childrencKs$||jf||_||jf||_dS)N)r r)rZclonekwrrr_copy_internalsGsz"aggregate_order_by._copy_internalscCs|jj|jjS)N)r _from_objectsr)rrrrrKsz aggregate_order_by._from_objects)N) __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|\}}xtt|||D]`\\}}} } } | dk rN|| |dk r\|jn| } | dk rr| |j| <t|}||| | fq,W||_tj |f|| d| d| dd| dd|_ | d} | dk rtj | d d |_ dS) a Create an :class:`.ExcludeConstraint` object. E.g.:: const = ExcludeConstraint( (Column('period'), '&&'), (Column('group'), '='), where=(Column('group') != 'some group') ) The constraint is normally embedded into the :class:`.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' ) ) :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:`.Column` object, and "operator" is a string containing the operator to use. In order to specify a column name when a :class:`.Column` object is not available, while ensuring that any necessary quoting rules take effect, an ad-hoc :class:`.Column` or :func:`.sql.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. Nname deferrable initially)r!r"r#usingZgistr T)Zallow_coercion_to_text) operatorszipZ"_extract_col_expression_collectionappendr!rZ_literal_as_columnZ _render_exprsrrgetr$r r )rrrcolumnsZ render_exprsZ expressionsr%exprcolumnZstrnameZ add_elementoperatorr!r rrrr^s2D     zExcludeConstraint.__init__c sLfddjD}j|jjjjjd}|j j|S)Ncsg|]}|j|fqSr)r%).0col)rrr sz*ExcludeConstraint.copy..)r!r"r#r r$) r)keys __class__r!r"r#r r$dispatchZ_update)rrrcr)rrcopys zExcludeConstraint.copy) rrrrrr rZ_document_text_coercionrr4rrrrrPsgrcOst|d<tjj||S)zPostgreSQL-specific form of :class:`.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)argrrrrr6s r6N) ZarrayrZsqlrrrZ sql.schemarZ ColumnElementrrr6rrrrs     A