Check for database ownership

library/postgresql_db

@@ -27,17 +27,32 @@ 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 count(*) 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):
+    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 not db_exists(cursor, db):
         if owner:
             owner = " OWNER %s" % owner
         if template:
@@ -47,6 +62,10 @@ def db_create(cursor, db, owner, template, 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,11 +119,9 @@ 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":
+        elif state == "present":
             changed = db_create(cursor, db, owner, template, encoding)
     except Exception, e:
         module.fail_json(msg="Database query failed: %s" % e)