From b7795560ca10404dbcced133a4be9ba8a588a7a4 Mon Sep 17 00:00:00 2001 From: Rishi Mondal <146999057+MAVRICK-1@users.noreply.github.com> Date: Thu, 8 Feb 2024 08:13:07 +0530 Subject: [PATCH] Requested changes --- iyp/crawlers/rovista/validating_rov.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/iyp/crawlers/rovista/validating_rov.py b/iyp/crawlers/rovista/validating_rov.py index 2581307..1e8484b 100644 --- a/iyp/crawlers/rovista/validating_rov.py +++ b/iyp/crawlers/rovista/validating_rov.py @@ -40,21 +40,21 @@ def run(self): # Break the loop if there's no more data if len(data) < batch_size: break - logging.info('Pushing nodes to neo4j...\n') + logging.info('Pushing nodes to neo4j...') # get ASNs and prefixes IDs self.asn_id = self.iyp.batch_get_nodes_by_single_prop('AS', 'asn', asns) - tag_id_not_vali = self.iyp.get_node('Tag', {'label': 'Not Validating RPKI ROV'}, create=True) - tag_id_vali = self.iyp.get_node('Tag', {'label': 'Validating RPKI ROV'}, create=True) + tag_id_not_valid = self.iyp.get_node('Tag', {'label': 'Not Validating RPKI ROV'}, create=True) + tag_id_valid = self.iyp.get_node('Tag', {'label': 'Validating RPKI ROV'}, create=True) # Compute links links = [] for entry in entries: asn_qid = self.asn_id[entry['asn']] if entry['ratio'] > 0.5: - links.append({'src_id': asn_qid, 'dst_id': tag_id_vali, 'props': [self.reference, entry]}) + links.append({'src_id': asn_qid, 'dst_id': tag_id_valid, 'props': [self.reference, {'ratio': entry['ratio']}) else: - links.append({'src_id': asn_qid, 'dst_id': tag_id_not_vali, 'props': [self.reference, entry]}) + links.append({'src_id': asn_qid, 'dst_id': tag_id_not_valid, 'props': [self.reference, {'ratio': entry['ratio']}) - logging.info('Pushing links to neo4j...\n') + logging.info('Pushing links to neo4j...') # Push all links to IYP self.iyp.batch_add_links('CATEGORIZED', links)