mirror of https://github.com/avast/PurpleDome
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
42 lines
1.8 KiB
Python
42 lines
1.8 KiB
Python
#!/usr/bin/env python3
|
|
|
|
# A standalon document generator. Takes an attack log and generates a doc using templates. Functionality will later be merged into PurpleDome
|
|
|
|
import json
|
|
from jinja2 import Environment, FileSystemLoader, select_autoescape
|
|
# from pprint import pprint
|
|
|
|
|
|
def generate(jfile, outfile):
|
|
env = Environment(
|
|
loader=FileSystemLoader("templates", encoding='utf-8', followlinks=False),
|
|
autoescape=select_autoescape(),
|
|
trim_blocks=True,
|
|
lstrip_blocks=True
|
|
)
|
|
template = env.get_template("attack_description.rst")
|
|
|
|
with open(jfile) as fh:
|
|
events = json.load(fh)
|
|
|
|
print(template.render(events=events))
|
|
# pprint(events)
|
|
# dest = os.path.join(self.get_plugin_path(), "filebeat.conf")
|
|
# with open(dest, "wt") as fh:
|
|
# res = template.render({"playground": self.get_playground()})
|
|
# fh.write(res)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
# generate("loot/2021_07_19___16_28_45/attack.json", "tools/human_readable_documentation/contents.rst") # Working example for a short run
|
|
# generate("loot/2021_07_20___08_26_33/attack.json", "tools/human_readable_documentation/contents.rst") # FIN 7 #1
|
|
# generate("loot/2021_07_20___10_07_36/attack.json", "tools/human_readable_documentation/contents.rst") # FIN 7 #2 The one Fabrizio got
|
|
#generate("loot/2021_07_28___12_09_00/attack.json",
|
|
# "tools/human_readable_documentation/contents.rst") # FIN 7 The last minute locally generated thing
|
|
|
|
generate("loot/2021_08_30___14_40_23/attack.json",
|
|
"tools/human_readable_documentation/contents.rst") # FIN 7 With genereated files added
|
|
|
|
# generate("loot/2021_07_19___15_10_45/attack.json", "tools/human_readable_documentation/contents.rst")
|
|
# generate("removeme.json", "tools/human_readable_documentation/contents.rst")
|