Issue #208: fix - save the source data to populate the reame files.

This commit is contained in:
Steven Black 2016-12-10 23:52:42 -05:00
parent 8fbd424850
commit 962010ddcc

View File

@ -88,6 +88,7 @@ defaults = {
"targetip" : "0.0.0.0", "targetip" : "0.0.0.0",
"ziphosts" : False, "ziphosts" : False,
"sourcedatafilename" : "update.json", "sourcedatafilename" : "update.json",
"sourcedata" : [],
"readmefilename" : "readme.md", "readmefilename" : "readme.md",
"readmetemplate" : os.path.join(BASEDIR_PATH, "readme_template.md"), "readmetemplate" : os.path.join(BASEDIR_PATH, "readme_template.md"),
"readmedata" : {}, "readmedata" : {},
@ -265,6 +266,7 @@ def getUpdateURLsFromFile(source):
if os.path.exists(pathToUpdateFile): if os.path.exists(pathToUpdateFile):
updateFile = open(pathToUpdateFile, "r") updateFile = open(pathToUpdateFile, "r")
updateData = json.load(updateFile) updateData = json.load(updateFile)
settings["sourcedata"].append(updateData)
retURLs = [updateData["url"]] retURLs = [updateData["url"]]
updateFile.close() updateFile.close()
else: else:
@ -428,7 +430,8 @@ def updateReadmeData():
extensionsKey = "-".join(settings["extensions"]) extensionsKey = "-".join(settings["extensions"])
generationData = {"location": os.path.join(settings["outputsubfolder"], ""), generationData = {"location": os.path.join(settings["outputsubfolder"], ""),
"entries": settings["numberofrules"]} "entries": settings["numberofrules"],
"sourcedata": settings["sourcedata"]}
settings["readmedata"][extensionsKey] = generationData settings["readmedata"][extensionsKey] = generationData
with open(settings["readmedatafilename"], "w") as f: with open(settings["readmedatafilename"], "w") as f:
json.dump(settings["readmedata"], f) json.dump(settings["readmedata"], f)