From cc4be53dd4d83c33e13d4574610e793e758076aa Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?F=C3=A9lix=20Sipma?= Date: Fri, 4 Jan 2013 12:48:29 +0100 Subject: [PATCH] use double-quotes in postgresql commands (fix bug with "-") --- postgresql_db | 8 ++++---- postgresql_user | 20 ++++++++++---------- 2 files changed, 14 insertions(+), 14 deletions(-) diff --git a/postgresql_db b/postgresql_db index af1b472f257..d1ae5fe4011 100644 --- a/postgresql_db +++ b/postgresql_db @@ -78,7 +78,7 @@ else: # def set_owner(cursor, db, owner): - query = "ALTER DATABASE %s OWNER TO %s" % (db, owner) + query = "ALTER DATABASE \"%s\" OWNER TO \"%s\"" % (db, owner) cursor.execute(query) return True @@ -95,7 +95,7 @@ def db_exists(cursor, db): def db_delete(cursor, db): if db_exists(cursor, db): - query = "DROP DATABASE %s" % db + query = "DROP DATABASE \"%s\"" % db cursor.execute(query) return True else: @@ -104,9 +104,9 @@ def db_delete(cursor, db): def db_create(cursor, db, owner, template, encoding): if not db_exists(cursor, db): if owner: - owner = " OWNER %s" % owner + owner = " OWNER \"%s\"" % owner if template: - template = " TEMPLATE %s" % template + template = " TEMPLATE \"%s\"" % template if encoding: encoding = " ENCODING '%s'" % encoding query = "CREATE DATABASE %s%s%s%s" % (db, owner, template, encoding) diff --git a/postgresql_user b/postgresql_user index 70e7f956cbe..2f71574ed4c 100644 --- a/postgresql_user +++ b/postgresql_user @@ -136,7 +136,7 @@ def user_exists(cursor, user): def user_add(cursor, user, password, role_attr_flags): """Create a new user with write access to the database""" - query = "CREATE USER %(user)s with PASSWORD '%(password)s' %(role_attr_flags)s" + query = "CREATE USER \"%(user)s\" with PASSWORD '%(password)s' %(role_attr_flags)s" cursor.execute(query % {"user": user, "password": password, "role_attr_flags": role_attr_flags}) return True @@ -154,11 +154,11 @@ def user_alter(cursor, user, password, role_attr_flags): if password is not None: # Update the role attributes, including password. - alter = "ALTER USER %(user)s WITH PASSWORD '%(password)s' %(role_attr_flags)s" + alter = "ALTER USER \"%(user)s\" WITH PASSWORD '%(password)s' %(role_attr_flags)s" cursor.execute(alter % {"user": user, "password": password, "role_attr_flags": role_attr_flags}) else: # Update the role attributes, excluding password. - alter = "ALTER USER %(user)s WITH %(role_attr_flags)s" + alter = "ALTER USER \"%(user)s\" WITH %(role_attr_flags)s" cursor.execute(alter % {"user": user, "role_attr_flags": role_attr_flags}) # Grab new role attributes. cursor.execute(select, {"user": user}) @@ -175,7 +175,7 @@ def user_delete(cursor, user): """Try to remove a user. Returns True if successful otherwise False""" cursor.execute("SAVEPOINT ansible_pgsql_user_delete") try: - cursor.execute("DROP USER %s" % user) + cursor.execute("DROP USER \"%s\"" % user) except: cursor.execute("ROLLBACK TO SAVEPOINT ansible_pgsql_user_delete") cursor.execute("RELEASE SAVEPOINT ansible_pgsql_user_delete") @@ -185,7 +185,7 @@ def user_delete(cursor, user): return True def has_table_privilege(cursor, user, table, priv): - query = 'SELECT has_table_privilege(%s, %s, %s)' + query = 'SELECT has_table_privilege(\'%s\', \'%s\', \'%s\')' cursor.execute(query, (user, table, priv)) return cursor.fetchone()[0] @@ -202,14 +202,14 @@ def get_table_privileges(cursor, user, table): def grant_table_privilege(cursor, user, table, priv): prev_priv = get_table_privileges(cursor, user, table) - query = 'GRANT %s ON TABLE %s TO %s' % (priv, table, user) + query = 'GRANT %s ON TABLE \"%s\" TO \"%s\"' % (priv, table, user) cursor.execute(query) curr_priv = get_table_privileges(cursor, user, table) return len(curr_priv) > len(prev_priv) def revoke_table_privilege(cursor, user, table, priv): prev_priv = get_table_privileges(cursor, user, table) - query = 'REVOKE %s ON TABLE %s FROM %s' % (priv, table, user) + query = 'REVOKE %s ON TABLE \"%s\" FROM \"%s\"' % (priv, table, user) cursor.execute(query) curr_priv = get_table_privileges(cursor, user, table) return len(curr_priv) < len(prev_priv) @@ -235,20 +235,20 @@ def get_database_privileges(cursor, user, db): return o def has_database_privilege(cursor, user, db, priv): - query = 'SELECT has_database_privilege(%s, %s, %s)' + query = 'SELECT has_database_privilege(\'%s\', \'%s\', \'%s\')' cursor.execute(query, (user, db, priv)) return cursor.fetchone()[0] def grant_database_privilege(cursor, user, db, priv): prev_priv = get_database_privileges(cursor, user, db) - query = 'GRANT %s ON DATABASE %s TO %s' % (priv, db, user) + query = 'GRANT %s ON DATABASE \"%s\" TO \"%s\"' % (priv, db, user) cursor.execute(query) curr_priv = get_database_privileges(cursor, user, db) return len(curr_priv) > len(prev_priv) def revoke_database_privilege(cursor, user, db, priv): prev_priv = get_database_privileges(cursor, user, db) - query = 'REVOKE %s ON DATABASE %s FROM %s' % (priv, db, user) + query = 'REVOKE %s ON DATABASE \"%s\" FROM \"%s\"' % (priv, db, user) cursor.execute(query) curr_priv = get_database_privileges(cursor, user, db) return len(curr_priv) < len(prev_priv)