def __init__(self, *args, **kwargs):
super(engine, self).__init__(*args, **kwargs)
self.entity_link_manager = Entitylink()
self.db_configuration_manager = DBConfiguration()
self.reload_configuration()
class engine(Engine):
etype = 'linklist'
def __init__(self, *args, **kwargs):
super(engine, self).__init__(*args, **kwargs)
self.entity_link_manager = Entitylink()
self.db_configuration_manager = DBConfiguration()
self.reload_configuration()
def beat(self):
self.reload_configuration()
def reload_configuration(self):
configuration = self.db_configuration_manager.find_one(
query={'crecord_type': 'linklistfieldsurl'}
)
if configuration is not None:
self.link_field = configuration['field_list']
self.logger.debug(
'searching for urls in events fields : {}'.format(
self.link_field
)
)
def work(self, event, *args, **kwargs):
"""
Check if event contains any url interesting keys.
When one of self.link_field found, it is added to the event entity link
"""
link_key = 'event_links'
event_links = {}
for key in self.link_field:
if key in event and event[key]:
event_links[key] = event[key]
if event_links:
self.logger.debug('found links into the event {}: {}'.format(
event['rk'],
event_links
))
# Proceed link update in case some link found in the event
entity = self.entity_link_manager.get_or_create_from_event(event)
_id = entity['_id']
links_to_integrate = {}
# integrate previous links if any
if link_key in entity:
for link in entity[link_key]:
links_to_integrate[link['label']] = link['url']
# New event link definition upsert (override)
links_to_integrate.update(event_links)
# Define new data context as in upsert mode
context = {link_key: []}
for key in links_to_integrate:
context[link_key].append({
'label': key,
'url': links_to_integrate[key]
})
# Push changes to db
self.entity_link_manager.put(_id, context)