mirrors/dotdrop
1
0
Fork 0
mirror of https://github.com/deadc0de6/dotdrop.git synced 2026-02-15 02:55:06 +00:00
Code Activity

adding ability to use dynamic variables in dotfile src/dst

Browse Source
This commit is contained in:
deadc0de6
2018-11-27 13:15:35 +01:00
parent d569201b0b
commit c45c17184e
5 changed files with 145 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
dotdrop/templategen.py
View File

@@ -23,7 +23,7 @@ COMMENT_END = '@@#}'
class Templategen:
def __init__(self, profile, base='.', variables={}, debug=False):
def __init__(self, profile='', base='.', variables={}, debug=False):
self.base = base.rstrip(os.sep)
self.debug = debug
self.log = Logger()
@@ -39,7 +39,8 @@ class Templategen:
comment_end_string=COMMENT_END)
# adding variables
self.env.globals['env'] = os.environ
self.env.globals['profile'] = profile
if profile:
self.env.globals['profile'] = profile
self.env.globals.update(variables)
# adding header method
self.env.globals['header'] = self._header
@@ -47,10 +48,17 @@ class Templategen:
self.env.globals['exists'] = jhelpers.exists
def generate(self, src):
"""render template from path"""
if not os.path.exists(src):
return ''
return self._handle_file(src)
def generate_string(self, string):
"""render template from string"""
if not string:
return ''
return self.env.from_string(string).render()
def _header(self, prepend=''):
"""add a comment usually in the header of a dotfile"""
return '{}{}'.format(prepend, utils.header())