Produits Entr'ouvert » Authentic 2

# authentic2 - versatile identity manager

# Copyright (C) 2010-2020 Entr'ouvert

#

# This program is free software: you can redistribute it and/or modify it

# under the terms of the GNU Affero General Public License as published

# by the Free Software Foundation, either version 3 of the License, or

# (at your option) any later version.

#

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

# GNU Affero General Public License for more details.

#

# You should have received a copy of the GNU Affero General Public License

# along with this program.  If not, see <http://www.gnu.org/licenses/>.

from django.template import VariableDoesNotExist as DjangoVariableDoesNotExist

from django.template import engines

from django.template import TemplateSyntaxError as DjangoTemplateSyntaxError

from django.utils.encoding import force_str

from django.utils.translation import ugettext_lazy as _

class TemplateError(Exception):

    pass

class Template(object):

    def __init__(self, value, raises=False, autoescape=False):

        self.value = value

        self.raises = raises

        if not autoescape:

            self.value = '{%% autescape off %%}%s{%% endautoescape %%}' % \

                    self.value

        try:

            self.template = engines['django'].from_string(value)

        except DjangoTemplateSyntaxError as e:

            if self.raises:

                raise TemplateError(_('template syntax error: %s') % e)

            # no rendering

            self.render = lambda x,y: str(x.value)

        else:

            self.render = self.django_render

    def django_render(self, context={}):

        excepted = False

        try:

            rendered = self.template.render(context)

        except DjangoTemplateSyntaxError as e:

            if raises:

                raise TemplateError(_('template syntax error: %s') % e)

        except DjangoVariableDoesNotExist as e:

            if raises:

                raise TemplateError(_('template variable missing: %s') % e)

        else:

            return force_str(rendered)

        return force_str(self.value)