hello_extension.hello_extension

Complexity

Total Complexity

3

Size/Duplication

Total Lines	25
Duplicated Lines	0 %

Importance

Changes

0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	HelloExtension.parse()	0	6	1
A	HelloExtension._hello()	0	3	1
A	HelloExtension.__init__()	0	3	1

"""Provides custom extension, exposing a ``hello`` command."""
from jinja2 import nodes
from jinja2.ext import Extension


class HelloExtension(Extension):
    """Simple jinja2 extension for cookiecutter test purposes."""

    tags = {'hello'}

    def __init__(self, environment):
        """Hello Extension Constructor."""
        super().__init__(environment)

    def _hello(self, name):
        """Do actual tag replace when invoked by parser."""
        return f'Hello {name}!'

    def parse(self, parser):
        """Work when something match `tags` variable."""
        lineno = next(parser.stream).lineno
        node = parser.parse_expression()
        call_method = self.call_method('_hello', [node], lineno=lineno)
        return nodes.Output([call_method], lineno=lineno)