source: mauRepo/MolesManager/trunk/src/MolesManager/molesSessionMiddleware.py @ 8155

'''
Created on 9 Jan 2012

@author: mnagni
'''
from MolesManager.moles3epb import Moles3EPB
from libs.migration.client import MigrationThread
from MolesManager.settings import RUN_MIGRATION, MIGRATION_INTERVAL,\
    MIGRATION_DB_CONNECTION, MIGRATION_DB_SCRIPT, MOLES3_DB_CONNECTION,\
    MOLES3_DB_SCRIPT
from libs.commons_db import DbManager
from libs.migration.MigrationEPB import MigrationEPB


class MolesSessionMiddleware(object):
    
    #This attribute should be not here.
    #unfortunately I cannot find any start/stop signals from Django
    _migrationThread = MigrationThread(interval = MIGRATION_INTERVAL)
    _epbInitialized = False
    

    
    def _getMolesSession(self):
        if not MolesSessionMiddleware._epbInitialized:
            migrationDB = DbManager(MIGRATION_DB_CONNECTION, MIGRATION_DB_SCRIPT)
            MigrationEPB.overrrideDBManager(migrationDB)       

            molesDB = DbManager(MOLES3_DB_CONNECTION, MOLES3_DB_SCRIPT)
            Moles3EPB.overrrideDBManager(molesDB)
            MolesSessionMiddleware._epbInitialized = True
            return Moles3EPB.getNewMolesSession()
        else:
            return Moles3EPB.getNewMolesSession()
    
    def _migration(self, runMigration = RUN_MIGRATION):
        if runMigration and not MolesSessionMiddleware._migrationThread.isAlive():
            #t.setDaemon(False)
            MolesSessionMiddleware._migrationThread.start()
            
    """
        Represents the access to the Moles database.
        Creates from the existing db connections pools a new session
        to be used by all the DB operations involved in the actual HTTPRequest
    """  
    
    def process_request(self, request):        
        self._migration() #see the note on  MolesSessionMiddleware._migration
        
        request.moles_session = self._getMolesSession()


    def process_response(self, request, response):
        if hasattr(request, 'moles_session'):
            request.moles_session.close()

        return response


    def process_exception(self, request, exception):
        try:
            session = request.moles_session            
        except AttributeError:
            return
        session.rollback()
        session.close()