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()
