136 lines
4.8 KiB
Python
136 lines
4.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
""" Import lektor plugin stuff """
|
|
from os import makedirs
|
|
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
|
|
makedirs('assets/svg/', exist_ok=True)
|
|
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)
|
|
}
|
|
)
|
|
)
|
|
)
|
|
|
|
def create_filepath(svg_label, svg_name, svg_suffix):
|
|
""" Generate File path """
|
|
if svg_label == 'ansible-galaxy':
|
|
if bool(svg_suffix):
|
|
return f"/svg/{svg_name}_{svg_suffix}.svg"
|
|
return f"/svg/{svg_name}.svg"
|
|
if bool(svg_suffix):
|
|
return f"/svg/{svg_name}_{svg_label}_{svg_suffix}.svg"
|
|
return f"/svg/{svg_name}_{svg_label}.svg"
|
|
|
|
def create_color(svg_label, svg_state, svg_color):
|
|
""" Create Color based on Input or predefined patterns """
|
|
label_color_map = {
|
|
'ansible-galaxy': '#FF6600',
|
|
'license': '#064ccf',
|
|
'maintainance': {
|
|
'well': '#cd02fa',
|
|
'true': '#8a00a6',
|
|
'poor': '#5d0070',
|
|
'false': '#220029',
|
|
'default': '#000000'
|
|
},
|
|
'default': '#e74c3c'
|
|
}
|
|
if bool(svg_color):
|
|
return svg_color
|
|
if svg_label in label_color_map:
|
|
label_item = label_color_map[svg_label]
|
|
if isinstance(label_item, dict):
|
|
return label_item.get(svg_state, label_item.get('default', '#000000'))
|
|
return label_item
|
|
return label_color_map['default']
|
|
|
|
# pylint: disable=unused-argument
|
|
def generate_svg(inputdict, **options):
|
|
""" Prepare Variables to geneate File based on a Jinja2 Template and return 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'])
|
|
if 'suffix' in dict(inputdict).keys():
|
|
svg_suffix = str(dict(inputdict)['suffix'])
|
|
else:
|
|
svg_suffix = False
|
|
if 'color' in dict(inputdict).keys():
|
|
svg_color = str(dict(inputdict)['color'])
|
|
else:
|
|
svg_color = False
|
|
|
|
filepath = create_filepath(svg_label, svg_name, svg_suffix)
|
|
color = create_color(svg_label, svg_state, svg_color)
|
|
|
|
# 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
|