From: Mike Bayer Date: Wed, 12 Mar 2014 18:46:55 +0000 (-0400) Subject: - Fixed bug in new :paramref:`.MetaData.naming_convention` feature X-Git-Tag: rel_0_9_4~66 X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=d85d6f9a3f1d3132abcd917d4378b2c4e14aec65;p=thirdparty%2Fsqlalchemy%2Fsqlalchemy.git - Fixed bug in new :paramref:`.MetaData.naming_convention` feature where the name of a check constraint making use of the `"%(constraint_name)s"` token would get doubled up for the constraint generated by a boolean or enum type, and overall duplicate events would cause the `"%(constraint_name)s"` token to keep compounding itself. fixes #2991 --- diff --git a/doc/build/changelog/changelog_09.rst b/doc/build/changelog/changelog_09.rst index fc39ea347a..2a8c24f91e 100644 --- a/doc/build/changelog/changelog_09.rst +++ b/doc/build/changelog/changelog_09.rst @@ -14,6 +14,17 @@ .. changelog:: :version: 0.9.4 + .. change:: + :tags: bug, sql + :tickets: 2991 + + Fixed bug in new :paramref:`.MetaData.naming_convention` feature + where the name of a check constraint making use of the + `"%(constraint_name)s"` token would get doubled up for the + constraint generated by a boolean or enum type, and overall + duplicate events would cause the `"%(constraint_name)s"` token + to keep compounding itself. + .. change:: :tags: feature, orm diff --git a/lib/sqlalchemy/sql/naming.py b/lib/sqlalchemy/sql/naming.py index ee99ccbcef..bbb8431211 100644 --- a/lib/sqlalchemy/sql/naming.py +++ b/lib/sqlalchemy/sql/naming.py @@ -22,7 +22,7 @@ class ConventionDict(object): self._is_fk = isinstance(const, ForeignKeyConstraint) self.table = table self.convention = convention - self._const_name = const.name + self._const_name = const._orig_name = getattr(const, '_orig_name', const.name) def _key_table_name(self): return self.table.name diff --git a/lib/sqlalchemy/sql/sqltypes.py b/lib/sqlalchemy/sql/sqltypes.py index fee424e659..ec1d66459c 100644 --- a/lib/sqlalchemy/sql/sqltypes.py +++ b/lib/sqlalchemy/sql/sqltypes.py @@ -1131,7 +1131,7 @@ class Enum(String, SchemaType): _create_rule=util.portable_instancemethod( self._should_create_constraint) ) - table.append_constraint(e) + assert e.table is table def adapt(self, impltype, **kw): schema = kw.pop('schema', self.schema) @@ -1268,7 +1268,7 @@ class Boolean(TypeEngine, SchemaType): _create_rule=util.portable_instancemethod( self._should_create_constraint) ) - table.append_constraint(e) + assert e.table is table @property def python_type(self): diff --git a/test/sql/test_metadata.py b/test/sql/test_metadata.py index ce95634fdc..fd166bc17a 100644 --- a/test/sql/test_metadata.py +++ b/test/sql/test_metadata.py @@ -7,7 +7,7 @@ from sqlalchemy import Integer, String, UniqueConstraint, \ CheckConstraint, ForeignKey, MetaData, Sequence, \ ForeignKeyConstraint, PrimaryKeyConstraint, ColumnDefault, Index, event,\ events, Unicode, types as sqltypes, bindparam, \ - Table, Column + Table, Column, Boolean, Enum from sqlalchemy import schema, exc import sqlalchemy as tsa from sqlalchemy.testing import fixtures @@ -2731,3 +2731,39 @@ class NamingConventionTest(fixtures.TestBase): ['user.id', 'user.version']) a1.append_constraint(fk) eq_(fk.name, "fk_address_HASH_address") + + def test_schematype_ck_name_boolean(self): + m1 = MetaData(naming_convention={ + "ck": "ck_%(table_name)s_%(constraint_name)s"}) + + u1 = Table('user', m1, + Column('x', Boolean(name='foo')) + ) + eq_( + [c for c in u1.constraints + if isinstance(c, CheckConstraint)][0].name, "ck_user_foo" + ) + + def test_schematype_ck_name_enum(self): + m1 = MetaData(naming_convention={ + "ck": "ck_%(table_name)s_%(constraint_name)s"}) + + u1 = Table('user', m1, + Column('x', Enum('a', 'b', name='foo')) + ) + eq_( + [c for c in u1.constraints + if isinstance(c, CheckConstraint)][0].name, "ck_user_foo" + ) + + def test_ck_constraint_redundant_event(self): + u1 = self._fixture(naming_convention={ + "ck": "ck_%(table_name)s_%(constraint_name)s"}) + + ck1 = CheckConstraint(u1.c.version > 3, name='foo') + u1.append_constraint(ck1) + u1.append_constraint(ck1) + u1.append_constraint(ck1) + + eq_(ck1.name, "ck_user_foo") +