Skip to content

Commit

Permalink
[CLI] Refactor painter arguments to table instead of commandline (#24456
Browse files Browse the repository at this point in the history
)

Co-authored-by: Nick Brassel <[email protected]>
  • Loading branch information
elpekenin and tzarc authored Nov 21, 2024
1 parent 9c86583 commit 88afd53
Show file tree
Hide file tree
Showing 3 changed files with 30 additions and 10 deletions.
4 changes: 1 addition & 3 deletions lib/python/qmk/cli/painter/convert_graphics.py
Original file line number Diff line number Diff line change
Expand Up @@ -60,9 +60,7 @@ def painter_convert_graphics(cli):
return

# Work out the text substitutions for rendering the output data
args_str = " ".join((f"--{arg} {getattr(cli.args, arg.replace('-', '_'))}" for arg in ["input", "output", "format", "no-rle", "no-deltas"]))
command = f"qmk painter-convert-graphics {args_str}"
subs = generate_subs(cli, out_bytes, image_metadata=metadata, command=command)
subs = generate_subs(cli, out_bytes, image_metadata=metadata, command_name="painter_convert_graphics")

# Render and write the header file
header_text = render_header(subs)
Expand Down
4 changes: 1 addition & 3 deletions lib/python/qmk/cli/painter/make_font.py
Original file line number Diff line number Diff line change
Expand Up @@ -61,10 +61,8 @@ def painter_convert_font_image(cli):
return

# Work out the text substitutions for rendering the output data
args_str = " ".join((f"--{arg} {getattr(cli.args, arg.replace('-', '_'))}" for arg in ["input", "output", "no-ascii", "unicode-glyphs", "format", "no-rle"]))
command = f"qmk painter-convert-font-image {args_str}"
metadata = {"glyphs": _generate_font_glyphs_list(not cli.args.no_ascii, cli.args.unicode_glyphs)}
subs = generate_subs(cli, out_bytes, font_metadata=metadata, command=command)
subs = generate_subs(cli, out_bytes, font_metadata=metadata, command_name="painter_convert_font_image")

# Render and write the header file
header_text = render_header(subs)
Expand Down
32 changes: 28 additions & 4 deletions lib/python/qmk/painter.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@
import datetime
import math
import re
from pathlib import Path
from string import Template
from PIL import Image, ImageOps

Expand Down Expand Up @@ -137,18 +138,40 @@ def _render_image_metadata(metadata):
return "\n".join(lines)


def generate_subs(cli, out_bytes, *, font_metadata=None, image_metadata=None, command):
def command_args_str(cli, command_name):
"""Given a command name, introspect milc to get the arguments passed in."""

args = {}
max_length = 0
for arg_name, was_passed in cli.args_passed[command_name].items():
max_length = max(max_length, len(arg_name))

val = getattr(cli.args, arg_name.replace("-", "_"))

# do not leak full paths, keep just file name
if isinstance(val, Path):
val = val.name

args[arg_name] = val

return "\n".join(f"// {arg_name.ljust(max_length)} | {val}" for arg_name, val in args.items())


def generate_subs(cli, out_bytes, *, font_metadata=None, image_metadata=None, command_name):
if font_metadata is not None and image_metadata is not None:
raise ValueError("Cant generate subs for font and image at the same time")

args = command_args_str(cli, command_name)

subs = {
"year": datetime.date.today().strftime("%Y"),
"input_file": cli.args.input.name,
"sane_name": re.sub(r"[^a-zA-Z0-9]", "_", cli.args.input.stem),
"byte_count": len(out_bytes),
"bytes_lines": render_bytes(out_bytes),
"format": cli.args.format,
"generator_command": command,
"generator_command": command_name.replace("_", "-"),
"command_args": args,
}

if font_metadata is not None:
Expand All @@ -167,7 +190,7 @@ def generate_subs(cli, out_bytes, *, font_metadata=None, image_metadata=None, co
subs.update({
"generated_type": "image",
"var_prefix": "gfx",
"generator_command": command,
"generator_command": command_name,
"metadata": _render_image_metadata(image_metadata),
})

Expand All @@ -183,7 +206,8 @@ def generate_subs(cli, out_bytes, *, font_metadata=None, image_metadata=None, co
// Copyright ${year} QMK -- generated source code only, ${generated_type} retains original copyright
// SPDX-License-Identifier: GPL-2.0-or-later
// This file was auto-generated by `${generator_command}`
// This file was auto-generated by `${generator_command}` with arguments:
${command_args}
"""


Expand Down

0 comments on commit 88afd53

Please sign in to comment.