def __exit__(self, exc, value, tb, *args, **kwargs):
transaction = current_transaction()
name = callable_name(self.__wrapped__.__exit__)
with FuncWrapper(transaction, name):
if self.is_commit_on_exit(exc, value, tb):
with DatabaseTrace(transaction, 'COMMIT',
self._pam_dbapi2_module,
self._pam_connect_params):
return self.__wrapped__.__exit__(exc, value, tb)
else:
with DatabaseTrace(transaction, 'ROLLBACK',
self._pam_dbapi2_module,
self._pam_connect_params):
return self.__wrapped__.__exit__(exc, value, tb)
def executemany(self, sql, seq_of_parameters):
transaction = current_transaction()
try:
parameters = seq_of_parameters[0]
except (TypeError, IndexError):
parameters = DEFAULT
if parameters is not DEFAULT:
with DatabaseTrace(transaction, sql, self._pam_dbapi2_module,
self._pam_connect_params,
self._pam_cursor_params, parameters):
return self.__wrapped__.executemany(sql, seq_of_parameters)
else:
with DatabaseTrace(transaction, sql, self._pam_dbapi2_module,
self._pam_connect_params,
self._pam_cursor_params):
return self.__wrapped__.executemany(sql, seq_of_parameters)
def commit(self):
transaction = current_transaction()
with DatabaseTrace(transaction,
'COMMIT',
self._pam_dbapi2_module,
database_name=self._pam_connect_params[0][0]):
return self.__wrapped__.commit()
def execute(self, sql, parameters=DEFAULT):
transaction = current_transaction()
if parameters is not DEFAULT:
with DatabaseTrace(transaction,
sql,
self._pam_dbapi2_module,
self._pam_connect_params,
database_name=self._pam_connect_params[0]):
return self.__wrapped__.execute(sql, parameters)
else:
with DatabaseTrace(transaction,
sql,
self._pam_dbapi2_module,
self._pam_connect_params,
database_name=self._pam_connect_params[0]):
return self.__wrapped__.execute(sql)
def callproc(self, procedure_name, parameters=DEFAULT):
transaction = current_transaction()
with DatabaseTrace(transaction, 'CALL %s' % procedure_name,
self._pam_dbapi2_module, self._pam_connect_params):
if parameters is not DEFAULT:
return self.__wrapped__.callproc(procedure_name, parameters)
else:
return self.__wrapped__.callproc(procedure_name)
def rollback(self):
transaction = current_transaction()
with DatabaseTrace(transaction,
'ROLLBACK',
self._pam_dbapi2_module,
self._pam_connect_params,
database_name=self._pam_connect_params[0]):
return self.__wrapped__.rollback()
def executescript(self, sql_script):
transaction = current_transaction()
with DatabaseTrace(transaction,
sql_script,
self._pam_dbapi2_module,
self._pam_connect_params,
database_name=self._pam_connect_params[0]):
return self.__wrapped__.executescript(sql_script)
def executemany(self, sql, seq_of_parameters):
transaction = current_transaction()
with DatabaseTrace(transaction,
sql,
self._pam_dbapi2_module,
self._pam_connect_params,
database_name=self._pam_connect_params[0]):
return self.__wrapped__.executemany(sql, seq_of_parameters)
def execute(self, sql, parameters=DEFAULT, *args, **kwargs):
transaction = current_transaction()
if parameters is not DEFAULT:
with DatabaseTrace(transaction, sql, self._pam_dbapi2_module,
self._pam_connect_params,
self._pam_cursor_params, parameters,
(args, kwargs)):
return self.__wrapped__.execute(sql, parameters, *args,
**kwargs)
else:
with DatabaseTrace(transaction,
sql,
self._pam_dbapi2_module,
self._pam_connect_params,
self._pam_cursor_params,
None, (args, kwargs),
database_name=self._pam_connect_params[0][0]):
return self.__wrapped__.execute(sql, **kwargs)
def commit(self):
transaction = current_transaction()
with DatabaseTrace(
transaction,
'COMMIT',
self._pam_dbapi2_module,
database_name=self._pam_connect_params[1]['database'],
host=self._pam_connect_params[1].get('host'),
port=self._pam_connect_params[1].get('port')):
return self.__wrapped__.commit()
def rollback(self):
transaction = current_transaction()
with DatabaseTrace(
transaction,
'ROLLBACK',
self._pam_dbapi2_module,
self._pam_connect_params,
database_name=self._pam_connect_params[1]['database'],
host=self._pam_connect_params[1].get('host'),
port=self._pam_connect_params[1].get('port')):
return self.__wrapped__.rollback()
def execute(self, sql, parameters=DEFAULT, *args, **kwargs):
transaction = current_transaction()
if parameters is not DEFAULT:
with DatabaseTrace(transaction,
sql,
self._pam_dbapi2_module,
self._pam_connect_params,
self._pam_cursor_params,
sql_parameters=parameters,
host=self._pam_connect_params[1].get('host'),
port=self._pam_connect_params[1].get('port')):
return self.__wrapped__.execute(sql, parameters, *args,
**kwargs)
else:
with DatabaseTrace(transaction,
sql,
self._pam_dbapi2_module,
self._pam_connect_params,
self._pam_cursor_params,
host=self._pam_connect_params[1].get('host'),
port=self._pam_connect_params[1].get('port')):
return self.__wrapped__.execute(sql, **kwargs)