From 7208502bfb252331e084a7807d1222445aa0ed90 Mon Sep 17 00:00:00 2001 From: Sebastian Lohff Date: Sun, 14 Jun 2020 00:24:18 +0200 Subject: [PATCH] Save graph in crawl converter --- bin/conv.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/bin/conv.py b/bin/conv.py index a5ae97f..526d71b 100755 --- a/bin/conv.py +++ b/bin/conv.py @@ -7,7 +7,7 @@ os.environ.setdefault("DJANGO_SETTINGS_MODULE", "dnmapper.settings") import django django.setup() -from backend.crawler import convert_crawl +from backend.crawler import convert_crawl, net_to_json from bgpdata.models import CrawlRun @@ -21,7 +21,13 @@ def main(): except CrawlRun.DoesNotExist: parser.error("CrawlRun with id {} does not exist".format(args.crawl_id)) - convert_crawl(crawl) + net = convert_crawl(crawl) + if net.nodes and net.edges: + crawl.graph = net_to_json(net) + crawl.save() + print("Crawl updated") + else: + print("Crawl had no nodes or edges, abort") if __name__ == '__main__':