o [hn@sddlmZddlmZddlmZddlmZddlmZ ddl m Z ddl m Z dd lmZd ZGd d d e Zeed dZGddde ZGdddeZGdddeZdS))ext)util)schema) _generative)Insert) ClauseElement)alias)public_factory)rinsertc@sFeZdZdZejddZe     d ddZe d ddZ dS) ra 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. .. 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:`_expression.Insert.excluded` excluded)name)r tablecolumns)selfr/home/ubuntu/experiments/live_experiments/Pythonexperiments/Otree/venv/lib/python3.10/site-packages/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:`_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\_: 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:`_schema.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:`_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). .. versionadded:: 1.1 .. seealso:: :ref:`postgresql_insert_on_conflict` )OnConflictDoUpdate_post_values_clauser constraintindex_elements index_whereZset_whererrron_conflict_do_update1s: 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. .. versionadded:: 1.1 .. seealso:: :ref:`postgresql_insert_on_conflict` )OnConflictDoNothingrrrrrrrron_conflict_do_nothingpszInsert.on_conflict_do_nothingNNNNNNNN) __name__ __module__ __qualname____doc__rZmemoized_propertyr rrrrrrrrs  >rz.dialects.postgresql.insertz.dialects.postgresql.Insertc@seZdZdddZdS)OnConflictClauseNcCs|durt|tjst|tjtjtjfrt|dp|}|durb|dur(t dt|tjr8||_ d|_ d|_ n*t|tjrJ|j }|jdd}nt|tjrW|j}|j}n |j}|jdd}|durqd|_ ||_ ||_ dS|durd|_ |_ |_ dSdS)Nr z8'constraint' and 'index_elements' are mutually exclusive postgresqlr) isinstancer string_typesrIndex ConstraintrZExcludeConstraintgetattr ValueErrorconstraint_targetinferred_target_elementsZinferred_target_whereclauseZ expressionsZdialect_optionsgetrrrrrr__init__sJ      zOnConflictClause.__init__r)r r!r"r/rrrrr$sr$c@seZdZdZdS)rrN)r r!r"__visit_name__rrrrrsrcs,eZdZdZ     dfdd ZZS)rrNcsftt|j|||d|jdur|jdurtdt|tr |s$tddd|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 s z/OnConflictDoUpdate.__init__..) superrr/r-r,r+r&dictitemsZupdate_values_to_setZupdate_whereclauser __class__rrr/s    zOnConflictDoUpdate.__init__r)r r!r"r0r/ __classcell__rrr8rrsrN)rrsqlrZsql.baserZsql.dmlrZStandardInsertZ sql.elementsrZsql.expressionr Zutil.langhelpersr __all__r r$rrrrrrs        ,