repytex/notes_tools/generate_bilan/texenv.py

#!/usr/bin/env python
# encoding: utf-8

import jinja2, os
from .filters import includegraphics

import logging
logger = logging.getLogger("generate_bilan")

__all__ = ["texenv"]

# Definition of jinja syntax for latex
texenv = jinja2.Environment(
    block_start_string = '\Block{',
    # Gros WTF!! Si on le met en maj ça ne marche pas alors que c'est en maj dans le template...
    block_end_string = '}',
    variable_start_string = '\Var{',
    variable_end_string = '}',
    comment_start_string = '\#{',
    comment_end_string = '}',
    line_statement_prefix = '%-',
    line_comment_prefix = '%#',
    loader = jinja2.ChoiceLoader([jinja2.PackageLoader("notes_tools.generate_bilan", "templates"),
        jinja2.FileSystemLoader(['./']),
        ]),
    extensions = ['jinja2.ext.do']
)

# Filters

from .filters import includegraphics
texenv.filters['includegraphics'] = includegraphics

from notes_tools.tools.marks_plottings import radar_on, marks_hist, parallele_on
texenv.filters['radar_on'] = radar_on
texenv.filters['marks_hist'] = marks_hist
texenv.filters['parallele_on'] = parallele_on

def feed_template(target, datas, template):
    """ Get the template and feed it to create bilans

    :param target: path where the bilan will be saved
    :param datas: dictonnary to feed the template
    :param template: the template
    """
    logger.info("Getting template {}".format(template))
    bilan = texenv.get_template(template)

    path_to_target = target.dirname()
    if not path_to_target.exists():
        path_to_target.mkdir()

    with open(target, "w") as f:
        f.write(bilan.render(**datas, directory=path_to_target))
    logger.info("{} est construit! Ya plus qu'à compiler!".format(target))
if __name__ == '__main__':
    print(texenv.list_templates())
    texenv.get_template("tpl_bilan.tex")



# -----------------------------
# Reglages pour 'vim'
# vim:set autoindent expandtab tabstop=4 shiftwidth=4:
# cursor: 16 del