
    Ng+                       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'd(S )*    )annotations)Any)Optional   )ext   )_OnConflictConstraintT)_OnConflictIndexElementsT)_OnConflictIndexWhereT)_OnConflictSetT)_OnConflictWhereT   )util)	coercions)roles)schema)_DMLTableArgument)_exclusive_against)_generative)ColumnCollection)ReadOnlyColumnCollectionInsert)ClauseElement)KeyedColumnElement)alias)Self)r   inserttabler   returnr   c                     t          |           S )as  Construct a PostgreSQL-specific variant :class:`_postgresql.Insert`
    construct.

    .. container:: inherited_member

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

    r   )r   s    ^/var/www/html/ai-engine/env/lib/python3.11/site-packages/sqlalchemy/dialects/postgresql/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d
S )r   zPostgreSQL-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.

    
postgresqlFr    6ReadOnlyColumnCollection[str, KeyedColumnElement[Any]]c                8    t          | j        d          j        S )ak  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.

        .. tip::  The :attr:`_postgresql.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.

        .. seealso::

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

        excluded)name)r   r   columns)selfs    r"   r(   zInsert.excludedF   s    6 TZj11199r#   _post_values_clausezCThis Insert construct already has an ON CONFLICT clause established)msgsN
constraintr	   index_elementsr
   index_wherer   set_r   wherer   r   c                6    t          |||||          | _        | S )aS  
        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\_:
         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:`_postgresql.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).


        .. seealso::

            :ref:`postgresql_insert_on_conflict`

        )OnConflictDoUpdater,   )r+   r.   r/   r0   r1   r2   s         r"   on_conflict_do_updatezInsert.on_conflict_do_updatek   s)    z $6T5$
 $
  r#   c                2    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.

        .. seealso::

            :ref:`postgresql_insert_on_conflict`

        )OnConflictDoNothingr,   r+   r.   r/   r0   s       r"   on_conflict_do_nothingzInsert.on_conflict_do_nothing   s%    @ $7$
 $
  r#   )r    r&   NNNNN)r.   r	   r/   r
   r0   r   r1   r   r2   r   r    r   NNN)r.   r	   r/   r
   r0   r   r    r   )__name__
__module____qualname____doc__stringify_dialectinherit_cacher   memoized_propertyr(   r   _on_conflict_exclusiver   r5   r9    r#   r"   r   r   9   s          %M	: : : :8 0/! $0
    .248.2 $#'> > > >  [>@  .248.2	! ! ! !  [! ! !r#   c                  B    e Zd ZU dZded<   ded<   ded<   	 	 	 dddZdS )OnConflictClauser%   zOptional[str]constraint_targetr
   inferred_target_elementsr   inferred_target_whereclauseNr.   r	   r/   r0   c                   |Mt          |t                    s8t          |t          j        t          j        f          rt          |d          p|}||t          d          t          |t                    r|| _        d | _	        d | _
        nt          |t          j                  r(|j        }|j        d                             d          }nPt          |t          j                  r|j        }|j        }n'|j        }|j        d                             d          }|d | _        || _	        || _
        d S |d x| _        x| _	        | _
        d S d S )Nr)   z8'constraint' and 'index_elements' are mutually exclusiver%   r2   )
isinstancestrr   
Constraintr   ExcludeConstraintgetattr
ValueErrorrG   rH   rI   Indexexpressionsdialect_optionsgetr*   r2   r8   s       r"   __init__zOnConflictClause.__init__   s    !j#.. G:"C$9:4 4 G %Z88FJ
!) N   *c** )3&04-3700J55 !+!7(8FJJ  J(=>> !+!3(.!+!3(8FJJ  %%)D",:D)/:D,,, D" T%B000  r#   r;   )r.   r	   r/   r
   r0   r   )r<   r=   r>   r@   __annotations__rU   rD   r#   r"   rF   rF      sd         $$$$$77777777 .248.2	, , , , , , ,r#   rF   c                      e Zd ZdZdS )r7   r9   N)r<   r=   r>   __visit_name__rD   r#   r"   r7   r7   	  s        -NNNr#   r7   c                  0     e Zd ZdZ	 	 	 	 	 dd fdZ xZS )r4   r5   Nr.   r	   r/   r
   r0   r   r1   r   r2   r   c                   t                                          |||           | j        | j        t	          d          t          |t                    r|st	          d          n4t          |t                    rt          |          }nt	          d          d |                                D             | _	        || _
        d S )N)r.   r/   r0   zVEither constraint or index_elements, but not both, must be specified unless DO NOTHINGz*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                V    g | ]&\  }}t          j        t          j        |          |f'S rD   )r   expectr   DMLColumnRole).0keyvalues      r"   
<listcomp>z/OnConflictDoUpdate.__init__.<locals>.<listcomp>2  sA     %
 %
 %
U e1377?%
 %
 %
r#   )superrU   rH   rG   rP   rK   dictr   itemsupdate_values_to_setupdate_whereclause)r+   r.   r/   r0   r1   r2   	__class__s         r"   rU   zOnConflictDoUpdate.__init__  s     	!)# 	 	
 	
 	
 )1&.D  
 dD!! 
	 O !MNNNO.// 	::DD$  
%
 %
"jjll%
 %
 %
! #(r#   r:   )
r.   r	   r/   r
   r0   r   r1   r   r2   r   )r<   r=   r>   rX   rU   __classcell__)rg   s   @r"   r4   r4     sX        ,N .248.2 $#'&( &( &( &( &( &( &( &( &( &( &(r#   r4   N)r   r   r    r   )(
__future__r   typingr   r    r   _typingr	   r
   r   r   r   r   sqlr   r   r   sql._typingr   sql.baser   r   r   r   sql.dmlr   StandardInsertsql.elementsr   r   sql.expressionr   util.typingr   __all__r   rF   r7   r4   rD   r#   r"   <module>rv      s~   # " " " " "                   , , , , , , / / / / / / , , , , , , % % % % % % ' ' ' ' ' '                         , , , , , , * * * * * * # # # # # # ( ( ( ( ( ( 0 0 0 0 0 0 / / / / / / ) ) ) ) ) ) . . . . . . # # # # # #          (W W W W W^ W W Wt3 3 3 3 3} 3 3 3l. . . . .* . . .)( )( )( )( )() )( )( )( )( )(r#   