diff --git a/bin/import.py b/bin/import.py index 027e493..2f34479 100644 --- a/bin/import.py +++ b/bin/import.py @@ -12,10 +12,22 @@ if args.f is not None: md5 = args.f[0].split(".")[0] r.sadd('processed', md5) lnumber = 0 + fields = None for line in sys.stdin: if lnumber == 0: - for field in line.rstrip().split(","): + fields = line.rstrip().split(",") + for field in fields: r.sadd('type', field) + else: + elements = line.rstrip().split(",") + i = 0 + for element in elements: + try: + r.sadd('e:'+fields[i], element) + except IndexError: + print("Empty fields") + i = i+1 + lnumber = lnumber + 1 else: argParser.print_help()