mirror of
https://github.com/deadc0de6/dotdrop.git
synced 2026-02-05 20:48:00 +00:00
268 lines
9.0 KiB
Python
268 lines
9.0 KiB
Python
"""
|
|
author: deadc0de6 (https://github.com/deadc0de6)
|
|
Copyright (c) 2017, deadc0de6
|
|
|
|
jinja2 template generator
|
|
"""
|
|
|
|
import os
|
|
from jinja2 import Environment, FileSystemLoader, \
|
|
ChoiceLoader, FunctionLoader, TemplateNotFound, \
|
|
StrictUndefined
|
|
from jinja2.exceptions import UndefinedError
|
|
|
|
|
|
# local imports
|
|
import dotdrop.utils as utils
|
|
from dotdrop.logger import Logger
|
|
import dotdrop.jhelpers as jhelpers
|
|
from dotdrop.exceptions import UndefinedException
|
|
|
|
BLOCK_START = '{%@@'
|
|
BLOCK_END = '@@%}'
|
|
VAR_START = '{{@@'
|
|
VAR_END = '@@}}'
|
|
COMMENT_START = '{#@@'
|
|
COMMENT_END = '@@#}'
|
|
|
|
|
|
class Templategen:
|
|
|
|
def __init__(self, base='.', variables={},
|
|
func_file=[], filter_file=[], debug=False):
|
|
"""constructor
|
|
@base: directory path where to search for templates
|
|
@variables: dictionary of variables for templates
|
|
@func_file: file path to load functions from
|
|
@filter_file: file path to load filters from
|
|
@debug: enable debug
|
|
"""
|
|
self.base = base.rstrip(os.sep)
|
|
self.debug = debug
|
|
self.log = Logger()
|
|
self.variables = {}
|
|
loader1 = FileSystemLoader(self.base)
|
|
loader2 = FunctionLoader(self._template_loader)
|
|
loader = ChoiceLoader([loader1, loader2])
|
|
self.env = Environment(loader=loader,
|
|
trim_blocks=True, lstrip_blocks=True,
|
|
keep_trailing_newline=True,
|
|
block_start_string=BLOCK_START,
|
|
block_end_string=BLOCK_END,
|
|
variable_start_string=VAR_START,
|
|
variable_end_string=VAR_END,
|
|
comment_start_string=COMMENT_START,
|
|
comment_end_string=COMMENT_END,
|
|
undefined=StrictUndefined)
|
|
|
|
# adding variables
|
|
self.variables['env'] = os.environ
|
|
if variables:
|
|
self.variables.update(variables)
|
|
|
|
# adding header method
|
|
self.env.globals['header'] = self._header
|
|
# adding helper methods
|
|
if self.debug:
|
|
self.log.dbg('load global functions:')
|
|
self._load_funcs_to_dic(jhelpers, self.env.globals)
|
|
if func_file:
|
|
for f in func_file:
|
|
if self.debug:
|
|
self.log.dbg('load custom functions from {}'.format(f))
|
|
self._load_path_to_dic(f, self.env.globals)
|
|
if filter_file:
|
|
for f in filter_file:
|
|
if self.debug:
|
|
self.log.dbg('load custom filters from {}'.format(f))
|
|
self._load_path_to_dic(f, self.env.filters)
|
|
if self.debug:
|
|
self._debug_dict('template additional variables', variables)
|
|
|
|
def generate(self, src):
|
|
"""
|
|
render template from path
|
|
may raise a UndefinedException
|
|
in case a variable is undefined
|
|
"""
|
|
if not os.path.exists(src):
|
|
return ''
|
|
try:
|
|
return self._handle_file(src)
|
|
except UndefinedError as e:
|
|
err = 'undefined variable: {}'.format(e.message)
|
|
raise UndefinedException(err)
|
|
|
|
def generate_string(self, string):
|
|
"""
|
|
render template from string
|
|
may raise a UndefinedException
|
|
in case a variable is undefined
|
|
"""
|
|
if not string:
|
|
return ''
|
|
try:
|
|
return self.env.from_string(string).render(self.variables)
|
|
except UndefinedError as e:
|
|
err = 'undefined variable: {}'.format(e.message)
|
|
raise UndefinedException(err)
|
|
|
|
def add_tmp_vars(self, newvars={}):
|
|
"""add vars to the globals, make sure to call restore_vars"""
|
|
saved_variables = self.variables.copy()
|
|
if not newvars:
|
|
return saved_variables
|
|
self.variables.update(newvars)
|
|
return saved_variables
|
|
|
|
def restore_vars(self, saved_globals):
|
|
"""restore globals from add_tmp_vars"""
|
|
self.variables = saved_globals.copy()
|
|
|
|
def update_variables(self, variables):
|
|
"""update variables"""
|
|
self.variables.update(variables)
|
|
|
|
def _load_path_to_dic(self, path, dic):
|
|
mod = utils.get_module_from_path(path)
|
|
if not mod:
|
|
self.log.warn('cannot load module \"{}\"'.format(path))
|
|
return
|
|
self._load_funcs_to_dic(mod, dic)
|
|
|
|
def _load_funcs_to_dic(self, mod, dic):
|
|
"""dynamically load functions from module to dic"""
|
|
if not mod or not dic:
|
|
return
|
|
funcs = utils.get_module_functions(mod)
|
|
for name, func in funcs:
|
|
if self.debug:
|
|
self.log.dbg('load function \"{}\"'.format(name))
|
|
dic[name] = func
|
|
|
|
def _header(self, prepend=''):
|
|
"""add a comment usually in the header of a dotfile"""
|
|
return '{}{}'.format(prepend, utils.header())
|
|
|
|
def _handle_file(self, src):
|
|
"""generate the file content from template"""
|
|
try:
|
|
import magic
|
|
filetype = magic.from_file(src, mime=True)
|
|
if self.debug:
|
|
self.log.dbg('using \"magic\" for filetype identification')
|
|
except ImportError:
|
|
_, filetype = utils.run(['file', '-b', '--mime-type', src],
|
|
raw=False, debug=self.debug)
|
|
if self.debug:
|
|
self.log.dbg('using \"file\" for filetype identification')
|
|
filetype = filetype.strip()
|
|
istext = self._is_text(filetype)
|
|
if self.debug:
|
|
self.log.dbg('filetype \"{}\": {}'.format(src, filetype))
|
|
if self.debug:
|
|
self.log.dbg('is text \"{}\": {}'.format(src, istext))
|
|
if not istext:
|
|
return self._handle_bin_file(src)
|
|
return self._handle_text_file(src)
|
|
|
|
def _is_text(self, fileoutput):
|
|
"""return if `file -b` output is ascii text"""
|
|
out = fileoutput.lower()
|
|
if out.startswith('text'):
|
|
return True
|
|
if 'empty' in out:
|
|
return True
|
|
if 'json' in out:
|
|
return True
|
|
return False
|
|
|
|
def _template_loader(self, relpath):
|
|
"""manually load template when outside of base"""
|
|
path = os.path.join(self.base, relpath)
|
|
path = os.path.normpath(path)
|
|
if not os.path.exists(path):
|
|
raise TemplateNotFound(path)
|
|
with open(path, 'r') as f:
|
|
content = f.read()
|
|
return content
|
|
|
|
def _handle_text_file(self, src):
|
|
"""write text to file"""
|
|
template_rel_path = os.path.relpath(src, self.base)
|
|
try:
|
|
template = self.env.get_template(template_rel_path)
|
|
content = template.render(self.variables)
|
|
except UnicodeDecodeError:
|
|
data = self._read_bad_encoded_text(src)
|
|
content = self.generate_string(data)
|
|
return content.encode('utf-8')
|
|
|
|
def _handle_bin_file(self, src):
|
|
"""write binary to file"""
|
|
# this is dirty
|
|
if not src.startswith(self.base):
|
|
src = os.path.join(self.base, src)
|
|
with open(src, 'rb') as f:
|
|
content = f.read()
|
|
return content
|
|
|
|
def _read_bad_encoded_text(self, path):
|
|
"""decode non utf-8 data"""
|
|
with open(path, 'rb') as f:
|
|
data = f.read()
|
|
return data.decode('utf-8', 'replace')
|
|
|
|
@staticmethod
|
|
def is_template(path):
|
|
"""recursively check if any file is a template within path"""
|
|
path = os.path.expanduser(path)
|
|
if not os.path.exists(path):
|
|
return False
|
|
if os.path.isfile(path):
|
|
# is file
|
|
return Templategen._is_template(path)
|
|
for entry in os.listdir(path):
|
|
fpath = os.path.join(path, entry)
|
|
if not os.path.isfile(fpath):
|
|
# recursively explore directory
|
|
if Templategen.is_template(fpath):
|
|
return True
|
|
else:
|
|
# check if file is a template
|
|
if Templategen._is_template(fpath):
|
|
return True
|
|
return False
|
|
|
|
@staticmethod
|
|
def var_is_template(string):
|
|
"""check if variable contains template(s)"""
|
|
return VAR_START in str(string)
|
|
|
|
@staticmethod
|
|
def _is_template(path):
|
|
"""test if file pointed by path is a template"""
|
|
if not os.path.isfile(path):
|
|
return False
|
|
try:
|
|
with open(path, 'r') as f:
|
|
data = f.read()
|
|
except UnicodeDecodeError:
|
|
# is binary so surely no template
|
|
return False
|
|
markers = [BLOCK_START, VAR_START, COMMENT_START]
|
|
for marker in markers:
|
|
if marker in data:
|
|
return True
|
|
return False
|
|
|
|
def _debug_dict(self, title, elems):
|
|
"""pretty print dict"""
|
|
if not self.debug:
|
|
return
|
|
self.log.dbg('{}:'.format(title))
|
|
if not elems:
|
|
return
|
|
for k, v in elems.items():
|
|
self.log.dbg(' - \"{}\": {}'.format(k, v))
|