import sys from diagnosis.preprocess.preprocessor import Preprocessor from diagnosis.preprocess.utils import utils from diagnosis.preprocess.event import config class ReadFile(Preprocessor): config: config.ReadFileConfig def __init__(self, configs, data): super(ReadFile, self).__init__(configs, data) def execute(self): result = [] try: result = utils.read_csv(self.config.data[0]) except: raise Exception("read event file failed", sys.exc_info()[1]) else: return result class ConvertEventsToTuple(Preprocessor): config: config.ConvertEventsToTupleConfig def __init__(self, configs, data): super(ConvertEventsToTuple, self).__init__(configs, data) def execute(self): result = [] try: result = utils.convert_list_to_tuple(self.config.data[0]) except: raise Exception("convert events failed", sys.exc_info()[1]) else: return result class CreateEventsSqlite(Preprocessor): config: config.CreateEventsSqliteConfig def __init__(self, configs, data): super(CreateEventsSqlite, self).__init__(configs, data) def execute(self): try: utils.create_events_sqlite( self.config.data[0], self.config.db_name, self.config.table_name) except: raise Exception("create metrics sqlite failed", sys.exc_info()[1])