o g^/@sddlmZddlmZddlmZddlmZddlmZddlmZddlm Z d d l m Z d d l m Z d d l m Z d d l mZd dl mZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm!Z!ddlm"Z"ddlm#Z#ddl$m%Z%ddl&m'Z'd Z(d/d%d&Z)Gd'd$d$eZGd(d)d)e Z*Gd*d+d+e*Z+Gd,d-d-e*Z,d.S)0) annotations)Any)List)Optional)Tuple)Union)ext)_OnConflictConstraintT)_OnConflictIndexElementsT)_OnConflictIndexWhereT)_OnConflictSetT)_OnConflictWhereT)util) coercions)roles)schema)_DMLTableArgument)_exclusive_against) _generative)ColumnCollection)ReadOnlyColumnCollectionInsert) ClauseElement) ColumnElement)KeyedColumnElement) TextClause)alias)Self)rinserttablerreturnrcCst|S)asConstruct 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#r%^/var/www/html/flaskapp/venv/lib/python3.10/site-packages/sqlalchemy/dialects/postgresql/dml.pyr"*sr"c@speZdZdZdZdZejdddZe ddd id Z e e ddddZ e e ddddZ d S) rzPostgreSQL-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]]cCst|jddjS)akProvide 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)selfr%r%r&r)KszInsert.excluded_post_values_clausezCThis Insert construct already has an ON CONFLICT clause established)msgsN constraintr index_elementsr index_wherer set_rwhererr!cCst||||||_|S)a$ 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. 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). .. seealso:: :ref:`postgresql_insert_on_conflict` )OnConflictDoUpdater-r,r/r0r1r2r3r%r%r&on_conflict_do_updateps< zInsert.on_conflict_do_updatecCst||||_|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/r0r1r%r%r&on_conflict_do_nothings zInsert.on_conflict_do_nothing)r$r(NNNNN) r/r r0r r1r r2rr3rr$r!NNN)r/r r0r r1r r$r!)__name__ __module__ __qualname____doc__stringify_dialect inherit_cachermemoized_propertyr)r_on_conflict_exclusiverr6r9r%r%r%r&r>s2  ?c@s<eZdZUdZded<ded<ded<   ddddZdS)OnConflictClauser'z Optional[str]constraint_targetz.Optional[List[Union[str, schema.Column[Any]]]]inferred_target_elementsz/Optional[Union[ColumnElement[Any], TextClause]]inferred_target_whereclauseNr/r r0r r1r cCs4|durt|tst|tjtjfrt|dp|}|dur^|dur%tdt|tr4||_d|_ d|_ n*t|tj rF|j }|j dd}nt|tjrS|j}|j}n |j}|j dd}|durd|_dd|D|_ |durtt|tjr|tjntj||_ dSd|_ dS|durd|_|_ |_ dSdS)Nr*z8'constraint' and 'index_elements' are mutually exclusiver'r3cSsg|] }ttj|qSr%)rexpectrDDLConstraintColumnRole).0columnr%r%r& s z-OnConflictClause.__init__..) isinstancestrr Constraintr ExcludeConstraintgetattr ValueErrorrErFrGIndex expressionsdialect_optionsgetr+r3rrHrStatementOptionRoleWhereHavingRoler8r%r%r&__init__s^           zOnConflictClause.__init__r;)r/r r0r r1r )r<r=r>r@__annotations__rYr%r%r%r&rDs rDc@seZdZdZdS)r7r9N)r<r=r>__visit_name__r%r%r%r&r7sr7cs@eZdZUdZded<ded<     ddfdd ZZS)r4r6z0List[Tuple[Union[schema.Column[Any], str], Any]]update_values_to_setzOptional[ColumnElement[Any]]update_whereclauseNr/r r0r r1r r2rr3rcstj|||d|jdur|jdurtdt|tr#|s"tdnt|tr-t|}ntddd|D|_ |durIt t j ||_dSd|_dS)N)r/r0r1zVEither 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 objectcSs"g|] \}}ttj||fqSr%)rrHr DMLColumnRole)rJkeyvaluer%r%r&rLKsz/OnConflictDoUpdate.__init__..)superrYrFrErRrMdictritemsr\rrHrrXr]r5 __class__r%r&rY)s6      zOnConflictDoUpdate.__init__r:) r/r r0r r1r r2rr3r)r<r=r>r[rZrY __classcell__r%r%rdr&r4#s r4N)r#rr$r)- __future__rtypingrrrrrr _typingr r r rrrsqlrrr sql._typingrsql.baserrrrsql.dmlrStandardInsert sql.elementsrrrrsql.expressionr util.typingr!__all__r"rDr7r4r%r%r%r&sF                             H