diff options
author | Mike Bayer <mike_mp@zzzcomputing.com> | 2010-07-21 11:33:47 -0400 |
---|---|---|
committer | Mike Bayer <mike_mp@zzzcomputing.com> | 2010-07-21 11:33:47 -0400 |
commit | dabe38bf56dd18876466f07667c386c56ba88de4 (patch) | |
tree | 7d4fa21f1b00668defac05dbcb68230ceddea72e /lib/sqlalchemy/sql/compiler.py | |
parent | 1b0bd4b23f572f581ded0395b78a3371f889ee21 (diff) | |
download | sqlalchemy-dabe38bf56dd18876466f07667c386c56ba88de4.tar.gz |
- Changed the scheme used to generate truncated
"auto" index names when using the "index=True"
flag on Column. The truncation only takes
place with the auto-generated name, not one
that is user-defined (an error would be
raised instead), and the truncation scheme
itself is now based on a fragment of an md5
hash of the identifier name, so that multiple
indexes on columns with similar names still
have unique names. [ticket:1855]
Diffstat (limited to 'lib/sqlalchemy/sql/compiler.py')
-rw-r--r-- | lib/sqlalchemy/sql/compiler.py | 30 |
1 files changed, 15 insertions, 15 deletions
diff --git a/lib/sqlalchemy/sql/compiler.py b/lib/sqlalchemy/sql/compiler.py index c54931b87..0383f9690 100644 --- a/lib/sqlalchemy/sql/compiler.py +++ b/lib/sqlalchemy/sql/compiler.py @@ -1179,7 +1179,18 @@ class DDLCompiler(engine.Compiled): def visit_drop_table(self, drop): return "\nDROP TABLE " + self.preparer.format_table(drop.element) - + + def _index_identifier(self, ident): + if isinstance(ident, sql._generated_label): + if len(ident) > self.dialect.max_identifier_length: + return ident[0:self.dialect.max_identifier_length - 8] + \ + "_" + util.md5_hex(ident)[-4:] + else: + return ident + else: + self.dialect.validate_identifier(ident) + return ident + def visit_create_index(self, create): index = create.element preparer = self.preparer @@ -1187,7 +1198,8 @@ class DDLCompiler(engine.Compiled): if index.unique: text += "UNIQUE " text += "INDEX %s ON %s (%s)" \ - % (preparer.quote(self._validate_identifier(index.name, True), index.quote), + % (preparer.quote(self._index_identifier(index.name), + index.quote), preparer.format_table(index.table), ', '.join(preparer.quote(c.name, c.quote) for c in index.columns)) @@ -1196,7 +1208,7 @@ class DDLCompiler(engine.Compiled): def visit_drop_index(self, drop): index = drop.element return "\nDROP INDEX " + \ - self.preparer.quote(self._validate_identifier(index.name, False), index.quote) + self.preparer.quote(self._index_identifier(index.name), index.quote) def visit_add_constraint(self, create): preparer = self.preparer @@ -1238,18 +1250,6 @@ class DDLCompiler(engine.Compiled): def post_create_table(self, table): return '' - def _validate_identifier(self, ident, truncate): - if truncate: - if len(ident) > self.dialect.max_identifier_length: - counter = getattr(self, 'counter', 0) - self.counter = counter + 1 - return ident[0:self.dialect.max_identifier_length - 6] + "_" + hex(self.counter)[2:] - else: - return ident - else: - self.dialect.validate_identifier(ident) - return ident - def get_column_default_string(self, column): if isinstance(column.server_default, schema.DefaultClause): if isinstance(column.server_default.arg, basestring): |