summaryrefslogtreecommitdiff
path: root/lib/sqlalchemy/dialects/postgresql/base.py
diff options
context:
space:
mode:
authorMike Bayer <mike_mp@zzzcomputing.com>2009-10-21 04:47:02 +0000
committerMike Bayer <mike_mp@zzzcomputing.com>2009-10-21 04:47:02 +0000
commit5a140299b3ec6c7d800484fd6b5acedacc4aabb7 (patch)
tree80cc5623b6281f6cec568b82764e4affdefab4f3 /lib/sqlalchemy/dialects/postgresql/base.py
parent7dddcd140339509fb3d208a37eb1ac0f1ea3aa3a (diff)
downloadsqlalchemy-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.py24
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):