diff options
author | Mike Bayer <mike_mp@zzzcomputing.com> | 2014-02-26 13:28:14 -0500 |
---|---|---|
committer | Mike Bayer <mike_mp@zzzcomputing.com> | 2014-02-26 13:28:14 -0500 |
commit | bf67069d264cba3feed8a48614289d605ed61a55 (patch) | |
tree | 3101f06439d9c9ab6a3f7dee1ab48cc5707d6f16 | |
parent | 6fadb57165248e7142ba8b4d1f3e32fa77054528 (diff) | |
download | sqlalchemy-bf67069d264cba3feed8a48614289d605ed61a55.tar.gz |
- Fixed issue in new :meth:`.TextClause.columns` method where the ordering
of columns given positionally would not be preserved. This could
have potential impact in positional situations such as applying the
resulting :class:`.TextAsFrom` object to a union.
-rw-r--r-- | doc/build/changelog/changelog_09.rst | 8 | ||||
-rw-r--r-- | lib/sqlalchemy/sql/elements.py | 14 | ||||
-rw-r--r-- | test/sql/test_selectable.py | 25 | ||||
-rw-r--r-- | test/sql/test_text.py | 14 |
4 files changed, 54 insertions, 7 deletions
diff --git a/doc/build/changelog/changelog_09.rst b/doc/build/changelog/changelog_09.rst index c29db6f92..a6245bdb7 100644 --- a/doc/build/changelog/changelog_09.rst +++ b/doc/build/changelog/changelog_09.rst @@ -15,6 +15,14 @@ :version: 0.9.4 .. change:: + :tags: bug, sql + + Fixed issue in new :meth:`.TextClause.columns` method where the ordering + of columns given positionally would not be preserved. This could + have potential impact in positional situations such as applying the + resulting :class:`.TextAsFrom` object to a union. + + .. change:: :tags: feature, sql :tickets: 2962, 2866 diff --git a/lib/sqlalchemy/sql/elements.py b/lib/sqlalchemy/sql/elements.py index 2846b3b51..1b49a7cd1 100644 --- a/lib/sqlalchemy/sql/elements.py +++ b/lib/sqlalchemy/sql/elements.py @@ -1446,13 +1446,13 @@ class TextClause(Executable, ClauseElement): """ - col_by_name = dict( - (col.key, col) for col in cols - ) - for key, type_ in types.items(): - col_by_name[key] = ColumnClause(key, type_) - - return selectable.TextAsFrom(self, list(col_by_name.values())) + input_cols = [ + ColumnClause(col.key, types.pop(col.key)) + if col.key in types + else col + for col in cols + ] + [ColumnClause(key, type_) for key, type_ in types.items()] + return selectable.TextAsFrom(self, input_cols) @property def type(self): diff --git a/test/sql/test_selectable.py b/test/sql/test_selectable.py index dbd73a836..9617cfdf7 100644 --- a/test/sql/test_selectable.py +++ b/test/sql/test_selectable.py @@ -388,6 +388,31 @@ class SelectableTest(fixtures.TestBase, AssertsExecutionResults, AssertsCompiled assert u.corresponding_column(s2.c.table2_coly) is u.c.coly assert s2.corresponding_column(u.c.coly) is s2.c.table2_coly + def test_union_of_alias(self): + s1 = select([table1.c.col1, table1.c.col2]) + s2 = select([table1.c.col1, table1.c.col2]).alias() + + u1 = union(s1, s2) + assert u1.corresponding_column(s1.c.col1) is u1.c.col1 + assert u1.corresponding_column(s2.c.col1) is u1.c.col1 + + u2 = union(s2, s1) + assert u2.corresponding_column(s1.c.col1) is u2.c.col1 + assert u2.corresponding_column(s2.c.col1) is u2.c.col1 + + def test_union_of_text(self): + s1 = select([table1.c.col1, table1.c.col2]) + s2 = text("select col1, col2 from foo").columns( + column('col1'), column('col2')) + + u1 = union(s1, s2) + assert u1.corresponding_column(s1.c.col1) is u1.c.col1 + assert u1.corresponding_column(s2.c.col1) is u1.c.col1 + + u2 = union(s2, s1) + assert u2.corresponding_column(s1.c.col1) is u2.c.col1 + assert u2.corresponding_column(s2.c.col1) is u2.c.col1 + def test_select_union(self): # like testaliasunion, but off a Select off the union. diff --git a/test/sql/test_text.py b/test/sql/test_text.py index 57dadfb16..ef63f9daa 100644 --- a/test/sql/test_text.py +++ b/test/sql/test_text.py @@ -322,6 +322,20 @@ class AsFromTest(fixtures.TestBase, AssertsCompiledSQL): } ) + def test_column_collection_ordered(self): + t = text("select a, b, c from foo").columns(column('a'), + column('b'), column('c')) + eq_(t.c.keys(), ['a', 'b', 'c']) + + def test_column_collection_pos_plus_bykey(self): + # overlapping positional names + type names + t = text("select a, b, c from foo").columns(column('a'), + column('b'), b=Integer, c=String) + eq_(t.c.keys(), ['a', 'b', 'c']) + eq_(t.c.b.type._type_affinity, Integer) + eq_(t.c.c.type._type_affinity, String) + + def _xy_table_fixture(self): m = MetaData() t = Table('t', m, Column('x', Integer), Column('y', Integer)) |