lektor-render-template/lektor_render_template.py

# -*- coding: utf-8 -*-
""" Import lektor plugin stuff """
from lektor.pluginsystem import Plugin
from lektor.project import Project
from lektor.context import get_ctx

from scour.scour import scourString
from scour.scour import sanitizeOptions as sanitizeScourOptions
from scour.scour import parse_args as parseScourArgs

def optimize_svg(source_svg):
    """ using scour to optimize and minify svg """
    scour_options = parseScourArgs([
        "--enable-id-stripping",
        "--enable-comment-stripping",
        "--shorten-ids",
        "--indent=none",
        "--no-line-breaks"])
    scour_options = sanitizeScourOptions(scour_options)
    optimized_svg = scourString(source_svg, scour_options)
    return optimized_svg


# pylint: disable=too-many-arguments
def render_template(env, pad, filepath, label_name, label_value, color):
    """ Render my custom Jinja2 Template 'galaxy.svg' with the given variables as input and pass it to a svg optimizer """
    # Prepare Final Variables for galaxy.svg
    font_spacing = int(7)
    rendered_title = f"{label_value}: {label_name}"
    center_position = int(len(label_value))*font_spacing+15
    right_width = int(len(label_name))*font_spacing
    overal_width = int((len(label_value)+len(label_name))*font_spacing)+int(31)
    rendered_sizes = f"width=\"{overal_width}\" height=\"20\" viewBox=\"0 0 {overal_width} 20\""
    accent_multiplier = int((overal_width - 121)/font_spacing)
    right_box_position = float(int(overal_width - 132)-0.007)

    # Create an optimized SVG File
    with open(f"assets/{filepath}", 'w', encoding='UTF-8') as outputfile:
        outputfile.write(
          optimize_svg(
            env.render_template(
              'galaxy.svg', pad=pad, this={
                'title': str(rendered_title),
                'font_spacing': str(font_spacing),
                'center_position': str(center_position),
                'right_width': str(right_width),
                'right_box_position': str(right_box_position),
                'latest_accent': str(right_box_position),
                'name': str(label_name),
                'accent_multiplier': str(accent_multiplier),
                "sizes": str(rendered_sizes),
                "inputvalue": str(label_value),
                "color_higlight": str(color)
              }
            )
          )
        )

# pylint: disable=unused-argument
def generate_svg(inputdict, **options):
    """ Prepare Variables to geneate File based on a Jinja2 Template and returning its filepath """
    # Setup Lektor Env, CTX and Pad
    project = Project.discover()
    env = project.make_env()
    ctx = get_ctx()
    if ctx is not None:
        pad = ctx.pad
    else:
        pad = env.new_pad()

    # Sorting Input
    svg_name = str(dict(inputdict)['name'])
    svg_state = str(dict(inputdict)['state'])
    svg_label = str(dict(inputdict)['label'])

    # File Path
    if svg_label == 'ansible-galaxy':
        filepath =  f"/svg/{svg_name}.svg"
    else:
        filepath =  f"/svg/{svg_name}_{svg_label}.svg"

    # File Color
    match svg_label:
        case 'ansible-galaxy':
            color = '#FF6600'
        case 'license':
            color = '#064ccf'
        case 'maintainance':
            match svg_state:
                case 'well':
                    color = '#cd02fa'
                case 'true':
                    color = '#8a00a6'
                case 'poor':
                    color = '#5d0070'
                case 'false':
                    color = '#220029'
                case _:
                    color = '#000000'
        case _:
            color = '#e74c3c'

    # Render Template and Return Filepath
    render_template(env, pad, filepath, svg_state, svg_label, color)
    return filepath

class RenderTemplatePlugin(Plugin):
    """ Create Plugin and listen to Filter Keyword """
    name = 'lektor_render_template'
    description = 'Lektor plugin to generate files form filters.'
    def on_setup_env(self, **extra):
        """ Create Listener for Filter """
        self.env.jinja_env.filters['svg'] = generate_svg