A couple of fixes to psycopg1

_psycopg.connect is no more. Also use symbolic consts instead of values
for the isolation level.
This commit is contained in:
Daniele Varrazzo 2011-12-15 13:06:32 +00:00
parent b5de04d2ff
commit 5a6a303d43

View File

@ -33,14 +33,14 @@ from psycopg2.extensions import cursor as _2cursor
from psycopg2.extensions import connection as _2connection from psycopg2.extensions import connection as _2connection
from psycopg2 import * from psycopg2 import *
del connect import psycopg2.extensions as _ext
_2connect = connect
def connect(*args, **kwargs): def connect(*args, **kwargs):
"""connect(dsn, ...) -> new psycopg 1.1.x compatible connection object""" """connect(dsn, ...) -> new psycopg 1.1.x compatible connection object"""
kwargs['connection_factory'] = connection kwargs['connection_factory'] = connection
conn = _2psycopg.connect(*args, **kwargs) conn = _2connect(*args, **kwargs)
conn.set_isolation_level(2) conn.set_isolation_level(_ext.ISOLATION_LEVEL_READ_COMMITTED)
return conn return conn
class connection(_2connection): class connection(_2connection):
@ -53,9 +53,9 @@ class connection(_2connection):
def autocommit(self, on_off=1): def autocommit(self, on_off=1):
"""autocommit(on_off=1) -> switch autocommit on (1) or off (0)""" """autocommit(on_off=1) -> switch autocommit on (1) or off (0)"""
if on_off > 0: if on_off > 0:
self.set_isolation_level(0) self.set_isolation_level(_ext.ISOLATION_LEVEL_AUTOCOMMIT)
else: else:
self.set_isolation_level(2) self.set_isolation_level(_ext.ISOLATION_LEVEL_READ_COMMITTED)
class cursor(_2cursor): class cursor(_2cursor):