U ovg^/@sddlmZddlmZddlmZddlmZddlmZddlmZddlm Z d d l m Z d d l m Z d d l m Z d d l mZd dl mZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm!Z!ddlm"Z"ddlm#Z#ddl$m%Z%ddl&m'Z'd Z(d!d"d#d$d%Z)Gd&d"d"eZGd'd(d(e Z*Gd)d*d*e*Z+Gd+d,d,e*Z,d-S).) annotations)Any)List)Optional)Tuple)Union)ext)_OnConflictConstraintT)_OnConflictIndexElementsT)_OnConflictIndexWhereT)_OnConflictSetT)_OnConflictWhereT)util) coercions)roles)schema)_DMLTableArgument)_exclusive_against) _generative)ColumnCollection)ReadOnlyColumnCollectionInsert) ClauseElement) ColumnElement)KeyedColumnElement) TextClause)alias)Self)rinsertrr)tablereturncCst|S)asConstruct 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%F/tmp/pip-unpacked-wheel-uqd9otp_/sqlalchemy/dialects/postgresql/dml.pyr"*sr"c @s|eZdZdZdZdZejddddZe ddd id Z e e dd d dddddddZ e e dd d dddddZ d S)rzPostgreSQL-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$cCst|jddjS)akProvide 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(KszInsert.excluded_post_values_clausezCThis Insert construct already has an ON CONFLICT clause established)ZmsgsNr r r rrr!) constraintindex_elements index_whereset_wherer$cCst||||||_|S)a$ 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. An expression object representing 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/r0r1r%r%r&on_conflict_do_updateps<zInsert.on_conflict_do_update)r-r.r/r$cCst||||_|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_nothings zInsert.on_conflict_do_nothing)NNNNN)NNN)__name__ __module__ __qualname____doc__stringify_dialectZ inherit_cacherZmemoized_propertyr(rZ_on_conflict_exclusiverr4r7r%r%r%r&r>s2 ?c@s>eZdZUdZded<ded<ded<dd d d d d dZdS)OnConflictClauser'z Optional[str]constraint_targetz.Optional[List[Union[str, schema.Column[Any]]]]inferred_target_elementsz/Optional[Union[ColumnElement[Any], TextClause]]inferred_target_whereclauseNr r r r-r.r/cCs.|dk r2t|ts2t|tjtjfr2t|dp0|}|dk r|dk rJtdt|trh||_d|_ d|_ nTt|tj r|j }|j dd}n0t|tjr|j}|j}n|j}|j dd}|dk rd|_dd|D|_ |dk rtt|tjrtjntj|nd|_ n|dkr*d|_|_ |_ dS)Nr)z8'constraint' and 'index_elements' are mutually exclusiver'r1cSsg|]}ttj|qSr%)rexpectrZDDLConstraintColumnRole).0columnr%r%r& sz-OnConflictClause.__init__..) isinstancestrrZ Constraintr ZExcludeConstraintgetattr ValueErrorr>r?r@ZIndexZ expressionsZdialect_optionsgetr*r1rrBrZStatementOptionRoleWhereHavingRoler6r%r%r&__init__s\            zOnConflictClause.__init__)NNN)r8r9r:r<__annotations__rLr%r%r%r&r=s r=c@seZdZdZdS)r5r7N)r8r9r:__visit_name__r%r%r%r&r5sr5csBeZdZUdZded<ded<dddd d d d fd d ZZS)r2r4z0List[Tuple[Union[schema.Column[Any], str], Any]]update_values_to_setzOptional[ColumnElement[Any]]update_whereclauseNr r r rr)r-r.r/r0r1cstj|||d|jdkr.|jdkr.tdt|trF|sbtdnt|trZt|}ntddd|D|_ |dk rt t j |nd|_dS)NrAzVEither 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 objectcSs"g|]\}}ttj||fqSr%)rrBrZ DMLColumnRole)rCkeyvaluer%r%r&rEKsz/OnConflictDoUpdate.__init__..)superrLr?r>rIrFdictritemsrOrrBrrKrPr3 __class__r%r&rL)s6    zOnConflictDoUpdate.__init__)NNNNN)r8r9r:rNrMrL __classcell__r%r%rVr&r2#s r2N)- __future__rtypingrrrrrr _typingr r r rrrZsqlrrrZ sql._typingrZsql.baserrrrZsql.dmlrZStandardInsertZ sql.elementsrrrrZsql.expressionr Z util.typingr!__all__r"r=r5r2r%r%r%r&sD                            H