Reintroduce jinja2

This commit is contained in:
Joe Rickerby
2020-05-02 09:59:55 +01:00
parent 4f75eaabcd
commit c38bc9064a
2 changed files with 47 additions and 15 deletions
+16 -5
View File
@@ -1,18 +1,29 @@
import os
import io
from typing import List, Tuple
import jinja2
from typing import Union, Dict, Any
FilesDict = Dict[str, Union[str, jinja2.Template]]
class TemplateProject:
files: List[Tuple[str, str]]
default_files: FilesDict = {}
files: FilesDict
context: Dict[str, Any]
def __init__(self, files):
self.files = files
def __init__(self, *, extra_files: FilesDict):
self.files = self.default_files.copy()
self.files.update(extra_files)
self.context = {}
def generate(self, path):
for filename, content in self.files:
for filename, content in self.files.items():
file_path = os.path.join(path, filename)
os.makedirs(os.path.dirname(file_path), exist_ok=True)
with io.open(file_path, 'w', encoding='utf8') as f:
if isinstance(content, jinja2.Template):
content = content.render(self.context)
f.write(content)