"""Module to contain different types of methods."""
from importlib.util import find_spec
from pathlib import Path

from jinja2 import (
    ChoiceLoader,
    Environment,
    FileSystemLoader,
    PackageLoader,
    select_autoescape,
)


def jinja_loader(module_name):
    """Renders template to provide formatted output."""
    options = {}
    spec = find_spec(module_name)

    if spec is None or spec.origin is None:
        loader = PackageLoader(module_name)
    else:
        path = Path(spec.origin).resolve(strict=True)
        loader = FileSystemLoader(str(path.parent / "templates"))
    options["loader"] = ChoiceLoader([loader, PackageLoader(module_name)])
    if "autoescape" not in options:
        options["autoescape"] = select_autoescape(["html", "htm", "xml"])

    # bandit doesn't detect if we set "autoescape" dynamically
    env = Environment(**options)
    return env