From 054e071787a2d581cb3f7431e19ccda133b5363d Mon Sep 17 00:00:00 2001 From: zvecr Date: Mon, 5 Apr 2021 21:01:29 +0100 Subject: [PATCH] Add env subcommand --- qmk_cli/subcommands/__init__.py | 1 + qmk_cli/subcommands/env.py | 21 +++++++++++++++++++++ 2 files changed, 22 insertions(+) create mode 100644 qmk_cli/subcommands/env.py diff --git a/qmk_cli/subcommands/__init__.py b/qmk_cli/subcommands/__init__.py index 3b50b61..0eaf4ee 100644 --- a/qmk_cli/subcommands/__init__.py +++ b/qmk_cli/subcommands/__init__.py @@ -3,4 +3,5 @@ We list each subcommand here explicitly because all the reliable ways of searching for modules are slow and delay startup. """ from . import clone # noqa +from . import env # noqa from . import setup # noqa diff --git a/qmk_cli/subcommands/env.py b/qmk_cli/subcommands/env.py new file mode 100644 index 0000000..e1ab72e --- /dev/null +++ b/qmk_cli/subcommands/env.py @@ -0,0 +1,21 @@ +"""Prints environment information. +""" +import os + +from milc import cli + + +@cli.argument('var', default=None, nargs='?', help='Optional variable to query') +@cli.subcommand('Prints environment information.') +def env(cli): + data = { + 'QMK_HOME' : os.environ.get('QMK_HOME', "") + } + + if cli.args.var: + # dump out requested arg + print(data[cli.args.var]) + else: + # dump out everything + for key,val in data.items(): + print(f'{key}="{val}"')