diff --git a/postgresql_db b/postgresql_db index 8bfdc9fd20e..32ee9205909 100755 --- a/postgresql_db +++ b/postgresql_db @@ -27,26 +27,45 @@ else: # PostgreSQL module specific support methods. # +def set_owner(cursor, db, owner): + query = "ALTER DATABASE %s OWNER TO %s" % (db, owner) + cursor.execute(query) + return True + +def db_owned_by(cursor, db, user): + query = """SELECT * FROM pg_database JOIN pg_user ON datdba = usesysid + WHERE usename = %(user)s and datname = %(db)s""" + cursor.execute(query, {'db':db, 'user':user}) + return cursor.rowcount == 1 + def db_exists(cursor, db): query = "SELECT * FROM pg_database WHERE datname=%(db)s" cursor.execute(query, {'db': db}) return cursor.rowcount == 1 def db_delete(cursor, db): - query = "DROP DATABASE %s" % db - cursor.execute(query) - return True + if db_exists(cursor, db): + query = "DROP DATABASE %s" % db + cursor.execute(query) + return True + else: + return False def db_create(cursor, db, owner, template, encoding): - if owner: - owner = " OWNER %s" % owner - if template: - template = " TEMPLATE %s" % template - if encoding: - encoding = " ENCODING '%s'" % encoding - query = "CREATE DATABASE %s%s%s%s" % (db, owner, template, encoding) - cursor.execute(query) - return True + if not db_exists(cursor, db): + if owner: + owner = " OWNER %s" % owner + if template: + template = " TEMPLATE %s" % template + if encoding: + encoding = " ENCODING '%s'" % encoding + query = "CREATE DATABASE %s%s%s%s" % (db, owner, template, encoding) + cursor.execute(query) + return True + elif owner and not db_owned_by(cursor, db, owner): + return set_owner(cursor, db, owner) + else: + return False # =========================================== # Module execution. @@ -100,12 +119,10 @@ def main(): module.fail_json(msg="unable to connect to database: %s" % e) try: - if db_exists(cursor, db): - if state == "absent": - changed = db_delete(cursor, db) - else: - if state == "present": - changed = db_create(cursor, db, owner, template, encoding) + if state == "absent": + changed = db_delete(cursor, db) + elif state == "present": + changed = db_create(cursor, db, owner, template, encoding) except Exception, e: module.fail_json(msg="Database query failed: %s" % e)