Autocommit support for psycopg2 < 2.4.2

2012-08-14 15:55:47 -05:00 · 2012-08-14 15:55:47 -05:00 · 7dcd1bd223
commit 7dcd1bd223
parent 95fc5dd4a8
1 changed files with 6 additions and 1 deletions
--- 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)