summaryrefslogtreecommitdiff
path: root/lib/python
diff options
context:
space:
mode:
authorErovia <erovia@users.noreply.github.com>2019-10-08 21:50:21 +0200
committerskullydazed <skullydazed@users.noreply.github.com>2020-02-15 15:19:03 -0800
commit409c610543ffb62677d1162be868a2e44913a943 (patch)
treee23421ddcaa162925be37604672da87fb5d4548c /lib/python
parent348266bd5b6c6c79c292e3df9442a19d99873b5c (diff)
CLI: add support for list_keymaps
List all the available keymaps for a given keyboard
Diffstat (limited to 'lib/python')
-rw-r--r--lib/python/qmk/cli/list/__init__.py1
-rw-r--r--lib/python/qmk/cli/list/keymaps.py84
2 files changed, 85 insertions, 0 deletions
diff --git a/lib/python/qmk/cli/list/__init__.py b/lib/python/qmk/cli/list/__init__.py
index c36ba69548..d83cd20b5b 100644
--- a/lib/python/qmk/cli/list/__init__.py
+++ b/lib/python/qmk/cli/list/__init__.py
@@ -1 +1,2 @@
from . import keyboards
+from . import keymaps
diff --git a/lib/python/qmk/cli/list/keymaps.py b/lib/python/qmk/cli/list/keymaps.py
new file mode 100644
index 0000000000..709a5b9f90
--- /dev/null
+++ b/lib/python/qmk/cli/list/keymaps.py
@@ -0,0 +1,84 @@
+"""List the keymaps for a specific keyboard
+"""
+import os
+import re
+import glob
+from bs4 import UnicodeDammit
+
+from milc import cli
+
+def unicode_text(filename):
+ """Returns the contents of filename as a UTF-8 string. Tries to DTRT when it comes to encoding.
+ """
+ with open(filename, 'rb') as fd:
+ text = UnicodeDammit(fd.read())
+
+ if text.contains_replacement_characters:
+ log_warning('%s: Could not determine file encoding, some characters were replaced.' % (filename,))
+
+ return text.unicode_markup or ''
+
+
+def unicode_lines(filename):
+ """Returns the contents of filename as a UTF-8 string. Tries to DTRT when it comes to encoding.
+ """
+ return unicode_text(filename).split('\n')
+
+def parse_rules_mk(keyboard):
+ base_path = os.path.join(os.getcwd(), "keyboards", keyboard) + os.path.sep
+ rules_mk_path_wildcard = os.path.join(base_path, "**", "rules.mk")
+ paths = [path for path in glob.iglob(rules_mk_path_wildcard, recursive=True) if "keymaps" not in path]
+
+ rules_mk = list()
+ for file in paths:
+ rules_mk_content = unicode_lines(file)
+ parsed_file = dict()
+ for line in rules_mk_content:
+ found = re.search(r'^\s*(\w+)\s*=\s*((?:\s*\w+)+)', line)
+ if found:
+ parsed_file[found.group(1)] = found.group(2)
+ rules_mk.append(parsed_file)
+ return rules_mk
+
+@cli.argument('-kb', '--keyboard', help='Specify keyboard name. Example: 1upkeyboards/1up60hse')
+@cli.subcommand("List the keymaps for a specific keyboard")
+def list_keymaps(cli):
+ """List the keymaps for a specific keyboard
+ """
+ # ask for user input if keyboard was not provided in the command line
+ keyboard = cli.config.list_keymaps.keyboard if cli.config.list_keymaps.keyboard else input("Keyboard Name: ")
+
+ kb_base_path = os.path.join(os.getcwd(), "keyboards", keyboard) + os.path.sep
+ km_path_wildcard = os.path.join(kb_base_path, "**", "keymap.c")
+
+ # find everywhere we have rules.mk where keymaps isn't in the path
+ km_paths = [path for path in glob.iglob(km_path_wildcard, recursive=True) if 'keymaps' in path]
+
+ # replace the keyboard directory's path prefix with the keyboard's name and remove the keymap.c suffix
+ km_find_name = lambda path: path.replace(kb_base_path, keyboard + os.path.sep).replace("/keymaps/", ":").replace(os.path.sep + "keymap.c", "")
+ names = list(map(km_find_name, km_paths))
+
+ # get all the rules.mk files for the keyboard
+ rules_mk = parse_rules_mk(keyboard)
+
+ # find all the supported layouts
+ all_supported_layouts = set()
+ for version in rules_mk:
+ if "LAYOUTS" in version:
+ for layout in version["LAYOUTS"].split():
+ all_supported_layouts.add(layout)
+
+ for layout in all_supported_layouts:
+ cl_base_path = os.path.join(os.getcwd(), "layouts", "community", layout) + os.path.sep
+ cl_path_wildcard = os.path.join(cl_base_path, "**", "keymap.c")
+ # find all the keymap.c files under the community layouts
+ cl_paths = [path for path in glob.iglob(cl_path_wildcard, recursive=True)]
+ # replace the community layouts directory's path with the keyboard's name and remove the keymap.c suffix
+ cl_find_name = lambda path: path.replace(cl_base_path, keyboard + ":").replace(os.path.sep + "keymap.c", "")
+ names = names + list(map(cl_find_name, cl_paths))
+
+ names.sort()
+
+ for name in names:
+ # We echo instead of cli.log.info to allow easier piping of this output
+ cli.echo(name)