o
    XJ@h#                     @  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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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&d,d"d#Z'G d$d! d!eZG d%d& d&eZ(G d'd( d(e(Z)G d)d* d*e(Z*d+S )-    )annotations)Any)List)Optional)Tuple)Union   )_OnConflictIndexElementsT)_OnConflictIndexWhereT)_OnConflictSetT)_OnConflictWhereT   )util)	coercions)roles)schema)_DMLTableArgument)_exclusive_against)_generative)ColumnCollection)ReadOnlyColumnCollectionInsert)ClauseElement)ColumnElement)KeyedColumnElement)
TextClause)alias)Self)r   inserttabler   returnr   c                 C  s   t | S )aW  Construct a sqlite-specific variant :class:`_sqlite.Insert`
    construct.

    .. container:: inherited_member

        The :func:`sqlalchemy.dialects.sqlite.insert` function creates
        a :class:`sqlalchemy.dialects.sqlite.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:`_sqlite.Insert` construct includes additional methods
    :meth:`_sqlite.Insert.on_conflict_do_update`,
    :meth:`_sqlite.Insert.on_conflict_do_nothing`.

    r   )r     r"   Q/var/www/html/venv/lib/python3.10/site-packages/sqlalchemy/dialects/sqlite/dml.pyr   '   s   r   c                   @  sl   e Zd ZdZdZdZejdddZe	ddd	id
Z
ee
				ddddZee
		d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`

    sqliteFr!   6ReadOnlyColumnCollection[str, KeyedColumnElement[Any]]c                 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   r    columns)selfr"   r"   r#   r&   N   s   zInsert.excluded_post_values_clausezCThis Insert construct already has an ON CONFLICT clause established)ZmsgsNindex_elementsr	   index_wherer
   set_r   wherer   r   c                 C  s   t ||||| _| 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. An expression object representing 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).

        )OnConflictDoUpdater*   r)   r+   r,   r-   r.   r"   r"   r#   on_conflict_do_updaten   s   0zInsert.on_conflict_do_updatec                 C  s   t ||| _| 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.

        )OnConflictDoNothingr*   r)   r+   r,   r"   r"   r#   on_conflict_do_nothing   s   zInsert.on_conflict_do_nothing)r!   r%   NNNN)
r+   r	   r,   r
   r-   r   r.   r   r!   r   NN)r+   r	   r,   r
   r!   r   )__name__
__module____qualname____doc__stringify_dialectZinherit_cacher   Zmemoized_propertyr&   r   Z_on_conflict_exclusiver   r1   r4   r"   r"   r"   r#   r   ;   s.    3c                   @  s2   e Zd ZU dZded< ded< 		ddddZdS )OnConflictClauser$   z.Optional[List[Union[str, schema.Column[Any]]]]inferred_target_elementsz/Optional[Union[ColumnElement[Any], TextClause]]inferred_target_whereclauseNr+   r	   r,   r
   c                 C  sN   |d urdd |D | _ |d urttj|| _d S d | _d S d  | _ | _d S )Nc                 S  s   g | ]	}t tj|qS r"   )r   expectr   ZDDLConstraintColumnRole).0columnr"   r"   r#   
<listcomp>   s    z-OnConflictClause.__init__.<locals>.<listcomp>)r=   r   r?   r   WhereHavingRoler>   r3   r"   r"   r#   __init__   s   	zOnConflictClause.__init__r6   )r+   r	   r,   r
   )r7   r8   r9   r;   __annotations__rD   r"   r"   r"   r#   r<      s   
 r<   c                   @  s   e Zd ZdZdS )r2   r4   N)r7   r8   r9   __visit_name__r"   r"   r"   r#   r2      s    r2   c                      s>   e Zd ZU dZded< ded< 				dd fddZ  ZS )r/   r1   z0List[Tuple[Union[schema.Column[Any], str], Any]]update_values_to_setzOptional[ColumnElement[Any]]update_whereclauseNr+   r	   r,   r
   r-   r   r.   r   c                   s~   t  j||d t|tr|stdnt|trt|}ntddd | D | _|d ur:t	t
j|| _d S 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   r?   r   ZDMLColumnRole)r@   keyvaluer"   r"   r#   rB      s    z/OnConflictDoUpdate.__init__.<locals>.<listcomp>)superrD   
isinstancedict
ValueErrorr   itemsrG   r   r?   r   rC   rH   r0   	__class__r"   r#   rD      s*   


zOnConflictDoUpdate.__init__r5   )r+   r	   r,   r
   r-   r   r.   r   )r7   r8   r9   rF   rE   rD   __classcell__r"   r"   rP   r#   r/      s   
 r/   N)r    r   r!   r   )+
__future__r   typingr   r   r   r   r   Z_typingr	   r
   r   r    r   sqlr   r   r   Zsql._typingr   Zsql.baser   r   r   r   Zsql.dmlr   ZStandardInsertZsql.elementsr   r   r   r   Zsql.expressionr   Zutil.typingr   __all__r   r<   r2   r/   r"   r"   r"   r#   <module>   sB   
  