From 8e112ad2cc45910a8ca860f32a3a4e138bae2e73 Mon Sep 17 00:00:00 2001 From: Aubrey Stark-Toller Date: Fri, 5 Jan 2018 15:06:23 -0500 Subject: [PATCH] `ValuesBase.values` inconsistency fix Fixed bug in :meth:`.Insert.values` where using the "multi-values" format in combination with :class:`.Column` objects as keys rather than strings would fail. Pull request courtesy Aubrey Stark-Toller. Change-Id: I9d3b40b5950df8f5bfdc8b1d22f9c3afb277f17f Pull-request: https://github.com/zzzeek/sqlalchemy/pull/412 Fixes: #4162 (cherry picked from commit ab2165e6d6e9b33f772e9eb3668b3e32175093c4) --- doc/build/changelog/unreleased_11/4162.rst | 8 +++++ lib/sqlalchemy/sql/crud.py | 39 ++++++++++++---------- test/sql/test_insert.py | 33 ++++++++++++++++++ 3 files changed, 63 insertions(+), 17 deletions(-) create mode 100644 doc/build/changelog/unreleased_11/4162.rst diff --git a/doc/build/changelog/unreleased_11/4162.rst b/doc/build/changelog/unreleased_11/4162.rst new file mode 100644 index 0000000000..8f831f6ae3 --- /dev/null +++ b/doc/build/changelog/unreleased_11/4162.rst @@ -0,0 +1,8 @@ +.. change:: + :tags: bug, sql + :tickets: 4162 + :versions: 1.2.1 + + Fixed bug in :meth:`.Insert.values` where using the "multi-values" + format in combination with :class:`.Column` objects as keys rather + than strings would fail. Pull request courtesy Aubrey Stark-Toller. diff --git a/lib/sqlalchemy/sql/crud.py b/lib/sqlalchemy/sql/crud.py index 5739c22f9e..b71256bd5a 100644 --- a/lib/sqlalchemy/sql/crud.py +++ b/lib/sqlalchemy/sql/crud.py @@ -597,23 +597,28 @@ def _extend_values_for_multiparams(compiler, stmt, values, kw): values_0 = values values = [values] - values.extend( - [ - ( - c, - (_create_bind_param( - compiler, c, row[c.key], - name="%s_m%d" % (c.key, i + 1), **kw - ) if elements._is_literal(row[c.key]) - else compiler.process( - row[c.key].self_group(), **kw)) - if c.key in row else - _process_multiparam_default_bind(compiler, stmt, c, i, kw) - ) - for (c, param) in values_0 - ] - for i, row in enumerate(stmt.parameters[1:]) - ) + for i, row in enumerate(stmt.parameters[1:]): + extension = [] + for (col, param) in values_0: + if col in row or col.key in row: + key = col if col in row else col.key + + if elements._is_literal(row[key]): + new_param = _create_bind_param( + compiler, col, row[key], + name="%s_m%d" % (col.key, i + 1), **kw + ) + else: + new_param = compiler.process(row[key].self_group(), **kw) + else: + new_param = _process_multiparam_default_bind( + compiler, stmt, col, i, kw + ) + + extension.append((col, new_param)) + + values.append(extension) + return values diff --git a/test/sql/test_insert.py b/test/sql/test_insert.py index d3bc11a969..6d41a4dca5 100644 --- a/test/sql/test_insert.py +++ b/test/sql/test_insert.py @@ -839,6 +839,39 @@ class MultirowTest(_InsertTestBase, fixtures.TablesTest, AssertsCompiledSQL): checkparams=checkparams, dialect=dialect) + def test_named_with_column_objects(self): + table1 = self.tables.mytable + + values = [ + {table1.c.myid: 1, table1.c.name: 'a', table1.c.description: 'b'}, + {table1.c.myid: 2, table1.c.name: 'c', table1.c.description: 'd'}, + {table1.c.myid: 3, table1.c.name: 'e', table1.c.description: 'f'}, + ] + + checkparams = { + 'myid_m0': 1, + 'myid_m1': 2, + 'myid_m2': 3, + 'name_m0': 'a', + 'name_m1': 'c', + 'name_m2': 'e', + 'description_m0': 'b', + 'description_m1': 'd', + 'description_m2': 'f', + } + + dialect = default.DefaultDialect() + dialect.supports_multivalues_insert = True + + self.assert_compile( + table1.insert().values(values), + 'INSERT INTO mytable (myid, name, description) VALUES ' + '(:myid_m0, :name_m0, :description_m0), ' + '(:myid_m1, :name_m1, :description_m1), ' + '(:myid_m2, :name_m2, :description_m2)', + checkparams=checkparams, + dialect=dialect) + def test_positional(self): table1 = self.tables.mytable -- 2.47.2