From 52dd1c4a670d725e72bbd4993454232aca77cd91 Mon Sep 17 00:00:00 2001 From: Alexandre Dulaunoy Date: Mon, 24 May 2021 17:04:03 +0200 Subject: [PATCH] chg: [importer] parent option added - to reference an parent CyCAT OID --- backend/bin/importer.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/backend/bin/importer.py b/backend/bin/importer.py index cd26d26..4266cb3 100644 --- a/backend/bin/importer.py +++ b/backend/bin/importer.py @@ -5,6 +5,7 @@ import os parser = argparse.ArgumentParser(description='JSON importer for CyCAT backend') parser.add_argument('-f', '--file', help='JSON file to import') parser.add_argument('-t', '--type', help='CyCAT backend type', default=1) +parser.add_argument('-p', '--parent', help='Parent reference to add for the JSON imported', default=None) args = parser.parse_args() r = redis.Redis(host='127.0.0.1', port='3033') @@ -32,5 +33,8 @@ elif int(args.type) == 2: k = "t:{}".format(args.type) r.zadd(k, d, nx=False) r.hmset("{}:{}".format(args.type, uuid), record) + if args.parent is not None: + r.sadd("parent:{}".format(uuid), args.parent) + r.sadd("child:{}".format(args.parent), uuid) else: pass