qmk_firmware

decorators.py (3315B)

"""Helpful decorators that subcommands can use.
"""
import functools
from time import monotonic
from milc import cli
from qmk.keyboard import find_keyboard_from_dir, keyboard_folder
from qmk.keymap import find_keymap_from_dir
def _get_subcommand_name():
    """Handle missing cli.subcommand_name on older versions of milc
    """
    try:
        return cli.subcommand_name
    except AttributeError:
        return cli._subcommand.__name__
def automagic_keyboard(func):
    """Sets `cli.config.<subcommand>.keyboard` based on environment.
    This will rewrite cli.config.<subcommand>.keyboard if the user did not pass `--keyboard` and the directory they are currently in is a keyboard or keymap directory.
    """
    @functools.wraps(func)
    def wrapper(*args, **kwargs):
        cmd = _get_subcommand_name()
        # TODO: Workaround for if config file contains "old" keyboard name
        #         Potential long-term fix needs to be within global cli or milc
        if cli.config_source[cmd]['keyboard'] == 'config_file':
            cli.config[cmd]['keyboard'] = keyboard_folder(cli.config[cmd]['keyboard'])
        # Ensure that `--keyboard` was not passed and CWD is under `qmk_firmware/keyboards`
        if cli.config_source[cmd]['keyboard'] != 'argument':
            keyboard = find_keyboard_from_dir()
            if keyboard:
                cli.config[cmd]['keyboard'] = keyboard
                cli.config_source[cmd]['keyboard'] = 'keyboard_directory'
        return func(*args, **kwargs)
    return wrapper
def automagic_keymap(func):
    """Sets `cli.config.<subcommand>.keymap` based on environment.
    This will rewrite cli.config.<subcommand>.keymap if the user did not pass `--keymap` and the directory they are currently in is a keymap, layout, or user directory.
    """
    @functools.wraps(func)
    def wrapper(*args, **kwargs):
        cmd = _get_subcommand_name()
        # Ensure that `--keymap` was not passed and that we're under `qmk_firmware`
        if cli.config_source[cmd]['keymap'] != 'argument':
            keymap_name, keymap_type = find_keymap_from_dir()
            if keymap_name:
                cli.config[cmd]['keymap'] = keymap_name
                cli.config_source[cmd]['keymap'] = keymap_type
        return func(*args, **kwargs)
    return wrapper
def lru_cache(timeout=10, maxsize=128, typed=False):
    """Least Recently Used Cache- cache the result of a function.
    Args:
        timeout
            How many seconds to cache results for.
        maxsize
            The maximum size of the cache in bytes
        typed
            When `True` argument types will be taken into consideration, for example `3` and `3.0` will be treated as different keys.
    """
    def wrapper_cache(func):
        func = functools.lru_cache(maxsize=maxsize, typed=typed)(func)
        func.expiration = monotonic() + timeout
        @functools.wraps(func)
        def wrapped_func(*args, **kwargs):
            if monotonic() >= func.expiration:
                func.expiration = monotonic() + timeout
                func.cache_clear()
            return func(*args, **kwargs)
        wrapped_func.cache_info = func.cache_info
        wrapped_func.cache_clear = func.cache_clear
        return wrapped_func
    return wrapper_cache