관리-도구
편집 파일: dml.cpython-38.pyc
U -?�f�+ � @ sl d dl mZ d dlmZ d dlmZ ddlmZ ddlmZ ddlm Z dd lm Z dd lm Z dd lm Z d d lm Z d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dl m Z d dl m Z d dl m Z d dl!m"Z" dZ#d d d �d d �Z$G d!d � d e�ZG d"d#� d#e �Z%G d$d%� d%e%�Z&G d&d'� d'e%�Z'd(S ))� )� annotations)�Any)�Optional� )�ext� )�_OnConflictConstraintT)�_OnConflictIndexElementsT)�_OnConflictIndexWhereT)�_OnConflictSetT)�_OnConflictWhereT� )�util)� coercions)�roles)�schema)�_DMLTableArgument)�_exclusive_against)� _generative)�ColumnCollection)�ReadOnlyColumnCollection��Insert)� ClauseElement)�KeyedColumnElement)�alias)�Self)r �insertr r )�table�returnc C s t | �S )as Construct a PostgreSQL-specific variant :class:`_postgresql.Insert` construct. .. container:: inherited_member The :func:`sqlalchemy.dialects.postgresql.insert` function creates a :class:`sqlalchemy.dialects.postgresql.Insert`. This class is based on the dialect-agnostic :class:`_sql.Insert` construct which may be constructed using the :func:`_sql.insert` function in SQLAlchemy Core. The :class:`_postgresql.Insert` construct includes additional methods :meth:`_postgresql.Insert.on_conflict_do_update`, :meth:`_postgresql.Insert.on_conflict_do_nothing`. r )r � r �R/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/dialects/postgresql/dml.pyr % s r c @ s| e Zd ZdZdZdZejdd�dd��Ze ddd id �Z e e dd d ddddd�dd���Z e e dd d ddd�dd���Z d S )r z�PostgreSQL-specific implementation of INSERT. Adds methods for PG-specific syntaxes such as ON CONFLICT. The :class:`_postgresql.Insert` object is created using the :func:`sqlalchemy.dialects.postgresql.insert` function. � postgresqlFz6ReadOnlyColumnCollection[str, KeyedColumnElement[Any]])r c C s t | jdd�jS )ak Provide the ``excluded`` namespace for an ON CONFLICT statement PG's ON CONFLICT clause allows reference to the row that would be inserted, known as ``excluded``. This attribute provides all columns in this row to be referenceable. .. tip:: The :attr:`_postgresql.Insert.excluded` attribute is an instance of :class:`_expression.ColumnCollection`, which provides an interface the same as that of the :attr:`_schema.Table.c` collection described at :ref:`metadata_tables_and_columns`. With this collection, ordinary names are accessible like attributes (e.g. ``stmt.excluded.some_column``), but special names and dictionary method names should be accessed using indexed access, such as ``stmt.excluded["column name"]`` or ``stmt.excluded["values"]``. See the docstring for :class:`_expression.ColumnCollection` for further examples. .. seealso:: :ref:`postgresql_insert_on_conflict` - example of how to use :attr:`_expression.Insert.excluded` �excluded)�name)r r �columns)�selfr r r! r# F s zInsert.excluded�_post_values_clausezCThis Insert construct already has an ON CONFLICT clause established)ZmsgsNr r r r r r )� constraint�index_elements� index_where�set_�wherer c C s t |||||�| _| S )aS Specifies a DO UPDATE SET action for ON CONFLICT clause. Either the ``constraint`` or ``index_elements`` argument is required, but only one of these can be specified. :param constraint: The name of a unique or exclusion constraint on the table, or the constraint object itself if it has a .name attribute. :param index_elements: A sequence consisting of string column names, :class:`_schema.Column` objects, or other column expression objects that will be used to infer a target index. :param index_where: Additional WHERE criterion that can be used to infer a conditional target index. :param set\_: A dictionary or other mapping object where the keys are either names of columns in the target table, or :class:`_schema.Column` objects or other ORM-mapped columns matching that of the target table, and expressions or literals as values, specifying the ``SET`` actions to take. .. versionadded:: 1.4 The :paramref:`_postgresql.Insert.on_conflict_do_update.set_` parameter supports :class:`_schema.Column` objects from the target :class:`_schema.Table` as keys. .. warning:: This dictionary does **not** take into account Python-specified default UPDATE values or generation functions, e.g. those specified using :paramref:`_schema.Column.onupdate`. These values will not be exercised for an ON CONFLICT style of UPDATE, unless they are manually specified in the :paramref:`.Insert.on_conflict_do_update.set_` dictionary. :param where: Optional argument. If present, can be a literal SQL string or an acceptable expression for a ``WHERE`` clause that restricts the rows affected by ``DO UPDATE SET``. Rows not meeting the ``WHERE`` condition will not be updated (effectively a ``DO NOTHING`` for those rows). .. seealso:: :ref:`postgresql_insert_on_conflict` )�OnConflictDoUpdater' �r& r( r) r* r+ r, r r r! �on_conflict_do_updatek s = �z Insert.on_conflict_do_update)r( r) r* r c C s t |||�| _| S )a Specifies a DO NOTHING action for ON CONFLICT clause. The ``constraint`` and ``index_elements`` arguments are optional, but only one of these can be specified. :param constraint: The name of a unique or exclusion constraint on the table, or the constraint object itself if it has a .name attribute. :param index_elements: A sequence consisting of string column names, :class:`_schema.Column` objects, or other column expression objects that will be used to infer a target index. :param index_where: Additional WHERE criterion that can be used to infer a conditional target index. .. seealso:: :ref:`postgresql_insert_on_conflict` )�OnConflictDoNothingr' �r& r( r) r* r r r! �on_conflict_do_nothing� s �z Insert.on_conflict_do_nothing)NNNNN)NNN)�__name__� __module__� __qualname__�__doc__�stringify_dialectZ inherit_cacher Zmemoized_propertyr# r Z_on_conflict_exclusiver r/ r2 r r r r! r 9 s2 �� � @ �c @ s> e Zd ZU dZded< ded< ded< d d ddd �d d �ZdS )�OnConflictClauser" z Optional[str]�constraint_targetr �inferred_target_elementsr �inferred_target_whereclauseNr �r( r) r* c C s� |d k r2t |t�s2t |tjtjf�r2t|d�p0|}|d k r�|d k rJtd��t |t�rh|| _d | _ d | _ nTt |tj �r�|j }|j d �d�}n0t |tj�r�|j}|j}n|j}|j d �d�}|d k r�d | _|| _ || _ n|d kr�d | _ | _ | _ d S )Nr$ z8'constraint' and 'index_elements' are mutually exclusiver" r, )� isinstance�strr � Constraintr ZExcludeConstraint�getattr� ValueErrorr9 r: r; �IndexZ expressionsZdialect_options�getr% r, r1 r r r! �__init__� sH �� � �� �zOnConflictClause.__init__)NNN)r3 r4 r5 r7 �__annotations__rD r r r r! r8 � s �r8 c @ s e Zd ZdZdS )r0 r2 N)r3 r4 r5 �__visit_name__r r r r! r0 s r0 c s0 e Zd ZdZd dddddd�� fd d � Z� ZS ) r- r/ Nr r r r r )r( r) r* r+ r, c s� t � j|||d� | jd kr.| jd kr.td��t|t�rF|sbtd��n t|t�rZt|�}ntd��dd� |�� D �| _ || _ d S )Nr< zVEither constraint or index_elements, but not both, must be specified unless DO NOTHINGz*set parameter dictionary must not be emptyzqset parameter must be a non-empty dictionary or a ColumnCollection such as the `.c.` collection of a Table objectc S s" g | ]\}}t �tj|�|f�qS r )r �expectr Z DMLColumnRole)�.0�key�valuer r r! � <listcomp>2 s �z/OnConflictDoUpdate.__init__.<locals>.<listcomp>) �superrD r: r9 rA r= �dictr �itemsZupdate_values_to_setZupdate_whereclauser. �� __class__r r! rD s0 ���� ��zOnConflictDoUpdate.__init__)NNNNN)r3 r4 r5 rF rD � __classcell__r r rO r! r- s �r- N)(� __future__r �typingr r � r Z_typingr r r r r r �sqlr r r Z sql._typingr Zsql.baser r r r Zsql.dmlr ZStandardInsertZ sql.elementsr r Zsql.expressionr Z util.typingr �__all__r r8 r0 r- r r r r! �<module> s: 6