lektor-render-template/lektor_render_template.py

113 lines
4 KiB
Python

# -*- 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