diff options
| author | Mike Bayer <mike_mp@zzzcomputing.com> | 2009-10-21 04:47:02 +0000 |
|---|---|---|
| committer | Mike Bayer <mike_mp@zzzcomputing.com> | 2009-10-21 04:47:02 +0000 |
| commit | 5a140299b3ec6c7d800484fd6b5acedacc4aabb7 (patch) | |
| tree | 80cc5623b6281f6cec568b82764e4affdefab4f3 /lib/sqlalchemy/dialects/postgresql/base.py | |
| parent | 7dddcd140339509fb3d208a37eb1ac0f1ea3aa3a (diff) | |
| download | sqlalchemy-5a140299b3ec6c7d800484fd6b5acedacc4aabb7.tar.gz | |
- Corrected the "has_sequence" query to take current schema,
or explicit sequence-stated schema, into account.
[ticket:1576]
Diffstat (limited to 'lib/sqlalchemy/dialects/postgresql/base.py')
| -rw-r--r-- | lib/sqlalchemy/dialects/postgresql/base.py | 24 |
1 files changed, 17 insertions, 7 deletions
diff --git a/lib/sqlalchemy/dialects/postgresql/base.py b/lib/sqlalchemy/dialects/postgresql/base.py index 0bc5f08b0..061c3b066 100644 --- a/lib/sqlalchemy/dialects/postgresql/base.py +++ b/lib/sqlalchemy/dialects/postgresql/base.py @@ -539,13 +539,23 @@ class PGDialect(default.DefaultDialect): ) return bool(cursor.first()) - def has_sequence(self, connection, sequence_name): - cursor = connection.execute( - sql.text("SELECT relname FROM pg_class WHERE relkind = 'S' AND " - "relnamespace IN ( SELECT oid FROM pg_namespace WHERE nspname NOT LIKE 'pg_%' " - "AND nspname != 'information_schema' AND relname = :seqname)", - bindparams=[sql.bindparam('seqname', unicode(sequence_name), type_=sqltypes.Unicode)] - )) + def has_sequence(self, connection, sequence_name, schema=None): + if schema is None: + cursor = connection.execute( + sql.text("SELECT relname FROM pg_class c join pg_namespace n on " + "n.oid=c.relnamespace where relkind='S' and n.nspname=current_schema() and lower(relname)=:name", + bindparams=[sql.bindparam('name', unicode(sequence_name.lower()), type_=sqltypes.Unicode)] + ) + ) + else: + cursor = connection.execute( + sql.text("SELECT relname FROM pg_class c join pg_namespace n on " + "n.oid=c.relnamespace where relkind='S' and n.nspname=:schema and lower(relname)=:name", + bindparams=[sql.bindparam('name', unicode(sequence_name.lower()), type_=sqltypes.Unicode), + sql.bindparam('schema', unicode(schema), type_=sqltypes.Unicode)] + ) + ) + return bool(cursor.first()) def table_names(self, connection, schema): |
