
    g#                       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Z' G d  de      Z G d! d"e      Z( G d# d$e(      Z) G d% d&e(      Z*y'))    )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   insertr   c                    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   )tables    ^/var/www/api/v1/venv_getwork_v1/lib/python3.12/site-packages/sqlalchemy/dialects/sqlite/dml.pyr    r    '   s    " %=    c                      e Zd ZdZdZdZej                  	 	 dd       Z e	dddi      Z
ee
	 	 	 	 d	 	 	 	 	 	 	 	 	 dd	              Zee
	 	 d	 	 	 	 	 dd
              Zy)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                D    t        | j                  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)selfs    r#   r(   zInsert.excludedN   s    , TZZj1999r$   _post_values_clausezCThis Insert construct already has an ON CONFLICT clause established)msgsNc                ,    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+   index_elementsindex_whereset_wheres        r#   on_conflict_do_updatezInsert.on_conflict_do_updaten   s"    ` $6Ku$
  r$   c                (    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+   r0   r1   s      r#   on_conflict_do_nothingzInsert.on_conflict_do_nothing   s    * $7K$
  r$   )returnz6ReadOnlyColumnCollection[str, KeyedColumnElement[Any]]NNNN)
r0   r
   r1   r   r2   r   r3   r   r8   r   NN)r0   r
   r1   r   r8   r   )__name__
__module____qualname____doc__stringify_dialectinherit_cacher   memoized_propertyr(   r   _on_conflict_exclusiver   r4   r7    r$   r#   r   r   ;   s     !M	:	?: :. 0! $0
  59.2 $#'111 ,1 	1
 !1 
1  1f  59.21 , 
	  r$   c                  :    e Zd ZU dZded<   ded<   	 	 d	 	 	 d	dZy)
OnConflictClauser&   z.Optional[List[Union[str, schema.Column[Any]]]]inferred_target_elementsz/Optional[Union[ColumnElement[Any], TextClause]]inferred_target_whereclauseNc                   |k|D cg c]&  }t        j                  t        j                  |      ( c}| _        |*t        j                  t        j
                  |      | _        y d | _        y d x| _        | _        y c c}w )N)r   expectr   DDLConstraintColumnRolerF   WhereHavingRolerG   )r+   r0   r1   columns       r#   __init__zOnConflictClause.__init__   s    
 % --,F   !>!>G,-D) *	   )) ,  , D)0-s   +A=r:   )r0   r
   r1   r   )r;   r<   r=   r?   __annotations__rM   rC   r$   r#   rE   rE      s7     LL"  59.21 ,r$   rE   c                      e Zd ZdZy)r6   r7   N)r;   r<   r=   __visit_name__rC   r$   r#   r6   r6      s    -Nr$   r6   c                  R     e Zd ZU dZded<   ded<   	 	 	 	 d	 	 	 	 	 	 	 d fdZ xZS )	r/   r4   z0List[Tuple[Union[schema.Column[Any], str], Any]]update_values_to_setzOptional[ColumnElement[Any]]update_whereclausec                   t         |   ||       t        |t              r|s2t	        d      t        |t
              rt        |      }nt	        d      |j                         D cg c]+  \  }}t        j                  t        j                  |      |f- c}}| _        |*t        j                  t        j                  |      | _        y d | _        y c c}}w )N)r0   r1   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 object)superrM   
isinstancedict
ValueErrorr   itemsr   rI   r   DMLColumnRolerR   rK   rS   )r+   r0   r1   r2   r3   keyvalue	__class__s          r#   rM   zOnConflictDoUpdate.__init__   s     	)# 	 	

 dD! !MNN./:D$  #jjl%
*
U e1137?*%
!   U22E: 	  		%
s   *0Cr9   )r0   r
   r1   r   r2   r   r3   r   )r;   r<   r=   rP   rN   rM   __classcell__)r]   s   @r#   r/   r/      sR    ,NJJ44 59.2 $#'
1
 ,
 	

 !
 
r$   r/   N)r"   r   r8   r   )+
__future__r   typingr   r   r   r   r   _typingr
   r   r   r    r   sqlr   r   r   sql._typingr   sql.baser   r   r   r   sql.dmlr   StandardInsertsql.elementsr   r   r   r   sql.expressionr   util.typingr   __all__r    rE   r6   r/   rC   r$   r#   <module>rl      s    #      / , % '     , * # ( 0 / ) ) . & # 
(@^ @F} @.* .%
) %
r$   