관리-도구
편집 파일: ext.cpython-38.pyc
U -?�f�? � @ 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 eded�Zer�ddlmZ G dd� de j�ZG dd� de�Zdd� 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 ejfdd�Zedd� �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|�}|dkr4td��n2|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! �R/opt/hc_python/lib64/python3.8/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 d�Z � 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'} ) ) 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 <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@ ^ �� ��� zExcludeConstraint.__init__c s& t � �� � � fdd�| jD �| _d S )Nc s0 g | ](\}}}t |t�s |n� j| ||f�qS 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��|f�qS 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� fdd�Z� 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�� d��qS �r7 N)r7 r] �r r r r �getattr�rH rG r- r! r" rK ; 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 G 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 ] 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 s 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� fdd�Z� 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 tj�rB|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�� d��qS 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 �sqlr 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 7