qmk_firmware/lib/python/qmk/cli/find.py

"""Command to search through all keyboards and keymaps for a given search criteria."""

import os
from milc import cli
from qmk.search import filter_help, search_keymap_targets
from qmk.util import maybe_exit_config


@cli.argument('-f', '--filter', arg_only=True, action='append', default=[], help=f"Filter the list of keyboards based on their info.json data. Accepts the formats key==value, function(key), or function(key,value), eg. 'features.rgblight==true'. Valid functions are {filter_help()}. May be passed multiple times; all filters need to match. Value may include wildcards such as '*' and '?'.")  # fmt: off
@cli.argument('-p', '--print', arg_only=True, action='append', default=[], help="For each matched target, print the value of the supplied info.json key. May be passed multiple times.")
@cli.argument('-km', '--keymap', type=str, default='default', help="The keymap name to build. Default is 'default'.")
@cli.subcommand('Find builds which match supplied search criteria.')
def find(cli):
    """Search through all keyboards and keymaps for a given search criteria."""
    os.environ.setdefault('SKIP_SCHEMA_VALIDATION', '1')
    maybe_exit_config(should_exit=False, should_reraise=True)

    targets = search_keymap_targets([('all', cli.config.find.keymap)], cli.args.filter)
    for target in sorted(targets, key=lambda t: (t.keyboard, t.keymap)):
        print(f'{target}')

        for key in cli.args.print:
            print(f'    {key}={target.dotty.get(key, None)}')