def setup(self): self.dbwriter = DBWriter(self.server.results_file_path)
class LogCollector(SocketServer.BaseRequestHandler): def setup(self): self.dbwriter = DBWriter(self.server.results_file_path) def handle(self): str_message = self.request[0].strip() dic_message = json.loads(str_message) dic_payload = dic_message.get('payload', {}) if dic_message.get('dataType') == 'transactionStart': self.dbwriter.write_transaction_start(dic_payload) elif dic_message.get('dataType') == 'transactionEnd': self.dbwriter.write_transaction_end(dic_payload) elif dic_message.get('dataType') == 'request': self.dbwriter.write_request(dic_payload) elif dic_message.get('dataType') == 'spawns': self.dbwriter.write_spawns_info(dic_payload) elif dic_message.get('dataType') == 'sysmonitor': self.dbwriter.write_sysmonitor_info(dic_payload) elif dic_message.get('dataType') == 'message': self.dbwriter.write_message(dic_payload)