3
a&                 @   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ZG dd
 d
e	ZeeddZG dd deZdS )   )exc)util)_exclusive_against)_generative)ColumnCollection)Insert)ClauseElement)alias)public_factoryr   insertc               @   sJ   e Zd ZdZdZedd Zejdd Z	e
edddid	d
d ZdS )r   a  MySQL-specific implementation of INSERT.

    Adds methods for MySQL-specific syntaxes such as ON DUPLICATE KEY UPDATE.

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

    .. versionadded:: 1.2

    mysqlc             C   s   | j jS )a  Provide the "inserted" namespace for an ON DUPLICATE KEY UPDATE statement

        MySQL's ON DUPLICATE KEY UPDATE clause allows reference to the row
        that would be inserted, via a special function called ``VALUES()``.
        This attribute provides all columns in this row to be referenceable
        such that they will render within a ``VALUES()`` function inside the
        ON DUPLICATE KEY UPDATE clause.    The attribute is named ``.inserted``
        so as not to conflict with the existing
        :meth:`_expression.Insert.values` method.

        .. tip::  The :attr:`_mysql.Insert.inserted` 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.inserted.some_column``), but special names and
            dictionary method names should be accessed using indexed access,
            such as ``stmt.inserted["column name"]`` or
            ``stmt.inserted["values"]``.  See the docstring for
            :class:`_expression.ColumnCollection` for further examples.

        .. seealso::

            :ref:`mysql_insert_on_duplicate_key_update` - example of how
            to use :attr:`_expression.Insert.inserted`

        )inserted_aliascolumns)self r   C/tmp/pip-build-6_cqtusv/SQLAlchemy/sqlalchemy/dialects/mysql/dml.pyinserted   s    zInsert.insertedc             C   s   t | jddS )Nr   )name)r	   table)r   r   r   r   r   <   s    zInsert.inserted_alias_post_values_clausezDThis Insert construct already has an ON DUPLICATE KEY clause present)Zmsgsc             O   sV   |r|rt jd|r6t|dkr,t jd|d }n|}t| dd}t||| _dS )aE  
        Specifies the ON DUPLICATE KEY UPDATE clause.

        :param \**kw:  Column keys linked to UPDATE values.  The
         values may be any SQL expression or supported literal Python
         values.

        .. 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 DUPLICATE KEY UPDATE
           style of UPDATE, unless values are manually specified here.

        :param \*args: As an alternative to passing key/value parameters,
         a dictionary or list of 2-tuples can be passed as a single positional
         argument.

         Passing a single dictionary is equivalent to the keyword argument
         form::

            insert().on_duplicate_key_update({"name": "some name"})

         Passing a list of 2-tuples indicates that the parameter assignments
         in the UPDATE clause should be ordered as sent, in a manner similar
         to that described for the :class:`_expression.Update`
         construct overall
         in :ref:`updates_order_parameters`::

            insert().on_duplicate_key_update(
                [("name", "some name"), ("value", "some value")])

         .. versionchanged:: 1.3 parameters can be specified as a dictionary
            or list of 2-tuples; the latter form provides for parameter
            ordering.


        .. versionadded:: 1.2

        .. seealso::

            :ref:`mysql_insert_on_duplicate_key_update`

        z9Can't pass kwargs and positional arguments simultaneously   zDOnly a single dictionary or list of tuples is accepted positionally.    r   N)r   ArgumentErrorlengetattrOnDuplicateClauser   )r   argskwvaluesr   r   r   r   on_duplicate_key_update@   s    4
zInsert.on_duplicate_key_updateN)__name__
__module____qualname____doc__stringify_dialectpropertyr   r   Zmemoized_propertyr   r   r   r   r   r   r   r   r      s   
z.dialects.mysql.insertz.dialects.mysql.Insertc               @   s    e Zd ZdZdZdZdd ZdS )r   r   Nr   c             C   sx   || _ t|tr:|r:t|d tr:dd |D | _t|}t|trR|sntdnt|trft|}ntd|| _d S )Nr   c             S   s   g | ]\}}|qS r   r   ).0keyvaluer   r   r   
<listcomp>   s    z.OnDuplicateClause.__init__.<locals>.<listcomp>z-update parameter dictionary must not be emptyztupdate parameter must be a non-empty dictionary or a ColumnCollection such as the `.c.` collection of a Table object)	r   
isinstancelisttuple_parameter_orderingdict
ValueErrorr   update)r   r   r0   r   r   r   __init__   s    



zOnDuplicateClause.__init__)r    r!   r"   Z__visit_name__r-   r$   r1   r   r   r   r   r      s   r   N)r   r   ) r   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   <module>   s   x
