def receive_message(self): self.current_message = self.source_pipeline.receive() if not self.current_message: return None message = self.current_message.decode('utf-8') try: # Event Object return Event.from_unicode(message) except: # Report Object return message
def handle_message(self, xmpp_connection, message): try: event = Event.from_unicode(unicode(message.getBody())) for key in event.keys(): value = event.value(key) event.clear(key) key = key.replace(' ','_') event.add(key, value) self.send_message(event) except: pass
def process(self): record_list = self.receive_message() query = "delete from tor_network" self.cur.execute(query) record_list=ast.literal_eval(record_list) for record in record_list: eventdict = Event.from_unicode(record).to_dict() keys = ", ".join(eventdict.keys()) values = eventdict.values() fvalues = len(values) * "%s, " query = "INSERT INTO tor_network (" + keys + ") VALUES (" + fvalues[:-2] + ")" self.logger.info(values) self.cur.execute(query, values) self.con.commit() self.acknowledge_message()