diff --git a/tailbone/app.py b/tailbone/app.py index ab96fa7f..99f3d8cc 100644 --- a/tailbone/app.py +++ b/tailbone/app.py @@ -89,6 +89,8 @@ def make_rattail_config(settings): tailbone.db.Session.configure(bind=rattail_engines['default']) if hasattr(rattail_config, 'tempmon_engine'): tailbone.db.TempmonSession.configure(bind=rattail_config.tempmon_engine) + if hasattr(rattail_config, 'trainwreck_engine'): + tailbone.db.TrainwreckSession.configure(bind=rattail_config.trainwreck_engine) # Make sure rattail config object uses our scoped session, to avoid # unnecessary connections (and pooling limits). diff --git a/tailbone/db.py b/tailbone/db.py index 27579da3..667f5ebe 100644 --- a/tailbone/db.py +++ b/tailbone/db.py @@ -39,6 +39,7 @@ Session = scoped_session(sessionmaker(class_=SessionBase, rattail_config=None, r # not necessarily used, but here if you need it TempmonSession = scoped_session(sessionmaker()) +TrainwreckSession = scoped_session(sessionmaker()) class TailboneSessionDataManager(datamanager.SessionDataManager): @@ -148,6 +149,8 @@ def register(session, initial_state=datamanager.STATUS_ACTIVE, if tuple(int(x) for x in sa.__version__.split('.')) >= (0, 7): register(Session) register(TempmonSession) + register(TrainwreckSession) else: Session.configure(extension=ZopeTransactionExtension()) TempmonSession.configure(extension=ZopeTransactionExtension()) + TrainwreckSession.configure(extension=ZopeTransactionExtension())