U
|
¸ý°d´@ ã @ st d dl mZ d dlmZ d dlmZ d dlmZ ddlmZ ddlm Z dd l
|
m Z dd
|
l
|
m Z dd l
|
m Z dd l
|
mZ dd l
|
mZ ddl
|
mZ ddlmZ ddlmZ ddlmZ ededZerÐddlmZ G dd de jZG dd deZdd Z G dd d ej e Z G d d d e Z G d d d e Z G d!d" d"e Z!G d#d$ d$e Z"G d%d& d&e Z#G d'd( d(e Z$d)S )*é )Ú annotations)ÚAny)Ú TYPE_CHECKING)ÚTypeVaré )Útypes)ÚARRAYé )Ú coercions)Úelements)Ú
|
expression)Ú functions)Úroles)Úschema)ÚColumnCollectionConstraint)ÚTEXT)ÚInternalTraversalÚ_T)Úbound)Ú_TraverseInternalsTypec @ sr e Zd ZU dZd ZdZdejfdejfdejfgZ de
|
d< dd Z dd d Z d d Z ejfddZedd Zd
|
S )Úaggregate_order_byaà Represent 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;
|
|
.. versionchanged:: 1.2.13 - the ORDER BY argument may be multiple terms
|
|
.. seealso::
|
|
:class:`_functions.array_agg`
|
|
Ú
|
postgresqlÚtargetÚtypeÚorder_byr Ú_traverse_internalsc G sj t tj|¡| _| jj| _t|}|dkr4tdn2|dkrRt tj|d ¡| _nt j
|
|dtji| _d S )Nr z)at least one ORDER BY element is requiredr Z_literal_as_text_role) r
|
Úexpectr ÚExpressionElementRoler r ÚlenÚ TypeErrorr r Z
|
ClauseList)Úselfr r Z_lob© r! úYd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\sqlalchemy/dialects/postgresql/ext.pyÚ__init__J s
|
|
ÿÿÿzaggregate_order_by.__init__Nc C s | S ©Nr! )r Zagainstr! r! r" Ú
|
self_groupZ s z aggregate_order_by.self_groupc K s | j | jfS r$ ©r r )r Úkwargsr! r! r" Ú get_children] s z aggregate_order_by.get_childrenc K s$ || j f|| _ || jf|| _d S r$ r& )r ÚcloneÚkwr! r! r" Ú_copy_internals` s z"aggregate_order_by._copy_internalsc C s | j j| jj S r$ )r Ú _from_objectsr ©r r! r! r" r, d s z aggregate_order_by._from_objects)N)Ú__name__Ú
|
__module__Ú __qualname__Ú__doc__Ú__visit_name__Zstringify_dialectr Zdp_clauseelementZdp_typer Ú__annotations__r# r% r( r Z_cloner+ Úpropertyr, r! r! r! r" r s
|
ý
|
r c sP e Zd ZdZdZdZdZdZe ddd¡d d
|
Z
|
fd d Z dd dZ Z S )ÚExcludeConstraintzäA 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
|
|
Zexclude_constraintNFr Úwherez:class:`.ExcludeConstraint`z$:paramref:`.ExcludeConstraint.where`c O sú g }g }i | _ t| \}}tt tj|¡|D ]V\\}}} }
|
} |
|
dk rP| |
|
¡ |dk r^|jn| } | dk rt| | j | < | || | f¡ q.|| _t j
|
| f|| d¡| d¡| d¡d | dd¡| _ | d¡} | dk rèt tj| ¡| _| d i ¡| _dS )
|
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'}
|
)
|
)
|
|
: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 <value> when issuing DDL
|
for this constraint.
|
|
:param using:
|
Optional string. If set, emit USING <index_method> when issuing DDL
|
for this constraint. Defaults to 'gist'.
|
|
:param where:
|
Optional SQL expression construct or literal SQL string.
|
If set, emit WHERE <predicate> 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 <postgresql_operator_classes>`
|
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.
|
|
NÚnameÚ
|
deferrableÚ initially)r7 r8 r9 ÚusingZgistr6 Úops)Ú operatorsÚzipr
|
Z expect_col_expression_collectionr ZDDLConstraintColumnRoleÚappendr7 Ú _render_exprsr r# Úgetr: r ZStatementOptionRoler6 r; )r r r* ÚcolumnsZ render_exprsZ expressionsr< ÚexprÚcolumnZstrnameZ add_elementÚoperatorr7 r6 r! r! r" r# z s@ Z ÿü
|
|
ÿþû
|
zExcludeConstraint.__init__c s& t ¡ fdd| jD | _d S )Nc s0 g | ](\}}}t |ts |n j| ||fqS r! )Ú
|
isinstanceÚstrÚc)Ú.0rB r7 rD ©Útabler! r" Ú
|
<listcomp>þ s
|
üýz1ExcludeConstraint._set_parent.<locals>.<listcomp>)ÚsuperÚ _set_parentr? )r rJ r* ©Ú __class__rI r" rM û s
|
úz ExcludeConstraint._set_parentc sJ fdd j D } j| j j j j jd}|j j¡ |S )Nc s& g | ] \}}}t | j¡|fqS r! )r Z_copy_expressionÚparent)rH rB Ú_rD ©r Ú target_tabler! r" rK s ýþz+ExcludeConstraint._copy.<locals>.<listcomp>)r7 r8 r9 r6 r: ) r? rO r7 r8 r9 r6 r: ÚdispatchÚ_update)r rS r* r rG r! rR r" Ú_copy s ûúzExcludeConstraint._copy)N)r. r/ r0 r1 r2 r6 Ú inherit_cacheZ create_drop_stringify_dialectr Z_document_text_coercionr# rM rV Ú __classcell__r! r! rN r" r5 i s ý
|
| r5 c O s t |d< tjj| |S )zÏPostgreSQL-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.
|
|
Z_default_array_type)r r ÚfuncÚ array_agg)Úargr* r! r! r" rZ s rZ c s e Zd ZdZ fddZ ZS )Ú _regconfig_fnTc sn t |}t|dkrBtjtj| d¡t dd tj d}|g}ng } fdd|D }t
|
j || | d S )Nr r r7 )r7 Úapply_propagate_attrsÚtype_c s( g | ] }t jtj|t d d dqS ©r7 N)r7 r] ©r
|
r r r Úgetattr©rH rG r- r! r" rK 8 s ú
|
üz*_regconfig_fn.__init__.<locals>.<listcomp>) Úlistr r
|
r r r Úpopra r Ú REGCONFIGrL r# )r Úargsr' Ú initial_argÚ
|
addtl_argsrN r- r" r# ) s
|
û
|
ù z_regconfig_fn.__init__)r. r/ r0 rW r# rX r! r! rN r" r\ & s r\ c @ s e Zd ZdZdZejZdS )Ú to_tsvectoraÇ The 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)r. r/ r0 r1 rW r ZTSVECTORr r! r! r! r" ri D s ri c @ s e Zd ZdZdZejZdS )Ú
|
to_tsqueryaà The 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
|
|
TN©r. r/ r0 r1 rW r ÚTSQUERYr r! r! r! r" rj Z s rj c @ s e Zd ZdZdZejZdS )Úplainto_tsqueryaÒ The 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
|
|
TNrk r! r! r! r" rm p s rm c @ s e Zd ZdZdZejZdS )Úphraseto_tsqueryaÕ The 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
|
|
TNrk r! r! r! r" rn s rn c @ s e Zd ZdZdZejZdS )Úwebsearch_to_tsqueryaá The 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
|
|
TNrk r! r! r! r" ro s ro c s( e Zd ZdZdZeZ fddZ ZS )Ú ts_headlinea¾ The 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 s¤ t |}t|dk rd}n,t|d tjrB|d jjtjkrBd}nd}|rxt j
|
t j | d¡ t dd tjd}|g}ng } fdd |D }t j|| | d S )
|
Né Fr Tr r7 )r] r7 r^ c s( g | ] }t jtj|t d d dqS r_ r` rb r- r! r" rK ä s ú
|
üz(ts_headline.__init__.<locals>.<listcomp>)rc r rE r Ú ColumnElementr Z_type_affinityr rl r
|
r r r rd ra re rL r# )r rf r' Z has_regconfigrg rh rN r- r" r# Ç s. ÿþ
|
û
|
ù zts_headline.__init__) r. r/ r0 r1 rW r r r# rX r! r! rN r" rp ² s rp N)%Ú
|
__future__r Útypingr r r Ú r Úarrayr Zsqlr
|
r r r r r Z
|
sql.schemar Z sql.sqltypesr Z sql.visitorsr r r rr r r5 rZ ZGenericFunctionr\ ri rj rm rn ro rp r! r! r! r" Ú<module> s8 I 3
|
