diff options
author | mike bayer <mike_mp@zzzcomputing.com> | 2020-03-05 04:18:22 +0000 |
---|---|---|
committer | Gerrit Code Review <gerrit@bbpush.zzzcomputing.com> | 2020-03-05 04:18:22 +0000 |
commit | d72bda5ed23a46bcbf31d40684200dcb79012a33 (patch) | |
tree | cbed66fd751d44dc49a47b899609e1b7317e63b9 /lib/sqlalchemy/dialects/mysql/base.py | |
parent | c278900de7a28c98335d870f393328c868e75126 (diff) | |
parent | 57b2aae0d9efe91c2338e5a762e04366f86c2651 (diff) | |
download | sqlalchemy-d72bda5ed23a46bcbf31d40684200dcb79012a33.tar.gz |
Merge "Render VALUES within composed MySQL on duplicate key expressions"
Diffstat (limited to 'lib/sqlalchemy/dialects/mysql/base.py')
-rw-r--r-- | lib/sqlalchemy/dialects/mysql/base.py | 34 |
1 files changed, 25 insertions, 9 deletions
diff --git a/lib/sqlalchemy/dialects/mysql/base.py b/lib/sqlalchemy/dialects/mysql/base.py index 6ea8cbcb8..a97719532 100644 --- a/lib/sqlalchemy/dialects/mysql/base.py +++ b/lib/sqlalchemy/dialects/mysql/base.py @@ -743,6 +743,8 @@ from collections import defaultdict import re import sys +from sqlalchemy import literal_column +from sqlalchemy.sql import visitors from . import reflection as _reflection from .enumerated import ENUM from .enumerated import SET @@ -1303,17 +1305,31 @@ class MySQLCompiler(compiler.SQLCompiler): if coercions._is_literal(val): val = elements.BindParameter(None, val, type_=column.type) value_text = self.process(val.self_group(), use_schema=False) - elif isinstance(val, elements.BindParameter) and val.type._isnull: - val = val._clone() - val.type = column.type - value_text = self.process(val.self_group(), use_schema=False) - elif ( - isinstance(val, elements.ColumnClause) - and val.table is on_duplicate.inserted_alias - ): - value_text = "VALUES(" + self.preparer.quote(column.name) + ")" else: + + def replace(obj): + if ( + isinstance(obj, elements.BindParameter) + and obj.type._isnull + ): + obj = obj._clone() + obj.type = column.type + return obj + elif ( + isinstance(obj, elements.ColumnClause) + and obj.table is on_duplicate.inserted_alias + ): + obj = literal_column( + "VALUES(" + self.preparer.quote(column.name) + ")" + ) + return obj + else: + # element is not replaced + return None + + val = visitors.replacement_traverse(val, {}, replace) value_text = self.process(val.self_group(), use_schema=False) + name_text = self.preparer.quote(column.name) clauses.append("%s = %s" % (name_text, value_text)) |