U x^@sddlmZddlmZddlmZddlmZddlmZ ddl m Z ddl m Z dd lmZd ZGd d d e Zeed ZGddde ZGdddeZGdddeZdS))ext)util)schema) _generative)Insert) ClauseElement)alias)public_factory)rinsertc@s:eZdZdZejddZed ddZed ddZ dS) rzPostgreSQL-specific implementation of INSERT. Adds methods for PG-specific syntaxes such as ON CONFLICT. .. versionadded:: 1.1 cCst|jddjS)aProvide 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. .. seealso:: :ref:`postgresql_insert_on_conflict` - example of how to use :attr:`.Insert.excluded` excluded)name)r tablecolumns)selfrJ/tmp/pip-install-dq5v43_d/SQLAlchemy/sqlalchemy/dialects/postgresql/dml.pyr szInsert.excludedNcCst||||||_|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:`.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\_: Required argument. A dictionary or other mapping object with column names as keys and expressions or literals as values, specifying the ``SET`` actions to take. If the target :class:`.Column` specifies a ".key" attribute distinct from the column name, that key should be used. .. warning:: This dictionary does **not** take into account Python-specified default UPDATE values or generation functions, e.g. those specified using :paramref:`.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). .. versionadded:: 1.1 .. seealso:: :ref:`postgresql_insert_on_conflict` )OnConflictDoUpdate_post_values_clauser constraintindex_elements index_whereset_whererrron_conflict_do_update.s9zInsert.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:`.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. .. versionadded:: 1.1 .. seealso:: :ref:`postgresql_insert_on_conflict` )OnConflictDoNothingrrrrrrrron_conflict_do_nothingls zInsert.on_conflict_do_nothing)NNNNN)NNN) __name__ __module__ __qualname____doc__rZmemoized_propertyr rrrrrrrrs  =rz.dialects.postgresql.insertc@seZdZdddZdS)OnConflictClauseNcCs|dk r8t|tjs8t|tjtjtjfr8t|dp6|}|dk r|dk rPt dt|tjrp||_ d|_ d|_ nTt|tjr|j }|jdd}n0t|tjr|j}|j}n|j}|jdd}|dk rd|_ ||_ ||_ n|dkrd|_ |_ |_ dS)Nr z8'constraint' and 'index_elements' are mutually exclusiveZ postgresqlr) isinstancer string_typesrZIndexZ ConstraintrZExcludeConstraintgetattr ValueErrorconstraint_targetinferred_target_elementsZinferred_target_whereclauseZ expressionsZdialect_optionsgetrrrrrr__init__sJ     zOnConflictClause.__init__)NNN)rr r!r+rrrrr#sr#c@seZdZdZdS)rrN)rr r!__visit_name__rrrrrsrcs"eZdZdZdfdd ZZS)rrNcsftt|j|||d|jdkr2|jdkr2tdt|tr@|sHtddd|D|_ ||_ dS)N)rrrzVEither constraint or index_elements, but not both, must be specified unless DO NOTHINGz,set parameter must be a non-empty dictionarycSsg|]\}}||fqSrr).0keyvaluerrr sz/OnConflictDoUpdate.__init__..) superrr+r)r(r'r$dictitemsZupdate_values_to_setZupdate_whereclauser __class__rrr+s$ zOnConflictDoUpdate.__init__)NNNNN)rr r!r,r+ __classcell__rrr4rrsrN)rrZsqlrZsql.baserZsql.dmlrZStandardInsertZ sql.elementsrZsql.expressionr Zutil.langhelpersr __all__r r#rrrrrrs        { ,