diff --git a/app/doc_generator.py b/app/doc_generator.py index 9abc4ff..29dbb37 100644 --- a/app/doc_generator.py +++ b/app/doc_generator.py @@ -6,6 +6,7 @@ import json import os from jinja2 import Environment, FileSystemLoader, select_autoescape + class DocGenerator(): """ Generates human readable docs from attack logs """ diff --git a/app/experimentcontrol.py b/app/experimentcontrol.py index cbd6320..abfcfcd 100644 --- a/app/experimentcontrol.py +++ b/app/experimentcontrol.py @@ -19,7 +19,6 @@ from machine_control import Machine from plugins.base.attack import AttackPlugin - # TODO: Multi threading at least when starting machines class Experiment(): diff --git a/doc_generator.py b/doc_generator.py index adae904..c164c9f 100755 --- a/doc_generator.py +++ b/doc_generator.py @@ -2,9 +2,6 @@ # A standalone 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 import argparse from app.doc_generator import DocGenerator @@ -25,4 +22,4 @@ if __name__ == "__main__": arguments = create_parser().parse_args() dg = DocGenerator() - dg.generate(arguments.attack_log, arguments.outfile) \ No newline at end of file + dg.generate(arguments.attack_log, arguments.outfile)