o Th+@sfddlmZddlmZddlmZddlmZddlmZddlm Z dd lm Z dd lm Z dd lm Z d d lm Z d dlmZd dlmZd dlmZd dlmZd dlmZd dlmZd dlmZd dlmZd dlmZd dlmZd dlmZd dlm Z d dl!m"Z"dZ#d*d d!Z$Gd"ddeZGd#d$d$eZ%Gd%d&d&e%Z&Gd'd(d(e%Z'd)S)+) annotations)Any)Optional)ext)_OnConflictConstraintT)_OnConflictIndexElementsT)_OnConflictIndexWhereT)_OnConflictSetT)_OnConflictWhereT)util) coercions)roles)schema)_DMLTableArgument)_exclusive_against) _generative)ColumnCollection)ReadOnlyColumnCollectionInsert) ClauseElement)KeyedColumnElement)alias)Self)rinserttablerreturnrcCst|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)rr b/home/air/segue/gpt/backup/venv/lib/python3.10/site-packages/sqlalchemy/dialects/postgresql/dml.pyr%src@speZdZdZdZdZejdddZe ddd id Z e e ddddZ e e ddddZ 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. postgresqlFr6ReadOnlyColumnCollection[str, KeyedColumnElement[Any]]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)rrcolumns)selfr r r!r$FszInsert.excluded_post_values_clausezCThis Insert construct already has an ON CONFLICT clause established)msgsN constraintrindex_elementsr index_wherer set_r wherer rcCst||||||_|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_updateks= zInsert.on_conflict_do_updatecCst||||_|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)rr#NNNNN) r*rr+r r,r r-r r.r rrNNN)r*rr+r r,r rr)__name__ __module__ __qualname____doc__stringify_dialect inherit_cachermemoized_propertyr$r_on_conflict_exclusiverr1r4r r r r!r9s2  @c@s<eZdZUdZded<ded<ded<   ddd dZdS)OnConflictClauser"z Optional[str]constraint_targetr inferred_target_elementsr inferred_target_whereclauseNr*rr+r,cCs|durt|tst|tjtjfrt|dp|}|dur^|dur%tdt|tr4||_d|_ d|_ n*t|tj rF|j }|j dd}nt|tjrS|j}|j}n |j}|j dd}|durmd|_||_ ||_ dS|dur|d|_|_ |_ dSdS)Nr%z8'constraint' and 'index_elements' are mutually exclusiver"r.) isinstancestrr ConstraintrExcludeConstraintgetattr ValueErrorr@rArBIndex expressionsdialect_optionsgetr&r.r3r r r!__init__sH         zOnConflictClause.__init__r6)r*rr+r r,r )r7r8r9r;__annotations__rMr r r r!r?s r?c@seZdZdZdS)r2r4N)r7r8r9__visit_name__r r r r!r2 sr2cs.eZdZdZ     ddfd d ZZS)r/r1Nr*rr+r r,r r-r r.r cstj|||d|jdur|jdurtdt|tr#|s"tdnt|tr-t|}ntddd|D|_ ||_ dS)N)r*r+r,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 objectcSs"g|] \}}ttj||fqSr )rexpectr DMLColumnRole).0keyvaluer r r! 2sz/OnConflictDoUpdate.__init__..) superrMrAr@rHrCdictritemsupdate_values_to_setupdate_whereclauser0 __class__r r!rMs.      zOnConflictDoUpdate.__init__r5) r*rr+r r,r r-r r.r )r7r8r9rOrM __classcell__r r r[r!r/ sr/N)rrrr)( __future__rtypingrrr_typingrr r r r rsqlrrr sql._typingrsql.baserrrrsql.dmlrStandardInsert sql.elementsrrsql.expressionr util.typingr__all__rr?r2r/r r r r!s<                        6