source: mauRepo/MolesManager/trunk/src/libs/migration/processor/migrationProcess.py @ 8325

'''
Created on 10 Jan 2012

@author: mnagni
'''
from libs.migration.processor.dataEntity import DataEntityProcessor
from libs.migration.MigrationEPB import MigrationEPB
import logging
from logging import StreamHandler
from libs.migration.processor.EPBRepo import EPBRepo

class MigrationProcess():
    log = logging.getLogger('MigrationProcess')
    log.addHandler(StreamHandler())
    log.setLevel(logging.DEBUG)
    
    """
        Manages the migration process from Moles2 to Moles3.
    """
    def __init__(self):
        pass
    
    def process(self, dataEntityObject = None):
        """
            Starts the migration. If no DataEntityMigration instance is given then all the DataEntityMigration objects in the 
            data_entity_migration table are processed
            1) each DataEntityMigration is associated to a "dataent_xxxx" file, called "data entity" in Moles2
            2) each "data entity" is associated with a CEDA_ObservationCollection
            @param dataEntityObject: a DataEntityMigration instance to process.
            @return: a list of collected exceptions
        """
        exs = []
        data_ents = []
        data_ents.append(EPBRepo.migrationEPB.getDataEntityMigrationbyPath(dataEntityObject))
        """
        #Does the migrationObject exists?
        if dataEntityObject:
            data_ents.append(MigrationEPB.getDataEntityMigrationbyPath(dataEntityObject, session = self.migrationSessions.migrationSession))
        else: #then process all the DataEntities
            data_ents = MigrationEPB.getAllDataEntityMigration(session = self.migrationSessions.migrationSession)
        """
            
        #loops over the data entities
        for dataEntityMigration in [f for f in data_ents if f != None]:                         
            dep = DataEntityProcessor(dataEntityMigration)            
            exs.extend(dep.process())
            MigrationProcess.log.info("Done")            
        return exs 
        
    def commitAll(self):
        self._migrationSession.commit()
        self._molesSession.commit()

    def rollbackAll(self):
        self._migrationSession.rollback()
        self._molesSession.rollback()