PurpleDome/doc_generator.py

#!/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")