Strip quotes and ensure file_path is typed as a list

This commit is contained in:
Wes Lambert
2022-07-05 14:25:54 +00:00
parent 7fa9ca8fc6
commit e96206d065

View File

@@ -17,7 +17,7 @@ def searchFile(artifact, csvfiles):
dir = os.path.dirname(os.path.realpath(__file__)) dir = os.path.dirname(os.path.realpath(__file__))
found = [] found = []
for f in csvfiles: for f in csvfiles:
filename = dir + "/" + f filename = dir + "/" + f.strip("'")
with open(filename, "r") as csvfile: with open(filename, "r") as csvfile:
csvdata = csv.DictReader(csvfile) csvdata = csv.DictReader(csvfile)
for row in csvdata: for row in csvdata:
@@ -58,7 +58,7 @@ def analyze(conf, input):
meta = helpers.loadMetadata(__file__) meta = helpers.loadMetadata(__file__)
data = helpers.parseArtifact(input) data = helpers.parseArtifact(input)
helpers.checkSupportedType(meta, data["artifactType"]) helpers.checkSupportedType(meta, data["artifactType"])
search = searchFile(data["value"], conf['file_path']) search = searchFile(data["value"], conf['file_path'].strip("[]").split(', '))
results = prepareResults(search) results = prepareResults(search)
return results return results