mirrors/dotdrop
1
0
Fork 0
mirror of https://github.com/deadc0de6/dotdrop.git synced 2026-02-05 14:23:48 +00:00
Code Activity

Don't blindly sum 1 to get a relative path

Browse Source
This commit is contained in:
Pedro Algarvio
2018-01-04 15:31:01 +00:00
parent 2b2306e00a
commit 06ef339d1b
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
dotdrop/templategen.py
View File

@@ -21,7 +21,7 @@ COMMENT_END = '@@#}'
class Templategen:
def __init__(self, base='.'):
self.base = base
self.base = base.rstrip(os.sep)
loader = FileSystemLoader(self.base)
self.env = Environment(loader=loader,
trim_blocks=True, lstrip_blocks=True,
@@ -47,9 +47,9 @@ class Templategen:
return self._handle_text_file(src, profile)
def _handle_text_file(self, src, profile):
length = len(self.base) + 1
template_rel_path = src.split(self.base)[-1]
try:
template = self.env.get_template(src[length:])
template = self.env.get_template(template_rel_path)
content = template.render(profile=profile, env=os.environ)
except UnicodeDecodeError:
data = self._read_bad_encoded_text(src)