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

Merge pull request #16 from s0undt3ch/master

Browse Source 
Don't blindly sum 1 to get a relative path
This commit is contained in:
deadc0de
2018-01-05 08:52:01 +01:00
committed by GitHub
parent 2b2306e00a 06ef339d1b
commit deadab5d3a
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: class Templategen:
def __init__(self, base='.'): def __init__(self, base='.'):
self.base = base self.base = base.rstrip(os.sep)
loader = FileSystemLoader(self.base) loader = FileSystemLoader(self.base)
self.env = Environment(loader=loader, self.env = Environment(loader=loader,
trim_blocks=True, lstrip_blocks=True, trim_blocks=True, lstrip_blocks=True,
@@ -47,9 +47,9 @@ class Templategen:
return self._handle_text_file(src, profile) return self._handle_text_file(src, profile)
def _handle_text_file(self, src, profile): def _handle_text_file(self, src, profile):
length = len(self.base) + 1 template_rel_path = src.split(self.base)[-1]
try: try:
template = self.env.get_template(src[length:]) template = self.env.get_template(template_rel_path)
content = template.render(profile=profile, env=os.environ) content = template.render(profile=profile, env=os.environ)
except UnicodeDecodeError: except UnicodeDecodeError:
data = self._read_bad_encoded_text(src) data = self._read_bad_encoded_text(src)