U
    ÌZjÑ  ã                   @   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 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 )é   )Úutil)Ú	coercions)Úroles)Ú_exclusive_against)Ú_generative)ÚColumnCollection)ÚInsert)ÚClauseElement)Úalias)Úpublic_factory)r   Úinsertc                   @   sZ   e Zd ZdZdZdZejdd„ ƒZe	dddid�Z
ee
dd
d„ƒƒZee
ddd„ƒƒZd	S )r   a?  SQLite-specific implementation of INSERT.

    Adds methods for SQLite-specific syntaxes such as ON CONFLICT.

    The :class:`_sqlite.Insert` object is created using the
    :func:`sqlalchemy.dialects.sqlite.insert` function.

    .. versionadded:: 1.4

    .. seealso::

        :ref:`sqlite_on_conflict_insert`

    ÚsqliteFc                 C   s   t | jdd�jS )aÚ  Provide the ``excluded`` namespace for an ON CONFLICT statement

        SQLite'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:`_sqlite.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.

        Úexcluded)Úname)r
   ÚtableÚcolumns)Úself© r   úa/var/www/html/TRUCKING_PROJECT/venv/lib/python3.8/site-packages/sqlalchemy/dialects/sqlite/dml.pyr   )   s    zInsert.excludedÚ_post_values_clausezCThis Insert construct already has an ON CONFLICT clause established)ZmsgsNc                 C   s   t ||||ƒ| _dS )aö  
        Specifies a DO UPDATE SET action for ON CONFLICT clause.

        :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 or unique constraint.

        :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:`_sqlite.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).

        N)ÚOnConflictDoUpdater   ©r   Úindex_elementsÚindex_whereÚset_Úwherer   r   r   Úon_conflict_do_updateG   s    1   ÿzInsert.on_conflict_do_updatec                 C   s   t ||ƒ| _dS )aµ  
        Specifies a DO NOTHING action for ON CONFLICT clause.

        :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 or unique constraint.

        :param index_where:
         Additional WHERE criterion that can be used to infer a
         conditional target index.

        N)ÚOnConflictDoNothingr   ©r   r   r   r   r   r   Úon_conflict_do_nothing|   s     ÿzInsert.on_conflict_do_nothing)NNNN)NN)Ú__name__Ú
__module__Ú__qualname__Ú__doc__Ústringify_dialectZinherit_cacher   Zmemoized_propertyr   r   Z_on_conflict_exclusiver   r   r   r   r   r   r   r      s*   
 ÿþ    û3r   z.dialects.sqlite.insertz.dialects.sqlite.Insertc                   @   s   e Zd ZdZddd„ZdS )ÚOnConflictClauser   Nc                 C   s2   |d k	rd | _ || _|| _nd  | _  | _| _d S )N)Zconstraint_targetZinferred_target_elementsZinferred_target_whereclauser   r   r   r   Ú__init__š   s    þþzOnConflictClause.__init__)NN)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� t|tƒr,|sHtdƒ‚nt|tƒr@t|ƒ}ntdƒ‚dd„ | ¡ D ƒ| _|| _	d S )N)r   r   z*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 objectc                 S   s"   g | ]\}}t  tj|¡|f‘qS r   )r   Úexpectr   ZDMLColumnRole)Ú.0ÚkeyÚvaluer   r   r   Ú
<listcomp>Ä   s   ÿz/OnConflictDoUpdate.__init__.<locals>.<listcomp>)
Úsuperr   r&   Ú
isinstanceÚdictÚ
ValueErrorr   ÚitemsZupdate_values_to_setZupdate_whereclauser   ©Ú	__class__r   r   r&   ­   s     
þ



ÿþzOnConflictDoUpdate.__init__)NNNN)r    r!   r"   r'   r&   Ú__classcell__r   r   r2   r   r   ª   s       ûr   N)Ú r   Úsqlr   r   Zsql.baser   r   r   Zsql.dmlr   ZStandardInsertZsql.elementsr	   Zsql.expressionr
   Zutil.langhelpersr   Ú__all__r   r%   r   r   r   r   r   r   Ú<module>   s&   |  ÿ