diff options
Diffstat (limited to 'lib/python/qmk/cli/flash.py')
| -rw-r--r-- | lib/python/qmk/cli/flash.py | 47 | 
1 files changed, 30 insertions, 17 deletions
| diff --git a/lib/python/qmk/cli/flash.py b/lib/python/qmk/cli/flash.py index f669c3cb7e..f8497071ef 100644 --- a/lib/python/qmk/cli/flash.py +++ b/lib/python/qmk/cli/flash.py @@ -6,9 +6,11 @@ A bootloader must be specified.  import subprocess  from argparse import FileType -import qmk.path  from milc import cli -from qmk.commands import compile_configurator_json, create_make_command, find_keyboard_keymap, parse_configurator_json + +import qmk.path +from qmk.decorators import automagic_keyboard, automagic_keymap +from qmk.commands import compile_configurator_json, create_make_command, parse_configurator_json  def print_bootloader_help(): @@ -28,12 +30,15 @@ def print_bootloader_help():      cli.echo('For more info, visit https://docs.qmk.fm/#/flashing') [email protected]('-bl', '--bootloader', default='flash', help='The flash command, corresponding to qmk\'s make options of bootloaders.')  @cli.argument('filename', nargs='?', arg_only=True, type=FileType('r'), help='The configurator export JSON to compile.') [email protected]('-b', '--bootloaders', action='store_true', help='List the available bootloaders.') [email protected]('-bl', '--bootloader', default='flash', help='The flash command, corresponding to qmk\'s make options of bootloaders.')  @cli.argument('-km', '--keymap', help='The keymap to build a firmware for. Use this if you dont have a configurator file. Ignored when a configurator file is supplied.')  @cli.argument('-kb', '--keyboard', help='The keyboard to build a firmware for. Use this if you dont have a configurator file. Ignored when a configurator file is supplied.') [email protected]('-b', '--bootloaders', action='store_true', help='List the available bootloaders.') [email protected]('-n', '--dry-run', arg_only=True, action='store_true', help="Don't actually build, just show the make command to be run.")  @cli.subcommand('QMK Flash.') +@automagic_keyboard +@automagic_keymap  def flash(cli):      """Compile and or flash QMK Firmware or keyboard/layout @@ -42,12 +47,13 @@ def flash(cli):      If no file is supplied, keymap and keyboard are expected. -    If bootloader is omitted, the one according to the rules.mk will be used. - +    If bootloader is omitted the make system will use the configured bootloader for that keyboard.      """ +    command = '' +      if cli.args.bootloaders:          # Provide usage and list bootloaders -        cli.echo('usage: qmk flash [-h] [-b] [-kb KEYBOARD] [-km KEYMAP] [-bl BOOTLOADER] [filename]') +        cli.echo('usage: qmk flash [-h] [-b] [-n] [-kb KEYBOARD] [-km KEYMAP] [-bl BOOTLOADER] [filename]')          print_bootloader_help()          return False @@ -60,16 +66,23 @@ def flash(cli):          cli.log.info('Wrote keymap to {fg_cyan}%s/%s/keymap.c', keymap_path, user_keymap['keymap'])      else: -        # Perform the action the user specified -        user_keyboard, user_keymap = find_keyboard_keymap() -        if user_keyboard and user_keymap: +        if cli.config.flash.keyboard and cli.config.flash.keymap:              # Generate the make command for a specific keyboard/keymap. -            command = create_make_command(user_keyboard, user_keymap, cli.args.bootloader) +            command = create_make_command(cli.config.flash.keyboard, cli.config.flash.keymap, cli.args.bootloader) -        else: -            cli.log.error('You must supply a configurator export or both `--keyboard` and `--keymap`.') -            cli.echo('usage: qmk flash [-h] [-b] [-kb KEYBOARD] [-km KEYMAP] [-bl BOOTLOADER] [filename]') -            return False +        elif not cli.config.flash.keyboard: +            cli.log.error('Could not determine keyboard!') +        elif not cli.config.flash.keymap: +            cli.log.error('Could not determine keymap!') -    cli.log.info('Flashing keymap with {fg_cyan}%s\n\n', ' '.join(command)) -    subprocess.run(command) +    # Compile the firmware, if we're able to +    if command: +        cli.log.info('Compiling keymap with {fg_cyan}%s', ' '.join(command)) +        if not cli.args.dry_run: +            cli.echo('\n') +            subprocess.run(command) + +    else: +        cli.log.error('You must supply a configurator export, both `--keyboard` and `--keymap`, or be in a directory for a keyboard or keymap.') +        cli.echo('usage: qmk flash [-h] [-b] [-n] [-kb KEYBOARD] [-km KEYMAP] [-bl BOOTLOADER] [filename]') +        return False | 
