Mini Shell
"""A simple configuration system."""
# Copyright (c) IPython Development Team.
# Distributed under the terms of the Modified BSD License.
from __future__ import annotations
import argparse
import copy
import functools
import json
import os
import re
import sys
import typing as t
from logging import Logger
from traitlets.traitlets import Any, Container, Dict, HasTraits, List, TraitType, Undefined
from ..utils import cast_unicode, filefind, warnings
# -----------------------------------------------------------------------------
# Exceptions
# -----------------------------------------------------------------------------
class ConfigError(Exception):
pass
class ConfigLoaderError(ConfigError):
pass
class ConfigFileNotFound(ConfigError):
pass
class ArgumentError(ConfigLoaderError):
pass
# -----------------------------------------------------------------------------
# Argparse fix
# -----------------------------------------------------------------------------
# Unfortunately argparse by default prints help messages to stderr instead of
# stdout. This makes it annoying to capture long help screens at the command
# line, since one must know how to pipe stderr, which many users don't know how
# to do. So we override the print_help method with one that defaults to
# stdout and use our class instead.
class _Sentinel:
def __repr__(self) -> str:
return "<Sentinel deprecated>"
def __str__(self) -> str:
return "<deprecated>"
_deprecated = _Sentinel()
class ArgumentParser(argparse.ArgumentParser):
"""Simple argparse subclass that prints help to stdout by default."""
def print_help(self, file: t.Any = None) -> None:
if file is None:
file = sys.stdout
return super().print_help(file)
print_help.__doc__ = argparse.ArgumentParser.print_help.__doc__
# -----------------------------------------------------------------------------
# Config class for holding config information
# -----------------------------------------------------------------------------
def execfile(fname: str, glob: dict[str, Any]) -> None:
with open(fname, "rb") as f:
exec(compile(f.read(), fname, "exec"), glob, glob) # noqa: S102
class LazyConfigValue(HasTraits):
"""Proxy object for exposing methods on configurable containers
These methods allow appending/extending/updating
to add to non-empty defaults instead of clobbering them.
Exposes:
- append, extend, insert on lists
- update on dicts
- update, add on sets
"""
_value = None
# list methods
_extend: List[t.Any] = List()
_prepend: List[t.Any] = List()
_inserts: List[t.Any] = List()
def append(self, obj: t.Any) -> None:
"""Append an item to a List"""
self._extend.append(obj)
def extend(self, other: t.Any) -> None:
"""Extend a list"""
self._extend.extend(other)
def prepend(self, other: t.Any) -> None:
"""like list.extend, but for the front"""
self._prepend[:0] = other
def merge_into(self, other: t.Any) -> t.Any:
"""
Merge with another earlier LazyConfigValue or an earlier container.
This is useful when having global system-wide configuration files.
Self is expected to have higher precedence.
Parameters
----------
other : LazyConfigValue or container
Returns
-------
LazyConfigValue
if ``other`` is also lazy, a reified container otherwise.
"""
if isinstance(other, LazyConfigValue):
other._extend.extend(self._extend)
self._extend = other._extend
self._prepend.extend(other._prepend)
other._inserts.extend(self._inserts)
self._inserts = other._inserts
if self._update:
other.update(self._update)
self._update = other._update
return self
else:
# other is a container, reify now.
return self.get_value(other)
def insert(self, index: int, other: t.Any) -> None:
if not isinstance(index, int):
raise TypeError("An integer is required")
self._inserts.append((index, other))
# dict methods
# update is used for both dict and set
_update = Any()
def update(self, other: t.Any) -> None:
"""Update either a set or dict"""
if self._update is None:
if isinstance(other, dict):
self._update = {}
else:
self._update = set()
self._update.update(other)
# set methods
def add(self, obj: t.Any) -> None:
"""Add an item to a set"""
self.update({obj})
def get_value(self, initial: t.Any) -> t.Any:
"""construct the value from the initial one
after applying any insert / extend / update changes
"""
if self._value is not None:
return self._value # type:ignore[unreachable]
value = copy.deepcopy(initial)
if isinstance(value, list):
for idx, obj in self._inserts:
value.insert(idx, obj)
value[:0] = self._prepend
value.extend(self._extend)
elif isinstance(value, dict):
if self._update:
value.update(self._update)
elif isinstance(value, set):
if self._update:
value.update(self._update)
self._value = value
return value
def to_dict(self) -> dict[str, t.Any]:
"""return JSONable dict form of my data
Currently update as dict or set, extend, prepend as lists, and inserts as list of tuples.
"""
d = {}
if self._update:
d["update"] = self._update
if self._extend:
d["extend"] = self._extend
if self._prepend:
d["prepend"] = self._prepend
elif self._inserts:
d["inserts"] = self._inserts
return d
def __repr__(self) -> str:
if self._value is not None:
return f"<{self.__class__.__name__} value={self._value!r}>"
else:
return f"<{self.__class__.__name__} {self.to_dict()!r}>"
def _is_section_key(key: str) -> bool:
"""Is a Config key a section name (does it start with a capital)?"""
return bool(key and key[0].upper() == key[0] and not key.startswith("_"))
class Config(dict): # type:ignore[type-arg]
"""An attribute-based dict that can do smart merges.
Accessing a field on a config object for the first time populates the key
with either a nested Config object for keys starting with capitals
or :class:`.LazyConfigValue` for lowercase keys,
allowing quick assignments such as::
c = Config()
c.Class.int_trait = 5
c.Class.list_trait.append("x")
"""
def __init__(self, *args: t.Any, **kwds: t.Any) -> None:
dict.__init__(self, *args, **kwds)
self._ensure_subconfig()
def _ensure_subconfig(self) -> None:
"""ensure that sub-dicts that should be Config objects are
casts dicts that are under section keys to Config objects,
which is necessary for constructing Config objects from dict literals.
"""
for key in self:
obj = self[key]
if _is_section_key(key) and isinstance(obj, dict) and not isinstance(obj, Config):
setattr(self, key, Config(obj))
def _merge(self, other: t.Any) -> None:
"""deprecated alias, use Config.merge()"""
self.merge(other)
def merge(self, other: t.Any) -> None:
"""merge another config object into this one"""
to_update = {}
for k, v in other.items():
if k not in self:
to_update[k] = v
else: # I have this key
if isinstance(v, Config) and isinstance(self[k], Config):
# Recursively merge common sub Configs
self[k].merge(v)
elif isinstance(v, LazyConfigValue):
self[k] = v.merge_into(self[k])
else:
# Plain updates for non-Configs
to_update[k] = v
self.update(to_update)
def collisions(self, other: Config) -> dict[str, t.Any]:
"""Check for collisions between two config objects.
Returns a dict of the form {"Class": {"trait": "collision message"}}`,
indicating which values have been ignored.
An empty dict indicates no collisions.
"""
collisions: dict[str, t.Any] = {}
for section in self:
if section not in other:
continue
mine = self[section]
theirs = other[section]
for key in mine:
if key in theirs and mine[key] != theirs[key]:
collisions.setdefault(section, {})
collisions[section][key] = f"{mine[key]!r} ignored, using {theirs[key]!r}"
return collisions
def __contains__(self, key: t.Any) -> bool:
# allow nested contains of the form `"Section.key" in config`
if "." in key:
first, remainder = key.split(".", 1)
if first not in self:
return False
return remainder in self[first]
return super().__contains__(key)
# .has_key is deprecated for dictionaries.
has_key = __contains__
def _has_section(self, key: str) -> bool:
return _is_section_key(key) and key in self
def copy(self) -> dict[str, t.Any]:
return type(self)(dict.copy(self))
def __copy__(self) -> dict[str, t.Any]:
return self.copy()
def __deepcopy__(self, memo: t.Any) -> Config:
new_config = type(self)()
for key, value in self.items():
if isinstance(value, (Config, LazyConfigValue)):
# deep copy config objects
value = copy.deepcopy(value, memo)
elif type(value) in {dict, list, set, tuple}:
# shallow copy plain container traits
value = copy.copy(value)
new_config[key] = value
return new_config
def __getitem__(self, key: str) -> t.Any:
try:
return dict.__getitem__(self, key)
except KeyError:
if _is_section_key(key):
c = Config()
dict.__setitem__(self, key, c)
return c
elif not key.startswith("_"):
# undefined, create lazy value, used for container methods
v = LazyConfigValue()
dict.__setitem__(self, key, v)
return v
else:
raise
def __setitem__(self, key: str, value: t.Any) -> None:
if _is_section_key(key):
if not isinstance(value, Config):
raise ValueError(
"values whose keys begin with an uppercase "
f"char must be Config instances: {key!r}, {value!r}"
)
dict.__setitem__(self, key, value)
def __getattr__(self, key: str) -> t.Any:
if key.startswith("__"):
return dict.__getattr__(self, key) # type:ignore[attr-defined]
try:
return self.__getitem__(key)
except KeyError as e:
raise AttributeError(e) from e
def __setattr__(self, key: str, value: t.Any) -> None:
if key.startswith("__"):
return dict.__setattr__(self, key, value)
try:
self.__setitem__(key, value)
except KeyError as e:
raise AttributeError(e) from e
def __delattr__(self, key: str) -> None:
if key.startswith("__"):
return dict.__delattr__(self, key)
try:
dict.__delitem__(self, key)
except KeyError as e:
raise AttributeError(e) from e
class DeferredConfig:
"""Class for deferred-evaluation of config from CLI"""
def get_value(self, trait: TraitType[t.Any, t.Any]) -> t.Any:
raise NotImplementedError("Implement in subclasses")
def _super_repr(self) -> str:
# explicitly call super on direct parent
return super(self.__class__, self).__repr__()
class DeferredConfigString(str, DeferredConfig):
"""Config value for loading config from a string
Interpretation is deferred until it is loaded into the trait.
Subclass of str for backward compatibility.
This class is only used for values that are not listed
in the configurable classes.
When config is loaded, `trait.from_string` will be used.
If an error is raised in `.from_string`,
the original string is returned.
.. versionadded:: 5.0
"""
def get_value(self, trait: TraitType[t.Any, t.Any]) -> t.Any:
"""Get the value stored in this string"""
s = str(self)
try:
return trait.from_string(s)
except Exception:
# exception casting from string,
# let the original string lie.
# this will raise a more informative error when config is loaded.
return s
def __repr__(self) -> str:
return f"{self.__class__.__name__}({self._super_repr()})"
class DeferredConfigList(t.List[t.Any], DeferredConfig):
"""Config value for loading config from a list of strings
Interpretation is deferred until it is loaded into the trait.
This class is only used for values that are not listed
in the configurable classes.
When config is loaded, `trait.from_string_list` will be used.
If an error is raised in `.from_string_list`,
the original string list is returned.
.. versionadded:: 5.0
"""
def get_value(self, trait: TraitType[t.Any, t.Any]) -> t.Any:
"""Get the value stored in this string"""
if hasattr(trait, "from_string_list"):
src = list(self)
cast = trait.from_string_list
else:
# only allow one item
if len(self) > 1:
raise ValueError(
f"{trait.name} only accepts one value, got {len(self)}: {list(self)}"
)
src = self[0]
cast = trait.from_string
try:
return cast(src)
except Exception:
# exception casting from string,
# let the original value lie.
# this will raise a more informative error when config is loaded.
return src
def __repr__(self) -> str:
return f"{self.__class__.__name__}({self._super_repr()})"
# -----------------------------------------------------------------------------
# Config loading classes
# -----------------------------------------------------------------------------
class ConfigLoader:
"""A object for loading configurations from just about anywhere.
The resulting configuration is packaged as a :class:`Config`.
Notes
-----
A :class:`ConfigLoader` does one thing: load a config from a source
(file, command line arguments) and returns the data as a :class:`Config` object.
There are lots of things that :class:`ConfigLoader` does not do. It does
not implement complex logic for finding config files. It does not handle
default values or merge multiple configs. These things need to be
handled elsewhere.
"""
def _log_default(self) -> Logger:
from traitlets.log import get_logger
return t.cast(Logger, get_logger())
def __init__(self, log: Logger | None = None) -> None:
"""A base class for config loaders.
log : instance of :class:`logging.Logger` to use.
By default logger of :meth:`traitlets.config.application.Application.instance()`
will be used
Examples
--------
>>> cl = ConfigLoader()
>>> config = cl.load_config()
>>> config
{}
"""
self.clear()
if log is None:
self.log = self._log_default()
self.log.debug("Using default logger")
else:
self.log = log
def clear(self) -> None:
self.config = Config()
def load_config(self) -> Config:
"""Load a config from somewhere, return a :class:`Config` instance.
Usually, this will cause self.config to be set and then returned.
However, in most cases, :meth:`ConfigLoader.clear` should be called
to erase any previous state.
"""
self.clear()
return self.config
class FileConfigLoader(ConfigLoader):
"""A base class for file based configurations.
As we add more file based config loaders, the common logic should go
here.
"""
def __init__(self, filename: str, path: str | None = None, **kw: t.Any) -> None:
"""Build a config loader for a filename and path.
Parameters
----------
filename : str
The file name of the config file.
path : str, list, tuple
The path to search for the config file on, or a sequence of
paths to try in order.
"""
super().__init__(**kw)
self.filename = filename
self.path = path
self.full_filename = ""
def _find_file(self) -> None:
"""Try to find the file by searching the paths."""
self.full_filename = filefind(self.filename, self.path)
class JSONFileConfigLoader(FileConfigLoader):
"""A JSON file loader for config
Can also act as a context manager that rewrite the configuration file to disk on exit.
Example::
with JSONFileConfigLoader('myapp.json','/home/jupyter/configurations/') as c:
c.MyNewConfigurable.new_value = 'Updated'
"""
def load_config(self) -> Config:
"""Load the config from a file and return it as a Config object."""
self.clear()
try:
self._find_file()
except OSError as e:
raise ConfigFileNotFound(str(e)) from e
dct = self._read_file_as_dict()
self.config = self._convert_to_config(dct)
return self.config
def _read_file_as_dict(self) -> dict[str, t.Any]:
with open(self.full_filename) as f:
return t.cast("dict[str, t.Any]", json.load(f))
def _convert_to_config(self, dictionary: dict[str, t.Any]) -> Config:
if "version" in dictionary:
version = dictionary.pop("version")
else:
version = 1
if version == 1:
return Config(dictionary)
else:
raise ValueError(f"Unknown version of JSON config file: {version}")
def __enter__(self) -> Config:
self.load_config()
return self.config
def __exit__(self, exc_type: object, exc_value: object, traceback: object) -> None:
"""
Exit the context manager but do not handle any errors.
In case of any error, we do not want to write the potentially broken
configuration to disk.
"""
self.config.version = 1
json_config = json.dumps(self.config, indent=2)
with open(self.full_filename, "w") as f:
f.write(json_config)
class PyFileConfigLoader(FileConfigLoader):
"""A config loader for pure python files.
This is responsible for locating a Python config file by filename and
path, then executing it to construct a Config object.
"""
def load_config(self) -> Config:
"""Load the config from a file and return it as a Config object."""
self.clear()
try:
self._find_file()
except OSError as e:
raise ConfigFileNotFound(str(e)) from e
self._read_file_as_dict()
return self.config
def load_subconfig(self, fname: str, path: str | None = None) -> None:
"""Injected into config file namespace as load_subconfig"""
if path is None:
path = self.path
loader = self.__class__(fname, path)
try:
sub_config = loader.load_config()
except ConfigFileNotFound:
# Pass silently if the sub config is not there,
# treat it as an empty config file.
pass
else:
self.config.merge(sub_config)
def _read_file_as_dict(self) -> None:
"""Load the config file into self.config, with recursive loading."""
def get_config() -> Config:
"""Unnecessary now, but a deprecation warning is more trouble than it's worth."""
return self.config
namespace = dict( # noqa: C408
c=self.config,
load_subconfig=self.load_subconfig,
get_config=get_config,
__file__=self.full_filename,
)
conf_filename = self.full_filename
with open(conf_filename, "rb") as f:
exec(compile(f.read(), conf_filename, "exec"), namespace, namespace) # noqa: S102
class CommandLineConfigLoader(ConfigLoader):
"""A config loader for command line arguments.
As we add more command line based loaders, the common logic should go
here.
"""
def _exec_config_str(
self, lhs: t.Any, rhs: t.Any, trait: TraitType[t.Any, t.Any] | None = None
) -> None:
"""execute self.config.<lhs> = <rhs>
* expands ~ with expanduser
* interprets value with trait if available
"""
value = rhs
if isinstance(value, DeferredConfig):
if trait:
# trait available, reify config immediately
value = value.get_value(trait)
elif isinstance(rhs, DeferredConfigList) and len(rhs) == 1:
# single item, make it a deferred str
value = DeferredConfigString(os.path.expanduser(rhs[0]))
else:
if trait:
value = trait.from_string(value)
else:
value = DeferredConfigString(value)
*path, key = lhs.split(".")
section = self.config
for part in path:
section = section[part]
section[key] = value
return
def _load_flag(self, cfg: t.Any) -> None:
"""update self.config from a flag, which can be a dict or Config"""
if isinstance(cfg, (dict, Config)):
# don't clobber whole config sections, update
# each section from config:
for sec, c in cfg.items():
self.config[sec].update(c)
else:
raise TypeError("Invalid flag: %r" % cfg)
# match --Class.trait keys for argparse
# matches:
# --Class.trait
# --x
# -x
class_trait_opt_pattern = re.compile(r"^\-?\-[A-Za-z][\w]*(\.[\w]+)*$")
_DOT_REPLACEMENT = "__DOT__"
_DASH_REPLACEMENT = "__DASH__"
class _KVAction(argparse.Action):
"""Custom argparse action for handling --Class.trait=x
Always
"""
def __call__( # type:ignore[override]
self,
parser: argparse.ArgumentParser,
namespace: dict[str, t.Any],
values: t.Sequence[t.Any],
option_string: str | None = None,
) -> None:
if isinstance(values, str):
values = [values]
values = ["-" if v is _DASH_REPLACEMENT else v for v in values]
items = getattr(namespace, self.dest, None)
if items is None:
items = DeferredConfigList()
else:
items = DeferredConfigList(items)
items.extend(values)
setattr(namespace, self.dest, items)
class _DefaultOptionDict(dict): # type:ignore[type-arg]
"""Like the default options dict
but acts as if all --Class.trait options are predefined
"""
def _add_kv_action(self, key: str) -> None:
self[key] = _KVAction(
option_strings=[key],
dest=key.lstrip("-").replace(".", _DOT_REPLACEMENT),
# use metavar for display purposes
metavar=key.lstrip("-"),
)
def __contains__(self, key: t.Any) -> bool:
if "=" in key:
return False
if super().__contains__(key):
return True
if key.startswith("-") and class_trait_opt_pattern.match(key):
self._add_kv_action(key)
return True
return False
def __getitem__(self, key: str) -> t.Any:
if key in self:
return super().__getitem__(key)
else:
raise KeyError(key)
def get(self, key: str, default: t.Any = None) -> t.Any:
try:
return self[key]
except KeyError:
return default
class _KVArgParser(argparse.ArgumentParser):
"""subclass of ArgumentParser where any --Class.trait option is implicitly defined"""
def parse_known_args( # type:ignore[override]
self, args: t.Sequence[str] | None = None, namespace: argparse.Namespace | None = None
) -> tuple[argparse.Namespace | None, list[str]]:
# must be done immediately prior to parsing because if we do it in init,
# registration of explicit actions via parser.add_option will fail during setup
for container in (self, self._optionals):
container._option_string_actions = _DefaultOptionDict(container._option_string_actions)
return super().parse_known_args(args, namespace)
# type aliases
SubcommandsDict = t.Dict[str, t.Any]
class ArgParseConfigLoader(CommandLineConfigLoader):
"""A loader that uses the argparse module to load from the command line."""
parser_class = ArgumentParser
def __init__(
self,
argv: list[str] | None = None,
aliases: dict[str, str] | None = None,
flags: dict[str, str] | None = None,
log: t.Any = None,
classes: list[type[t.Any]] | None = None,
subcommands: SubcommandsDict | None = None,
*parser_args: t.Any,
**parser_kw: t.Any,
) -> None:
"""Create a config loader for use with argparse.
Parameters
----------
classes : optional, list
The classes to scan for *container* config-traits and decide
for their "multiplicity" when adding them as *argparse* arguments.
argv : optional, list
If given, used to read command-line arguments from, otherwise
sys.argv[1:] is used.
*parser_args : tuple
A tuple of positional arguments that will be passed to the
constructor of :class:`argparse.ArgumentParser`.
**parser_kw : dict
A tuple of keyword arguments that will be passed to the
constructor of :class:`argparse.ArgumentParser`.
aliases : dict of str to str
Dict of aliases to full traitlets names for CLI parsing
flags : dict of str to str
Dict of flags to full traitlets names for CLI parsing
log
Passed to `ConfigLoader`
Returns
-------
config : Config
The resulting Config object.
"""
classes = classes or []
super(CommandLineConfigLoader, self).__init__(log=log)
self.clear()
if argv is None:
argv = sys.argv[1:]
self.argv = argv
self.aliases = aliases or {}
self.flags = flags or {}
self.classes = classes
self.subcommands = subcommands # only used for argcomplete currently
self.parser_args = parser_args
self.version = parser_kw.pop("version", None)
kwargs = dict(argument_default=argparse.SUPPRESS) # noqa: C408
kwargs.update(parser_kw)
self.parser_kw = kwargs
def load_config(
self,
argv: list[str] | None = None,
aliases: t.Any = None,
flags: t.Any = _deprecated,
classes: t.Any = None,
) -> Config:
"""Parse command line arguments and return as a Config object.
Parameters
----------
argv : optional, list
If given, a list with the structure of sys.argv[1:] to parse
arguments from. If not given, the instance's self.argv attribute
(given at construction time) is used.
flags
Deprecated in traitlets 5.0, instantiate the config loader with the flags.
"""
if flags is not _deprecated:
warnings.warn(
"The `flag` argument to load_config is deprecated since Traitlets "
f"5.0 and will be ignored, pass flags the `{type(self)}` constructor.",
DeprecationWarning,
stacklevel=2,
)
self.clear()
if argv is None:
argv = self.argv
if aliases is not None:
self.aliases = aliases
if classes is not None:
self.classes = classes
self._create_parser()
self._argcomplete(self.classes, self.subcommands)
self._parse_args(argv)
self._convert_to_config()
return self.config
def get_extra_args(self) -> list[str]:
if hasattr(self, "extra_args"):
return self.extra_args
else:
return []
def _create_parser(self) -> None:
self.parser = self.parser_class(
*self.parser_args,
**self.parser_kw, # type:ignore[arg-type]
)
self._add_arguments(self.aliases, self.flags, self.classes)
def _add_arguments(self, aliases: t.Any, flags: t.Any, classes: t.Any) -> None:
raise NotImplementedError("subclasses must implement _add_arguments")
def _argcomplete(self, classes: list[t.Any], subcommands: SubcommandsDict | None) -> None:
"""If argcomplete is enabled, allow triggering command-line autocompletion"""
def _parse_args(self, args: t.Any) -> t.Any:
"""self.parser->self.parsed_data"""
uargs = [cast_unicode(a) for a in args]
unpacked_aliases: dict[str, str] = {}
if self.aliases:
unpacked_aliases = {}
for alias, alias_target in self.aliases.items():
if alias in self.flags:
continue
if not isinstance(alias, tuple): # type:ignore[unreachable]
alias = (alias,) # type:ignore[assignment]
for al in alias:
if len(al) == 1:
unpacked_aliases["-" + al] = "--" + alias_target
unpacked_aliases["--" + al] = "--" + alias_target
def _replace(arg: str) -> str:
if arg == "-":
return _DASH_REPLACEMENT
for k, v in unpacked_aliases.items():
if arg == k:
return v
if arg.startswith(k + "="):
return v + "=" + arg[len(k) + 1 :]
return arg
if "--" in uargs:
idx = uargs.index("--")
extra_args = uargs[idx + 1 :]
to_parse = uargs[:idx]
else:
extra_args = []
to_parse = uargs
to_parse = [_replace(a) for a in to_parse]
self.parsed_data = self.parser.parse_args(to_parse)
self.extra_args = extra_args
def _convert_to_config(self) -> None:
"""self.parsed_data->self.config"""
for k, v in vars(self.parsed_data).items():
*path, key = k.split(".")
section = self.config
for p in path:
section = section[p]
setattr(section, key, v)
class _FlagAction(argparse.Action):
"""ArgParse action to handle a flag"""
def __init__(self, *args: t.Any, **kwargs: t.Any) -> None:
self.flag = kwargs.pop("flag")
self.alias = kwargs.pop("alias", None)
kwargs["const"] = Undefined
if not self.alias:
kwargs["nargs"] = 0
super().__init__(*args, **kwargs)
def __call__(
self, parser: t.Any, namespace: t.Any, values: t.Any, option_string: str | None = None
) -> None:
if self.nargs == 0 or values is Undefined:
if not hasattr(namespace, "_flags"):
namespace._flags = []
namespace._flags.append(self.flag)
else:
setattr(namespace, self.alias, values)
class KVArgParseConfigLoader(ArgParseConfigLoader):
"""A config loader that loads aliases and flags with argparse,
as well as arbitrary --Class.trait value
"""
parser_class = _KVArgParser # type:ignore[assignment]
def _add_arguments(self, aliases: t.Any, flags: t.Any, classes: t.Any) -> None:
alias_flags: dict[str, t.Any] = {}
argparse_kwds: dict[str, t.Any]
argparse_traits: dict[str, t.Any]
paa = self.parser.add_argument
self.parser.set_defaults(_flags=[])
paa("extra_args", nargs="*")
# An index of all container traits collected::
#
# { <traitname>: (<trait>, <argparse-kwds>) }
#
# Used to add the correct type into the `config` tree.
# Used also for aliases, not to re-collect them.
self.argparse_traits = argparse_traits = {}
for cls in classes:
for traitname, trait in cls.class_traits(config=True).items():
argname = f"{cls.__name__}.{traitname}"
argparse_kwds = {"type": str}
if isinstance(trait, (Container, Dict)):
multiplicity = trait.metadata.get("multiplicity", "append")
if multiplicity == "append":
argparse_kwds["action"] = multiplicity
else:
argparse_kwds["nargs"] = multiplicity
argparse_traits[argname] = (trait, argparse_kwds)
for keys, (value, fhelp) in flags.items():
if not isinstance(keys, tuple):
keys = (keys,)
for key in keys:
if key in aliases:
alias_flags[aliases[key]] = value
continue
keys = ("-" + key, "--" + key) if len(key) == 1 else ("--" + key,)
paa(*keys, action=_FlagAction, flag=value, help=fhelp)
for keys, traitname in aliases.items():
if not isinstance(keys, tuple):
keys = (keys,)
for key in keys:
argparse_kwds = {
"type": str,
"dest": traitname.replace(".", _DOT_REPLACEMENT),
"metavar": traitname,
}
argcompleter = None
if traitname in argparse_traits:
trait, kwds = argparse_traits[traitname]
argparse_kwds.update(kwds)
if "action" in argparse_kwds and traitname in alias_flags:
# flag sets 'action', so can't have flag & alias with custom action
# on the same name
raise ArgumentError(
f"The alias `{key}` for the 'append' sequence "
f"config-trait `{traitname}` cannot be also a flag!'"
)
# For argcomplete, check if any either an argcompleter metadata tag or method
# is available. If so, it should be a callable which takes the command-line key
# string as an argument and other kwargs passed by argcomplete,
# and returns the a list of string completions.
argcompleter = trait.metadata.get("argcompleter") or getattr(
trait, "argcompleter", None
)
if traitname in alias_flags:
# alias and flag.
# when called with 0 args: flag
# when called with >= 1: alias
argparse_kwds.setdefault("nargs", "?")
argparse_kwds["action"] = _FlagAction
argparse_kwds["flag"] = alias_flags[traitname]
argparse_kwds["alias"] = traitname
keys = ("-" + key, "--" + key) if len(key) == 1 else ("--" + key,)
action = paa(*keys, **argparse_kwds)
if argcompleter is not None:
# argcomplete's completers are callables returning list of completion strings
action.completer = functools.partial( # type:ignore[attr-defined]
argcompleter, key=key
)
def _convert_to_config(self) -> None:
"""self.parsed_data->self.config, parse unrecognized extra args via KVLoader."""
extra_args = self.extra_args
for lhs, rhs in vars(self.parsed_data).items():
if lhs == "extra_args":
self.extra_args = ["-" if a == _DASH_REPLACEMENT else a for a in rhs] + extra_args
continue
if lhs == "_flags":
# _flags will be handled later
continue
lhs = lhs.replace(_DOT_REPLACEMENT, ".")
if "." not in lhs:
self._handle_unrecognized_alias(lhs)
trait = None
if isinstance(rhs, list):
rhs = DeferredConfigList(rhs)
elif isinstance(rhs, str):
rhs = DeferredConfigString(rhs)
trait = self.argparse_traits.get(lhs)
if trait:
trait = trait[0]
# eval the KV assignment
try:
self._exec_config_str(lhs, rhs, trait)
except Exception as e:
# cast deferred to nicer repr for the error
# DeferredList->list, etc
if isinstance(rhs, DeferredConfig):
rhs = rhs._super_repr()
raise ArgumentError(f"Error loading argument {lhs}={rhs}, {e}") from e
for subc in self.parsed_data._flags:
self._load_flag(subc)
def _handle_unrecognized_alias(self, arg: str) -> None:
"""Handling for unrecognized alias arguments
Probably a mistyped alias. By default just log a warning,
but users can override this to raise an error instead, e.g.
self.parser.error("Unrecognized alias: '%s'" % arg)
"""
self.log.warning("Unrecognized alias: '%s', it will have no effect.", arg)
def _argcomplete(self, classes: list[t.Any], subcommands: SubcommandsDict | None) -> None:
"""If argcomplete is enabled, allow triggering command-line autocompletion"""
try:
import argcomplete # noqa: F401
except ImportError:
return
from . import argcomplete_config
finder = argcomplete_config.ExtendedCompletionFinder() # type:ignore[no-untyped-call]
finder.config_classes = classes
finder.subcommands = list(subcommands or [])
# for ease of testing, pass through self._argcomplete_kwargs if set
finder(self.parser, **getattr(self, "_argcomplete_kwargs", {}))
class KeyValueConfigLoader(KVArgParseConfigLoader):
"""Deprecated in traitlets 5.0
Use KVArgParseConfigLoader
"""
def __init__(self, *args: t.Any, **kwargs: t.Any) -> None:
warnings.warn(
"KeyValueConfigLoader is deprecated since Traitlets 5.0."
" Use KVArgParseConfigLoader instead.",
DeprecationWarning,
stacklevel=2,
)
super().__init__(*args, **kwargs)
def load_pyconfig_files(config_files: list[str], path: str) -> Config:
"""Load multiple Python config files, merging each of them in turn.
Parameters
----------
config_files : list of str
List of config files names to load and merge into the config.
path : unicode
The full path to the location of the config files.
"""
config = Config()
for cf in config_files:
loader = PyFileConfigLoader(cf, path=path)
try:
next_config = loader.load_config()
except ConfigFileNotFound:
pass
except Exception:
raise
else:
config.merge(next_config)
return config
Zerion Mini Shell 1.0