o
    [hn                     @   s   d dl mZ ddl mZ ddlmZ ddlmZ ddlmZ	 ddl
mZ ddlmZ dd	lmZ d
ZG dd de	ZeeddZG dd deZG dd deZG dd deZdS )   )ext   )util)schema)_generative)Insert)ClauseElement)alias)public_factory)r   insertc                   @   sF   e Zd ZdZejdd Ze					d	ddZe	d
ddZ	dS )r   a  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

    c                 C   s   t | jddjS )a  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.

        .. seealso::

            :ref:`postgresql_insert_on_conflict` - example of how
            to use :attr:`_expression.Insert.excluded`

        excluded)name)r	   tablecolumns)self r   /home/ubuntu/experiments/live_experiments/Pythonexperiments/Otree/venv/lib/python3.10/site-packages/sqlalchemy/dialects/postgresql/dml.pyr   !   s   zInsert.excludedNc                 C   s   t |||||| _| 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_clauser   
constraintindex_elementsindex_whereZset_wherer   r   r   on_conflict_do_update1   s   :
zInsert.on_conflict_do_updatec                 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.

         .. versionadded:: 1.1

        .. seealso::

            :ref:`postgresql_insert_on_conflict`

        )OnConflictDoNothingr   r   r   r   r   r   r   r   on_conflict_do_nothingp   s   zInsert.on_conflict_do_nothingNNNNNNNN)
__name__
__module____qualname____doc__r   Zmemoized_propertyr   r   r   r   r   r   r   r   r      s    
>r   z.dialects.postgresql.insertz.dialects.postgresql.Insertc                   @   s   e Zd ZdddZdS )OnConflictClauseNc                 C   s  |d urt |tjst |tjtjtjfrt|dp|}|d urb|d ur(t	dt |tjr8|| _
d | _d | _n*t |tjrJ|j}|jd d}nt |tjrW|j}|j}n|j}|jd d}|d urqd | _
|| _|| _d S |d u rd  | _
 | _| _d S d S )Nr   z8'constraint' and 'index_elements' are mutually exclusive
postgresqlr   )
isinstancer   string_typesr   Index
Constraintr   ZExcludeConstraintgetattr
ValueErrorconstraint_targetinferred_target_elementsZinferred_target_whereclauseZexpressionsZdialect_optionsgetr   r   r   r   r   r   __init__   sJ   


zOnConflictClause.__init__r   )r    r!   r"   r/   r   r   r   r   r$      s    r$   c                   @   s   e Zd ZdZdS )r   r   N)r    r!   r"   __visit_name__r   r   r   r   r      s    r   c                       s,   e Zd ZdZ					d fdd	Z  ZS )r   r   Nc                    sf   t t| j|||d | jd u r| jd u rtdt|tr |s$tddd | D | _	|| _
d S )N)r   r   r   zVEither constraint or index_elements, but not both, must be specified unless DO NOTHINGz,set parameter must be a non-empty dictionaryc                 S   s   g | ]\}}||fqS r   r   ).0keyvaluer   r   r   
<listcomp>   s    z/OnConflictDoUpdate.__init__.<locals>.<listcomp>)superr   r/   r-   r,   r+   r&   dictitemsZupdate_values_to_setZupdate_whereclauser   	__class__r   r   r/      s    



zOnConflictDoUpdate.__init__r   )r    r!   r"   r0   r/   __classcell__r   r   r8   r   r      s    r   N) r   r   sqlr   Zsql.baser   Zsql.dmlr   ZStandardInsertZsql.elementsr   Zsql.expressionr	   Zutil.langhelpersr
   __all__r   r$   r   r   r   r   r   r   <module>   s    ,