Autocommit support for psycopg2 < 2.4.2

7dcd1bd2 · Pepe Barbe · Pepe Barbe · 95fc5dd4 · 7dcd1bd2
Commit 7dcd1bd2 authored Aug 14, 2012 by Pepe Barbe Committed by Pepe Barbe Aug 14, 2012
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 1 deletions

library/postgresql_db
+6 -1

No files found.
--- a/library/postgresql_db
+++ b/library/postgresql_db
@@ -89,7 +89,12 @@ def main():
    try:
        db_connection = psycopg2.connect(database="template1", **kw)
        # Enable autocommit so we can create databases
-        db_connection.autocommit = True
+        if psycopg2.__version__ >= '2.4.2':
+            db_connection.autocommit = True
+        else:
+            db_connection.set_isolation_level(psycopg2
+                                              .extensions
+                                              .ISOLATION_LEVEL_AUTOCOMMIT)
        cursor = db_connection.cursor()
    except Exception, e:
        module.fail_json(msg="unable to connect to database: %s" % e)