2019-07-15 19:14:27 +00:00
|
|
|
"""Functions that help us work with files and folders.
|
|
|
|
"""
|
2019-09-10 12:14:25 +00:00
|
|
|
import logging
|
2019-07-15 19:14:27 +00:00
|
|
|
import os
|
2021-02-07 21:02:51 +00:00
|
|
|
import argparse
|
2020-02-17 19:42:11 +00:00
|
|
|
from pathlib import Path
|
2019-07-15 19:14:27 +00:00
|
|
|
|
2020-05-26 20:05:41 +00:00
|
|
|
from qmk.constants import MAX_KEYBOARD_SUBFOLDERS, QMK_FIRMWARE
|
2019-09-10 12:14:25 +00:00
|
|
|
from qmk.errors import NoSuchKeyboardError
|
|
|
|
|
2020-01-07 20:54:21 +00:00
|
|
|
|
2020-02-17 19:42:11 +00:00
|
|
|
def is_keyboard(keyboard_name):
|
|
|
|
"""Returns True if `keyboard_name` is a keyboard we can compile.
|
|
|
|
"""
|
2020-05-27 07:43:22 +00:00
|
|
|
if keyboard_name:
|
|
|
|
keyboard_path = QMK_FIRMWARE / 'keyboards' / keyboard_name
|
|
|
|
rules_mk = keyboard_path / 'rules.mk'
|
2021-03-24 16:26:38 +00:00
|
|
|
|
2020-05-27 07:43:22 +00:00
|
|
|
return rules_mk.exists()
|
2020-02-17 19:42:11 +00:00
|
|
|
|
|
|
|
|
2023-11-01 22:37:05 +00:00
|
|
|
def under_qmk_firmware(path=Path(os.environ['ORIG_CWD'])):
|
2020-02-17 19:42:11 +00:00
|
|
|
"""Returns a Path object representing the relative path under qmk_firmware, or None.
|
|
|
|
"""
|
|
|
|
try:
|
2023-11-01 22:37:05 +00:00
|
|
|
return path.relative_to(QMK_FIRMWARE)
|
2020-02-17 19:42:11 +00:00
|
|
|
except ValueError:
|
|
|
|
return None
|
|
|
|
|
|
|
|
|
2020-11-07 17:56:08 +00:00
|
|
|
def keyboard(keyboard_name):
|
|
|
|
"""Returns the path to a keyboard's directory relative to the qmk root.
|
|
|
|
"""
|
|
|
|
return Path('keyboards') / keyboard_name
|
|
|
|
|
|
|
|
|
2023-05-19 06:05:43 +00:00
|
|
|
def keymaps(keyboard_name):
|
|
|
|
"""Returns all of the `keymaps/` directories for a given keyboard.
|
2019-07-15 19:14:27 +00:00
|
|
|
|
|
|
|
Args:
|
2020-02-17 19:42:11 +00:00
|
|
|
|
2020-11-07 17:56:08 +00:00
|
|
|
keyboard_name
|
2019-07-15 19:14:27 +00:00
|
|
|
The name of the keyboard. Example: clueboard/66/rev3
|
|
|
|
"""
|
2020-11-07 17:56:08 +00:00
|
|
|
keyboard_folder = keyboard(keyboard_name)
|
2023-05-19 06:05:43 +00:00
|
|
|
found_dirs = []
|
2020-02-17 19:42:11 +00:00
|
|
|
|
2022-02-08 19:03:30 +00:00
|
|
|
for _ in range(MAX_KEYBOARD_SUBFOLDERS):
|
2020-02-17 19:42:11 +00:00
|
|
|
if (keyboard_folder / 'keymaps').exists():
|
2023-05-19 06:05:43 +00:00
|
|
|
found_dirs.append((keyboard_folder / 'keymaps').resolve())
|
2019-07-15 19:14:27 +00:00
|
|
|
|
2020-02-17 19:42:11 +00:00
|
|
|
keyboard_folder = keyboard_folder.parent
|
2019-07-15 19:14:27 +00:00
|
|
|
|
2023-05-19 06:05:43 +00:00
|
|
|
if len(found_dirs) > 0:
|
|
|
|
return found_dirs
|
|
|
|
|
2020-02-17 19:42:11 +00:00
|
|
|
logging.error('Could not find the keymaps directory!')
|
2020-11-07 17:56:08 +00:00
|
|
|
raise NoSuchKeyboardError('Could not find keymaps directory for: %s' % keyboard_name)
|
2019-07-15 19:14:27 +00:00
|
|
|
|
|
|
|
|
2023-05-19 06:05:43 +00:00
|
|
|
def keymap(keyboard_name, keymap_name):
|
|
|
|
"""Locate the directory of a given keymap.
|
|
|
|
|
|
|
|
Args:
|
|
|
|
|
|
|
|
keyboard_name
|
|
|
|
The name of the keyboard. Example: clueboard/66/rev3
|
|
|
|
keymap_name
|
|
|
|
The name of the keymap. Example: default
|
|
|
|
"""
|
|
|
|
for keymap_dir in keymaps(keyboard_name):
|
|
|
|
if (keymap_dir / keymap_name).exists():
|
|
|
|
return (keymap_dir / keymap_name).resolve()
|
|
|
|
|
|
|
|
|
2019-07-15 19:14:27 +00:00
|
|
|
def normpath(path):
|
2020-02-17 19:42:11 +00:00
|
|
|
"""Returns a `pathlib.Path()` object for a given path.
|
2019-07-15 19:14:27 +00:00
|
|
|
|
2020-02-17 19:42:11 +00:00
|
|
|
This will use the path to a file as seen from the directory the script was called from. You should use this to normalize filenames supplied from the command line.
|
2019-07-15 19:14:27 +00:00
|
|
|
"""
|
2020-02-17 19:42:11 +00:00
|
|
|
path = Path(path)
|
|
|
|
|
|
|
|
if path.is_absolute():
|
2020-03-13 22:47:04 +00:00
|
|
|
return path
|
2019-07-15 19:14:27 +00:00
|
|
|
|
2020-02-17 19:42:11 +00:00
|
|
|
return Path(os.environ['ORIG_CWD']) / path
|
2021-02-07 21:02:51 +00:00
|
|
|
|
|
|
|
|
|
|
|
class FileType(argparse.FileType):
|
2022-03-19 20:58:30 +00:00
|
|
|
def __init__(self, *args, **kwargs):
|
2022-02-28 20:02:39 +00:00
|
|
|
# Use UTF8 by default for stdin
|
2022-03-19 20:58:30 +00:00
|
|
|
if 'encoding' not in kwargs:
|
|
|
|
kwargs['encoding'] = 'UTF-8'
|
|
|
|
return super().__init__(*args, **kwargs)
|
2022-02-28 20:02:39 +00:00
|
|
|
|
2021-02-07 21:02:51 +00:00
|
|
|
def __call__(self, string):
|
|
|
|
"""normalize and check exists
|
|
|
|
otherwise magic strings like '-' for stdin resolve to bad paths
|
|
|
|
"""
|
|
|
|
norm = normpath(string)
|
2022-02-28 20:02:39 +00:00
|
|
|
return norm if norm.exists() else super().__call__(string)
|