diff options
Diffstat (limited to 'lib/python')
-rw-r--r-- | lib/python/qmk/cli/__init__.py | 1 | ||||
-rwxr-xr-x | lib/python/qmk/cli/cd.py | 46 | ||||
-rw-r--r-- | lib/python/qmk/info.py | 12 |
3 files changed, 55 insertions, 4 deletions
diff --git a/lib/python/qmk/cli/__init__.py b/lib/python/qmk/cli/__init__.py index 094ea80b8..edf351d62 100644 --- a/lib/python/qmk/cli/__init__.py +++ b/lib/python/qmk/cli/__init__.py | |||
@@ -31,6 +31,7 @@ safe_commands = [ | |||
31 | subcommands = [ | 31 | subcommands = [ |
32 | 'qmk.cli.bux', | 32 | 'qmk.cli.bux', |
33 | 'qmk.cli.c2json', | 33 | 'qmk.cli.c2json', |
34 | 'qmk.cli.cd', | ||
34 | 'qmk.cli.cformat', | 35 | 'qmk.cli.cformat', |
35 | 'qmk.cli.chibios.confmigrate', | 36 | 'qmk.cli.chibios.confmigrate', |
36 | 'qmk.cli.clean', | 37 | 'qmk.cli.clean', |
diff --git a/lib/python/qmk/cli/cd.py b/lib/python/qmk/cli/cd.py new file mode 100755 index 000000000..c62c3f56c --- /dev/null +++ b/lib/python/qmk/cli/cd.py | |||
@@ -0,0 +1,46 @@ | |||
1 | """Open a shell in the QMK Home directory | ||
2 | """ | ||
3 | import sys | ||
4 | import os | ||
5 | |||
6 | from milc import cli | ||
7 | |||
8 | from qmk.path import under_qmk_firmware | ||
9 | |||
10 | |||
11 | @cli.subcommand('Go to QMK Home') | ||
12 | def cd(cli): | ||
13 | """Go to QMK Home | ||
14 | """ | ||
15 | if not sys.stdout.isatty(): | ||
16 | cli.log.error("This command is for interactive usage only. For non-interactive usage, 'cd $(qmk env QMK_HOME)' is more robust.") | ||
17 | sys.exit(1) | ||
18 | |||
19 | if not under_qmk_firmware(): | ||
20 | # Only do anything if the user is not under qmk_firmware already | ||
21 | # in order to reduce the possibility of starting multiple shells | ||
22 | cli.log.info("Spawning a subshell in your QMK_HOME directory.") | ||
23 | cli.log.info("Type 'exit' to get back to the parent shell.") | ||
24 | if not cli.platform.lower().startswith('windows'): | ||
25 | # For Linux/Mac/etc | ||
26 | # Check the user's login shell from 'passwd' | ||
27 | # alternatively fall back to $SHELL env var | ||
28 | # and finally to '/bin/bash'. | ||
29 | import getpass | ||
30 | import pwd | ||
31 | shell = pwd.getpwnam(getpass.getuser()).pw_shell | ||
32 | if not shell: | ||
33 | shell = os.environ.get('SHELL', '/bin/bash') | ||
34 | # Start the new subshell | ||
35 | os.execl(shell, shell) | ||
36 | else: | ||
37 | # For Windows | ||
38 | # Check the $SHELL env var | ||
39 | # and fall back to '/usr/bin/bash'. | ||
40 | qmk_env = os.environ.copy() | ||
41 | # Set the prompt for the new shell | ||
42 | qmk_env['MSYS2_PS1'] = qmk_env['PS1'] | ||
43 | # Start the new subshell | ||
44 | cli.run([os.environ.get('SHELL', '/usr/bin/bash')], env=qmk_env) | ||
45 | else: | ||
46 | cli.log.info("Already within qmk_firmware directory.") | ||
diff --git a/lib/python/qmk/info.py b/lib/python/qmk/info.py index d81f30438..dc42fdd4d 100644 --- a/lib/python/qmk/info.py +++ b/lib/python/qmk/info.py | |||
@@ -25,6 +25,13 @@ def _valid_community_layout(layout): | |||
25 | return (Path('layouts/default') / layout).exists() | 25 | return (Path('layouts/default') / layout).exists() |
26 | 26 | ||
27 | 27 | ||
28 | def _remove_newlines_from_labels(layouts): | ||
29 | for layout_name, layout_json in layouts.items(): | ||
30 | for key in layout_json['layout']: | ||
31 | if '\n' in key['label']: | ||
32 | key['label'] = key['label'].split('\n')[0] | ||
33 | |||
34 | |||
28 | def info_json(keyboard): | 35 | def info_json(keyboard): |
29 | """Generate the info.json data for a specific keyboard. | 36 | """Generate the info.json data for a specific keyboard. |
30 | """ | 37 | """ |
@@ -100,10 +107,7 @@ def info_json(keyboard): | |||
100 | _check_matrix(info_data) | 107 | _check_matrix(info_data) |
101 | 108 | ||
102 | # Remove newline characters from layout labels | 109 | # Remove newline characters from layout labels |
103 | for layout_name, layout_json in layouts.items(): | 110 | _remove_newlines_from_labels(layouts) |
104 | for key in layout_json['layout']: | ||
105 | if '\n' in key['label']: | ||
106 | key['label'] = key['label'].split('\n')[0] | ||
107 | 111 | ||
108 | return info_data | 112 | return info_data |
109 | 113 | ||