Starting some code around creating reports.

1 files changed, 29 insertions, 0 deletions

diff --git a/src/ee/project/report.py b/src/ee/project/report.py
new file mode 100644
index 0000000..5370f6c
--- /dev/null
+++ b/src/ee/project/report.py
@@ -0,0 +1,29 @@
+from pathlib import Path
+from typing import Optional
+
+from jinja2 import Environment, PackageLoader, select_autoescape
+
+from ee.tools import mk_parents
+
+
+class Message(object):
+    def __init__(self, object: Optional[str] = None, text: Optional[str] = None):
+        self.object = object
+        self.text = text
+
+
+def subsection_filter(s: str):
+    return "{}\n{}".format(s, "~" * len(s))
+
+
+def save_report(package: str, template_name: str, out_file: Path, **kwargs):
+    env = Environment(
+        loader=PackageLoader(package, "templates"),
+        autoescape=select_autoescape(["html", "xml"]),
+    )
+    env.filters["subsection"] = subsection_filter
+
+    mk_parents(out_file)
+    with out_file.open("w") as f:
+        template = env.get_template(template_name)
+        f.write(template.render(**kwargs))