1227 lines
43 KiB
Python
1227 lines
43 KiB
Python
|
from __future__ import annotations
|
||
|
|
||
|
import functools
|
||
|
import ipaddress
|
||
|
import logging
|
||
|
import os
|
||
|
import string
|
||
|
import sys
|
||
|
import traceback
|
||
|
import types
|
||
|
import warnings
|
||
|
from collections import Counter, UserString
|
||
|
from types import SimpleNamespace
|
||
|
from typing import (
|
||
|
Any,
|
||
|
Callable,
|
||
|
Collection,
|
||
|
Dict,
|
||
|
Generic,
|
||
|
Iterator,
|
||
|
List,
|
||
|
Mapping,
|
||
|
MutableMapping,
|
||
|
NamedTuple,
|
||
|
TypeVar,
|
||
|
Union,
|
||
|
overload,
|
||
|
)
|
||
|
from urllib.parse import quote as urlquote
|
||
|
from urllib.parse import urlsplit, urlunsplit
|
||
|
|
||
|
import markdown
|
||
|
import pathspec
|
||
|
import pathspec.gitignore
|
||
|
|
||
|
from mkdocs import plugins, theme, utils
|
||
|
from mkdocs.config.base import (
|
||
|
BaseConfigOption,
|
||
|
Config,
|
||
|
LegacyConfig,
|
||
|
PlainConfigSchemaItem,
|
||
|
ValidationError,
|
||
|
)
|
||
|
from mkdocs.exceptions import ConfigurationError
|
||
|
|
||
|
T = TypeVar('T')
|
||
|
SomeConfig = TypeVar('SomeConfig', bound=Config)
|
||
|
|
||
|
log = logging.getLogger(__name__)
|
||
|
|
||
|
|
||
|
class SubConfig(Generic[SomeConfig], BaseConfigOption[SomeConfig]):
|
||
|
"""
|
||
|
Subconfig Config Option.
|
||
|
|
||
|
New: If targeting MkDocs 1.4+, please pass a subclass of Config to the
|
||
|
constructor, instead of the old style of a sequence of ConfigOption instances.
|
||
|
Validation is then enabled by default.
|
||
|
|
||
|
A set of `config_options` grouped under a single config option.
|
||
|
By default, validation errors and warnings resulting from validating
|
||
|
`config_options` are ignored (`validate=False`). Users should typically
|
||
|
enable validation with `validate=True`.
|
||
|
"""
|
||
|
|
||
|
_config_file_path: str | None = None
|
||
|
config_class: type[SomeConfig]
|
||
|
|
||
|
@overload
|
||
|
def __init__(
|
||
|
self: SubConfig[SomeConfig], config_class: type[SomeConfig], /, *, validate: bool = True
|
||
|
):
|
||
|
"""Create a sub-config in a type-safe way, using fields defined in a Config subclass."""
|
||
|
|
||
|
@overload
|
||
|
def __init__(
|
||
|
self: SubConfig[LegacyConfig],
|
||
|
*config_options: PlainConfigSchemaItem,
|
||
|
validate: bool = False,
|
||
|
):
|
||
|
"""Create an untyped sub-config, using directly passed fields."""
|
||
|
|
||
|
def __init__(self, *config_options, validate=None):
|
||
|
super().__init__()
|
||
|
self.default = {}
|
||
|
self._do_validation = True if validate is None else validate
|
||
|
if type(self) is SubConfig:
|
||
|
if (
|
||
|
len(config_options) == 1
|
||
|
and isinstance(config_options[0], type)
|
||
|
and issubclass(config_options[0], Config)
|
||
|
):
|
||
|
(self.config_class,) = config_options
|
||
|
else:
|
||
|
self.config_class = functools.partial(LegacyConfig, config_options)
|
||
|
self._do_validation = False if validate is None else validate
|
||
|
|
||
|
def __class_getitem__(cls, config_class: type[Config]):
|
||
|
"""Eliminates the need to write `config_class = FooConfig` when subclassing SubConfig[FooConfig]."""
|
||
|
name = f'{cls.__name__}[{config_class.__name__}]'
|
||
|
return type(name, (cls,), dict(config_class=config_class))
|
||
|
|
||
|
def pre_validation(self, config: Config, key_name: str):
|
||
|
self._config_file_path = config.config_file_path
|
||
|
|
||
|
def run_validation(self, value: object) -> SomeConfig:
|
||
|
config = self.config_class(config_file_path=self._config_file_path)
|
||
|
try:
|
||
|
config.load_dict(value) # type: ignore
|
||
|
failed, warnings = config.validate()
|
||
|
except ConfigurationError as e:
|
||
|
raise ValidationError(str(e))
|
||
|
|
||
|
if self._do_validation:
|
||
|
# Capture errors and warnings
|
||
|
self.warnings.extend(f"Sub-option '{key}': {msg}" for key, msg in warnings)
|
||
|
if failed:
|
||
|
# Get the first failing one
|
||
|
key, err = failed[0]
|
||
|
raise ValidationError(f"Sub-option '{key}': {err}")
|
||
|
|
||
|
return config
|
||
|
|
||
|
|
||
|
class PropagatingSubConfig(SubConfig[SomeConfig], Generic[SomeConfig]):
|
||
|
"""
|
||
|
A SubConfig that must consist of SubConfigs with defined schemas.
|
||
|
|
||
|
Any value set on the top config gets moved to sub-configs with matching keys.
|
||
|
"""
|
||
|
|
||
|
def run_validation(self, value: object):
|
||
|
if isinstance(value, dict):
|
||
|
to_discard = set()
|
||
|
for k1, v1 in self.config_class._schema:
|
||
|
if isinstance(v1, SubConfig):
|
||
|
for k2, _ in v1.config_class._schema:
|
||
|
if k2 in value:
|
||
|
subdict = value.setdefault(k1, {})
|
||
|
if isinstance(subdict, dict):
|
||
|
to_discard.add(k2)
|
||
|
subdict.setdefault(k2, value[k2])
|
||
|
for k in to_discard:
|
||
|
del value[k]
|
||
|
return super().run_validation(value)
|
||
|
|
||
|
|
||
|
class OptionallyRequired(Generic[T], BaseConfigOption[T]):
|
||
|
"""
|
||
|
Soft-deprecated, do not use.
|
||
|
|
||
|
A subclass of BaseConfigOption that adds support for default values and
|
||
|
required values. It is a base class for config options.
|
||
|
"""
|
||
|
|
||
|
@overload
|
||
|
def __init__(self, default=None):
|
||
|
...
|
||
|
|
||
|
@overload
|
||
|
def __init__(self, default=None, *, required: bool):
|
||
|
...
|
||
|
|
||
|
def __init__(self, default=None, required=None):
|
||
|
super().__init__()
|
||
|
self.default = default
|
||
|
self._legacy_required = required
|
||
|
self.required = bool(required)
|
||
|
|
||
|
def validate(self, value):
|
||
|
"""
|
||
|
Perform some initial validation.
|
||
|
|
||
|
If the option is empty (None) and isn't required, leave it as such. If
|
||
|
it is empty but has a default, use that. Finally, call the
|
||
|
run_validation method on the subclass unless.
|
||
|
"""
|
||
|
if value is None:
|
||
|
if self.default is not None:
|
||
|
value = self.default
|
||
|
elif not self.required:
|
||
|
return None
|
||
|
elif self.required:
|
||
|
raise ValidationError("Required configuration not provided.")
|
||
|
|
||
|
return self.run_validation(value)
|
||
|
|
||
|
|
||
|
class ListOfItems(Generic[T], BaseConfigOption[List[T]]):
|
||
|
"""
|
||
|
Validates a homogeneous list of items.
|
||
|
|
||
|
E.g. for `config_options.ListOfItems(config_options.Type(int))` a valid item is `[1, 2, 3]`.
|
||
|
"""
|
||
|
|
||
|
required: bool | None = None # Only for subclasses to set.
|
||
|
|
||
|
def __init__(self, option_type: BaseConfigOption[T], default=None) -> None:
|
||
|
super().__init__()
|
||
|
self.default = default
|
||
|
self.option_type = option_type
|
||
|
self.option_type.warnings = self.warnings
|
||
|
|
||
|
def __repr__(self) -> str:
|
||
|
return f'{type(self).__name__}: {self.option_type}'
|
||
|
|
||
|
def pre_validation(self, config: Config, key_name: str):
|
||
|
self._config = config
|
||
|
self._key_name = key_name
|
||
|
|
||
|
def run_validation(self, value: object) -> list[T]:
|
||
|
if value is None:
|
||
|
if self.required or self.default is None:
|
||
|
raise ValidationError("Required configuration not provided.")
|
||
|
value = self.default
|
||
|
if not isinstance(value, list):
|
||
|
raise ValidationError(f'Expected a list of items, but a {type(value)} was given.')
|
||
|
if not value: # Optimization for empty list
|
||
|
return value
|
||
|
|
||
|
fake_config = LegacyConfig(())
|
||
|
try:
|
||
|
fake_config.config_file_path = self._config.config_file_path
|
||
|
except AttributeError:
|
||
|
pass
|
||
|
|
||
|
# Emulate a config-like environment for pre_validation and post_validation.
|
||
|
parent_key_name = getattr(self, '_key_name', '')
|
||
|
fake_keys = [f'{parent_key_name}[{i}]' for i in range(len(value))]
|
||
|
fake_config.data = dict(zip(fake_keys, value))
|
||
|
|
||
|
self.option_type.warnings = self.warnings
|
||
|
for key_name in fake_config:
|
||
|
self.option_type.pre_validation(fake_config, key_name)
|
||
|
for key_name in fake_config:
|
||
|
# Specifically not running `validate` to avoid the OptionallyRequired effect.
|
||
|
fake_config[key_name] = self.option_type.run_validation(fake_config[key_name])
|
||
|
for key_name in fake_config:
|
||
|
self.option_type.post_validation(fake_config, key_name)
|
||
|
|
||
|
return [fake_config[k] for k in fake_keys]
|
||
|
|
||
|
|
||
|
class DictOfItems(Generic[T], BaseConfigOption[Dict[str, T]]):
|
||
|
"""
|
||
|
Validates a dict of items. Keys are always strings.
|
||
|
|
||
|
E.g. for `config_options.DictOfItems(config_options.Type(int))` a valid item is `{"a": 1, "b": 2}`.
|
||
|
"""
|
||
|
|
||
|
required: bool | None = None # Only for subclasses to set.
|
||
|
|
||
|
def __init__(self, option_type: BaseConfigOption[T], default=None) -> None:
|
||
|
super().__init__()
|
||
|
self.default = default
|
||
|
self.option_type = option_type
|
||
|
self.option_type.warnings = self.warnings
|
||
|
|
||
|
def __repr__(self) -> str:
|
||
|
return f"{type(self).__name__}: {self.option_type}"
|
||
|
|
||
|
def pre_validation(self, config: Config, key_name: str):
|
||
|
self._config = config
|
||
|
self._key_name = key_name
|
||
|
|
||
|
def run_validation(self, value: object) -> dict[str, T]:
|
||
|
if value is None:
|
||
|
if self.required or self.default is None:
|
||
|
raise ValidationError("Required configuration not provided.")
|
||
|
value = self.default
|
||
|
if not isinstance(value, dict):
|
||
|
raise ValidationError(f"Expected a dict of items, but a {type(value)} was given.")
|
||
|
if not value: # Optimization for empty list
|
||
|
return value
|
||
|
|
||
|
fake_config = LegacyConfig(())
|
||
|
try:
|
||
|
fake_config.config_file_path = self._config.config_file_path
|
||
|
except AttributeError:
|
||
|
pass
|
||
|
|
||
|
# Emulate a config-like environment for pre_validation and post_validation.
|
||
|
fake_config.data = value
|
||
|
|
||
|
for key in fake_config:
|
||
|
self.option_type.pre_validation(fake_config, key)
|
||
|
for key in fake_config:
|
||
|
if not isinstance(key, str):
|
||
|
raise ValidationError(
|
||
|
f"Expected type: {str} for keys, but received: {type(key)} (key={key})"
|
||
|
)
|
||
|
for key in fake_config:
|
||
|
# Specifically not running `validate` to avoid the OptionallyRequired effect.
|
||
|
fake_config[key] = self.option_type.run_validation(fake_config[key])
|
||
|
for key in fake_config:
|
||
|
self.option_type.post_validation(fake_config, key)
|
||
|
|
||
|
return value
|
||
|
|
||
|
|
||
|
class ConfigItems(ListOfItems[LegacyConfig]):
|
||
|
"""
|
||
|
Deprecated: Use `ListOfItems(SubConfig(...))` instead of `ConfigItems(...)`.
|
||
|
|
||
|
Validates a list of mappings that all must match the same set of
|
||
|
options.
|
||
|
"""
|
||
|
|
||
|
@overload
|
||
|
def __init__(self, *config_options: PlainConfigSchemaItem):
|
||
|
...
|
||
|
|
||
|
@overload
|
||
|
def __init__(self, *config_options: PlainConfigSchemaItem, required: bool):
|
||
|
...
|
||
|
|
||
|
def __init__(self, *config_options: PlainConfigSchemaItem, required=None) -> None:
|
||
|
super().__init__(SubConfig(*config_options), default=[])
|
||
|
self._legacy_required = required
|
||
|
self.required = bool(required)
|
||
|
|
||
|
|
||
|
class Type(Generic[T], OptionallyRequired[T]):
|
||
|
"""
|
||
|
Type Config Option.
|
||
|
|
||
|
Validate the type of a config option against a given Python type.
|
||
|
"""
|
||
|
|
||
|
@overload
|
||
|
def __init__(self, type_: type[T], /, length: int | None = None, **kwargs):
|
||
|
...
|
||
|
|
||
|
@overload
|
||
|
def __init__(self, type_: tuple[type[T], ...], /, length: int | None = None, **kwargs):
|
||
|
...
|
||
|
|
||
|
def __init__(self, type_, /, length=None, **kwargs) -> None:
|
||
|
super().__init__(**kwargs)
|
||
|
self._type = type_
|
||
|
self.length = length
|
||
|
|
||
|
def run_validation(self, value: object) -> T:
|
||
|
if not isinstance(value, self._type):
|
||
|
msg = f"Expected type: {self._type} but received: {type(value)}"
|
||
|
elif self.length is not None and len(value) != self.length:
|
||
|
msg = (
|
||
|
f"Expected type: {self._type} with length {self.length}"
|
||
|
f" but received: {value!r} with length {len(value)}"
|
||
|
)
|
||
|
else:
|
||
|
return value
|
||
|
|
||
|
raise ValidationError(msg)
|
||
|
|
||
|
|
||
|
class Choice(Generic[T], OptionallyRequired[T]):
|
||
|
"""
|
||
|
Choice Config Option.
|
||
|
|
||
|
Validate the config option against a strict set of values.
|
||
|
"""
|
||
|
|
||
|
def __init__(self, choices: Collection[T], default: T | None = None, **kwargs) -> None:
|
||
|
super().__init__(default=default, **kwargs)
|
||
|
try:
|
||
|
length = len(choices)
|
||
|
except TypeError:
|
||
|
length = 0
|
||
|
|
||
|
if not length or isinstance(choices, str):
|
||
|
raise ValueError(f'Expected iterable of choices, got {choices}')
|
||
|
if default is not None and default not in choices:
|
||
|
raise ValueError(f'{default!r} is not one of {choices!r}')
|
||
|
|
||
|
self.choices = choices
|
||
|
|
||
|
def run_validation(self, value: object) -> T:
|
||
|
if value not in self.choices:
|
||
|
raise ValidationError(f"Expected one of: {self.choices} but received: {value!r}")
|
||
|
return value # type: ignore
|
||
|
|
||
|
|
||
|
class Deprecated(BaseConfigOption):
|
||
|
"""
|
||
|
Deprecated Config Option.
|
||
|
|
||
|
Raises a warning as the option is deprecated. Uses `message` for the
|
||
|
warning. If `move_to` is set to the name of a new config option, the value
|
||
|
is moved to the new option on pre_validation. If `option_type` is set to a
|
||
|
ConfigOption instance, then the value is validated against that type.
|
||
|
"""
|
||
|
|
||
|
def __init__(
|
||
|
self,
|
||
|
moved_to: str | None = None,
|
||
|
message: str | None = None,
|
||
|
removed: bool = False,
|
||
|
option_type: BaseConfigOption | None = None,
|
||
|
) -> None:
|
||
|
super().__init__()
|
||
|
self.default = None
|
||
|
self.moved_to = moved_to
|
||
|
if not message:
|
||
|
if removed:
|
||
|
message = "The configuration option '{}' was removed from MkDocs."
|
||
|
else:
|
||
|
message = (
|
||
|
"The configuration option '{}' has been deprecated and "
|
||
|
"will be removed in a future release."
|
||
|
)
|
||
|
if moved_to:
|
||
|
message += f" Use '{moved_to}' instead."
|
||
|
|
||
|
self.message = message
|
||
|
self.removed = removed
|
||
|
self.option = option_type or BaseConfigOption()
|
||
|
|
||
|
self.warnings = self.option.warnings
|
||
|
|
||
|
def pre_validation(self, config: Config, key_name: str):
|
||
|
self.option.pre_validation(config, key_name)
|
||
|
|
||
|
if config.get(key_name) is not None:
|
||
|
if self.removed:
|
||
|
raise ValidationError(self.message.format(key_name))
|
||
|
self.warnings.append(self.message.format(key_name))
|
||
|
|
||
|
if self.moved_to is not None:
|
||
|
*parent_keys, target_key = self.moved_to.split('.')
|
||
|
target: Any = config
|
||
|
|
||
|
for key in parent_keys:
|
||
|
if target.get(key) is None:
|
||
|
target[key] = {}
|
||
|
target = target[key]
|
||
|
|
||
|
if not isinstance(target, dict):
|
||
|
# We can't move it for the user
|
||
|
return
|
||
|
|
||
|
target[target_key] = config.pop(key_name)
|
||
|
|
||
|
def validate(self, value):
|
||
|
return self.option.validate(value)
|
||
|
|
||
|
def post_validation(self, config: Config, key_name: str):
|
||
|
self.option.post_validation(config, key_name)
|
||
|
|
||
|
def reset_warnings(self):
|
||
|
self.option.reset_warnings()
|
||
|
self.warnings = self.option.warnings
|
||
|
|
||
|
|
||
|
class _IpAddressValue(NamedTuple):
|
||
|
host: str
|
||
|
port: int
|
||
|
|
||
|
def __str__(self) -> str:
|
||
|
return f'{self.host}:{self.port}'
|
||
|
|
||
|
|
||
|
class IpAddress(OptionallyRequired[_IpAddressValue]):
|
||
|
"""
|
||
|
IpAddress Config Option.
|
||
|
|
||
|
Validate that an IP address is in an appropriate format
|
||
|
"""
|
||
|
|
||
|
def run_validation(self, value: object) -> _IpAddressValue:
|
||
|
if not isinstance(value, str) or ':' not in value:
|
||
|
raise ValidationError("Must be a string of format 'IP:PORT'")
|
||
|
host, port_str = value.rsplit(':', 1)
|
||
|
|
||
|
if host != 'localhost':
|
||
|
if host.startswith('[') and host.endswith(']'):
|
||
|
host = host[1:-1]
|
||
|
try:
|
||
|
# Validate and normalize IP Address
|
||
|
host = str(ipaddress.ip_address(host))
|
||
|
except ValueError as e:
|
||
|
raise ValidationError(e)
|
||
|
|
||
|
try:
|
||
|
port = int(port_str)
|
||
|
except Exception:
|
||
|
raise ValidationError(f"'{port_str}' is not a valid port")
|
||
|
|
||
|
return _IpAddressValue(host, port)
|
||
|
|
||
|
|
||
|
class URL(OptionallyRequired[str]):
|
||
|
"""
|
||
|
URL Config Option.
|
||
|
|
||
|
Validate a URL by requiring a scheme is present.
|
||
|
"""
|
||
|
|
||
|
@overload
|
||
|
def __init__(self, default=None, *, is_dir: bool = False):
|
||
|
...
|
||
|
|
||
|
@overload
|
||
|
def __init__(self, default=None, *, required: bool, is_dir: bool = False):
|
||
|
...
|
||
|
|
||
|
def __init__(self, default=None, required=None, is_dir: bool = False) -> None:
|
||
|
self.is_dir = is_dir
|
||
|
super().__init__(default, required=required)
|
||
|
|
||
|
def run_validation(self, value: object) -> str:
|
||
|
if not isinstance(value, str):
|
||
|
raise ValidationError(f"Expected a string, got {type(value)}")
|
||
|
if value == '':
|
||
|
return value
|
||
|
try:
|
||
|
parsed_url = urlsplit(value)
|
||
|
except (AttributeError, TypeError):
|
||
|
raise ValidationError("Unable to parse the URL.")
|
||
|
|
||
|
if parsed_url.scheme and parsed_url.netloc:
|
||
|
if self.is_dir and not parsed_url.path.endswith('/'):
|
||
|
parsed_url = parsed_url._replace(path=f'{parsed_url.path}/')
|
||
|
return urlunsplit(parsed_url)
|
||
|
|
||
|
raise ValidationError("The URL isn't valid, it should include the http:// (scheme)")
|
||
|
|
||
|
|
||
|
class Optional(Generic[T], BaseConfigOption[Union[T, None]]):
|
||
|
"""
|
||
|
Wraps a field and makes a None value possible for it when no value is set.
|
||
|
|
||
|
E.g. `my_field = config_options.Optional(config_options.Type(str))`
|
||
|
"""
|
||
|
|
||
|
def __init__(self, config_option: BaseConfigOption[T]) -> None:
|
||
|
if config_option.default is not None:
|
||
|
raise ValueError(
|
||
|
f"This option already has a default ({config_option.default!r}) "
|
||
|
f"and doesn't need to be wrapped into Optional"
|
||
|
)
|
||
|
super().__init__()
|
||
|
self.option = config_option
|
||
|
self.warnings = config_option.warnings
|
||
|
|
||
|
def __getattr__(self, key):
|
||
|
if key in ('option', 'warnings'):
|
||
|
raise AttributeError
|
||
|
return getattr(self.option, key)
|
||
|
|
||
|
def pre_validation(self, config: Config, key_name: str):
|
||
|
return self.option.pre_validation(config, key_name)
|
||
|
|
||
|
def run_validation(self, value: object) -> T | None:
|
||
|
if value is None:
|
||
|
return None
|
||
|
return self.option.validate(value)
|
||
|
|
||
|
def post_validation(self, config: Config, key_name: str):
|
||
|
result = self.option.post_validation(config, key_name) # type: ignore
|
||
|
self.warnings = self.option.warnings
|
||
|
return result
|
||
|
|
||
|
def reset_warnings(self):
|
||
|
self.option.reset_warnings()
|
||
|
self.warnings = self.option.warnings
|
||
|
|
||
|
|
||
|
class RepoURL(URL):
|
||
|
def __init__(self, *args, **kwargs):
|
||
|
warnings.warn(
|
||
|
"RepoURL is no longer used in MkDocs and will be removed.", DeprecationWarning
|
||
|
)
|
||
|
super().__init__(*args, **kwargs)
|
||
|
|
||
|
def post_validation(self, config: Config, key_name: str):
|
||
|
repo_host = urlsplit(config['repo_url']).netloc.lower()
|
||
|
edit_uri = config.get('edit_uri')
|
||
|
|
||
|
# derive repo_name from repo_url if unset
|
||
|
if config['repo_url'] is not None and config.get('repo_name') is None:
|
||
|
if repo_host == 'github.com':
|
||
|
config['repo_name'] = 'GitHub'
|
||
|
elif repo_host == 'bitbucket.org':
|
||
|
config['repo_name'] = 'Bitbucket'
|
||
|
elif repo_host == 'gitlab.com':
|
||
|
config['repo_name'] = 'GitLab'
|
||
|
else:
|
||
|
config['repo_name'] = repo_host.split('.')[0].title()
|
||
|
|
||
|
# derive edit_uri from repo_name if unset
|
||
|
if config['repo_url'] is not None and edit_uri is None:
|
||
|
if repo_host == 'github.com' or repo_host == 'gitlab.com':
|
||
|
edit_uri = 'edit/master/docs/'
|
||
|
elif repo_host == 'bitbucket.org':
|
||
|
edit_uri = 'src/default/docs/'
|
||
|
else:
|
||
|
edit_uri = ''
|
||
|
|
||
|
# ensure a well-formed edit_uri
|
||
|
if edit_uri and not edit_uri.endswith('/'):
|
||
|
edit_uri += '/'
|
||
|
|
||
|
config['edit_uri'] = edit_uri
|
||
|
|
||
|
|
||
|
class EditURI(Type[str]):
|
||
|
def __init__(self, repo_url_key: str) -> None:
|
||
|
super().__init__(str)
|
||
|
self.repo_url_key = repo_url_key
|
||
|
|
||
|
def post_validation(self, config: Config, key_name: str):
|
||
|
edit_uri = config.get(key_name)
|
||
|
repo_url = config.get(self.repo_url_key)
|
||
|
|
||
|
if edit_uri is None and repo_url is not None:
|
||
|
repo_host = urlsplit(repo_url).netloc.lower()
|
||
|
if repo_host == 'github.com' or repo_host == 'gitlab.com':
|
||
|
edit_uri = 'edit/master/docs/'
|
||
|
elif repo_host == 'bitbucket.org':
|
||
|
edit_uri = 'src/default/docs/'
|
||
|
|
||
|
# ensure a well-formed edit_uri
|
||
|
if edit_uri and not edit_uri.endswith('/'):
|
||
|
edit_uri += '/'
|
||
|
|
||
|
config[key_name] = edit_uri
|
||
|
|
||
|
|
||
|
class EditURITemplate(BaseConfigOption[str]):
|
||
|
class Formatter(string.Formatter):
|
||
|
def convert_field(self, value, conversion):
|
||
|
if conversion == 'q':
|
||
|
return urlquote(value, safe='')
|
||
|
return super().convert_field(value, conversion)
|
||
|
|
||
|
class Template(UserString):
|
||
|
def __init__(self, formatter, data) -> None:
|
||
|
super().__init__(data)
|
||
|
self.formatter = formatter
|
||
|
try:
|
||
|
self.format('', '')
|
||
|
except KeyError as e:
|
||
|
raise ValueError(f"Unknown template substitute: {e}")
|
||
|
|
||
|
def format(self, path, path_noext):
|
||
|
return self.formatter.format(self.data, path=path, path_noext=path_noext)
|
||
|
|
||
|
def __init__(self, edit_uri_key: str | None = None) -> None:
|
||
|
super().__init__()
|
||
|
self.edit_uri_key = edit_uri_key
|
||
|
|
||
|
def run_validation(self, value: object):
|
||
|
try:
|
||
|
return self.Template(self.Formatter(), value)
|
||
|
except Exception as e:
|
||
|
raise ValidationError(e)
|
||
|
|
||
|
def post_validation(self, config: Config, key_name: str):
|
||
|
if self.edit_uri_key and config.get(key_name) and config.get(self.edit_uri_key):
|
||
|
self.warnings.append(
|
||
|
f"The option '{self.edit_uri_key}' has no effect when '{key_name}' is set."
|
||
|
)
|
||
|
|
||
|
|
||
|
class RepoName(Type[str]):
|
||
|
def __init__(self, repo_url_key: str) -> None:
|
||
|
super().__init__(str)
|
||
|
self.repo_url_key = repo_url_key
|
||
|
|
||
|
def post_validation(self, config: Config, key_name: str):
|
||
|
repo_name = config.get(key_name)
|
||
|
repo_url = config.get(self.repo_url_key)
|
||
|
|
||
|
# derive repo_name from repo_url if unset
|
||
|
if repo_url is not None and repo_name is None:
|
||
|
repo_host = urlsplit(config['repo_url']).netloc.lower()
|
||
|
if repo_host == 'github.com':
|
||
|
repo_name = 'GitHub'
|
||
|
elif repo_host == 'bitbucket.org':
|
||
|
repo_name = 'Bitbucket'
|
||
|
elif repo_host == 'gitlab.com':
|
||
|
repo_name = 'GitLab'
|
||
|
else:
|
||
|
repo_name = repo_host.split('.')[0].title()
|
||
|
config[key_name] = repo_name
|
||
|
|
||
|
|
||
|
class FilesystemObject(Type[str]):
|
||
|
"""Base class for options that point to filesystem objects."""
|
||
|
|
||
|
existence_test: Callable[[str], bool] = staticmethod(os.path.exists)
|
||
|
name = 'file or directory'
|
||
|
|
||
|
def __init__(self, exists: bool = False, **kwargs) -> None:
|
||
|
super().__init__(str, **kwargs)
|
||
|
self.exists = exists
|
||
|
self.config_dir: str | None = None
|
||
|
|
||
|
def pre_validation(self, config: Config, key_name: str):
|
||
|
self.config_dir = (
|
||
|
os.path.dirname(config.config_file_path) if config.config_file_path else None
|
||
|
)
|
||
|
|
||
|
def run_validation(self, value: object) -> str:
|
||
|
value = super().run_validation(value)
|
||
|
if self.config_dir and not os.path.isabs(value):
|
||
|
value = os.path.join(self.config_dir, value)
|
||
|
if self.exists and not self.existence_test(value):
|
||
|
raise ValidationError(f"The path '{value}' isn't an existing {self.name}.")
|
||
|
return os.path.abspath(value)
|
||
|
|
||
|
|
||
|
class Dir(FilesystemObject):
|
||
|
"""
|
||
|
Dir Config Option.
|
||
|
|
||
|
Validate a path to a directory, optionally verifying that it exists.
|
||
|
"""
|
||
|
|
||
|
existence_test = staticmethod(os.path.isdir)
|
||
|
name = 'directory'
|
||
|
|
||
|
|
||
|
class DocsDir(Dir):
|
||
|
def post_validation(self, config: Config, key_name: str):
|
||
|
if not config.config_file_path:
|
||
|
return
|
||
|
|
||
|
# Validate that the dir is not the parent dir of the config file.
|
||
|
if os.path.dirname(config.config_file_path) == config[key_name]:
|
||
|
raise ValidationError(
|
||
|
f"The '{key_name}' should not be the parent directory of the"
|
||
|
f" config file. Use a child directory instead so that the"
|
||
|
f" '{key_name}' is a sibling of the config file."
|
||
|
)
|
||
|
|
||
|
|
||
|
class File(FilesystemObject):
|
||
|
"""
|
||
|
File Config Option.
|
||
|
|
||
|
Validate a path to a file, optionally verifying that it exists.
|
||
|
"""
|
||
|
|
||
|
existence_test = staticmethod(os.path.isfile)
|
||
|
name = 'file'
|
||
|
|
||
|
|
||
|
class ListOfPaths(ListOfItems[str]):
|
||
|
"""
|
||
|
List of Paths Config Option.
|
||
|
|
||
|
A list of file system paths. Raises an error if one of the paths does not exist.
|
||
|
|
||
|
For greater flexibility, prefer ListOfItems, e.g. to require files specifically:
|
||
|
|
||
|
config_options.ListOfItems(config_options.File(exists=True))
|
||
|
"""
|
||
|
|
||
|
@overload
|
||
|
def __init__(self, default=[]):
|
||
|
...
|
||
|
|
||
|
@overload
|
||
|
def __init__(self, default=[], *, required: bool):
|
||
|
...
|
||
|
|
||
|
def __init__(self, default=[], required=None) -> None:
|
||
|
super().__init__(FilesystemObject(exists=True), default)
|
||
|
self.required = required
|
||
|
|
||
|
|
||
|
class SiteDir(Dir):
|
||
|
"""
|
||
|
SiteDir Config Option.
|
||
|
|
||
|
Validates the site_dir and docs_dir directories do not contain each other.
|
||
|
"""
|
||
|
|
||
|
def post_validation(self, config: Config, key_name: str):
|
||
|
super().post_validation(config, key_name)
|
||
|
docs_dir = config['docs_dir']
|
||
|
site_dir = config['site_dir']
|
||
|
|
||
|
# Validate that the docs_dir and site_dir don't contain the
|
||
|
# other as this will lead to copying back and forth on each
|
||
|
# and eventually make a deep nested mess.
|
||
|
if (docs_dir + os.sep).startswith(site_dir.rstrip(os.sep) + os.sep):
|
||
|
raise ValidationError(
|
||
|
f"The 'docs_dir' should not be within the 'site_dir' as this "
|
||
|
f"can mean the source files are overwritten by the output or "
|
||
|
f"it will be deleted if --clean is passed to mkdocs build. "
|
||
|
f"(site_dir: '{site_dir}', docs_dir: '{docs_dir}')"
|
||
|
)
|
||
|
elif (site_dir + os.sep).startswith(docs_dir.rstrip(os.sep) + os.sep):
|
||
|
raise ValidationError(
|
||
|
f"The 'site_dir' should not be within the 'docs_dir' as this "
|
||
|
f"leads to the build directory being copied into itself and "
|
||
|
f"duplicate nested files in the 'site_dir'. "
|
||
|
f"(site_dir: '{site_dir}', docs_dir: '{docs_dir}')"
|
||
|
)
|
||
|
|
||
|
|
||
|
class Theme(BaseConfigOption[theme.Theme]):
|
||
|
"""
|
||
|
Theme Config Option.
|
||
|
|
||
|
Validate that the theme exists and build Theme instance.
|
||
|
"""
|
||
|
|
||
|
def __init__(self, default=None) -> None:
|
||
|
super().__init__()
|
||
|
self.default = default
|
||
|
|
||
|
def pre_validation(self, config: Config, key_name: str):
|
||
|
self.config_file_path = config.config_file_path
|
||
|
|
||
|
def run_validation(self, value: object) -> theme.Theme:
|
||
|
if value is None and self.default is not None:
|
||
|
theme_config = {'name': self.default}
|
||
|
elif isinstance(value, str):
|
||
|
theme_config = {'name': value}
|
||
|
elif isinstance(value, dict):
|
||
|
if 'name' not in value:
|
||
|
raise ValidationError("No theme name set.")
|
||
|
theme_config = value
|
||
|
else:
|
||
|
raise ValidationError(
|
||
|
f'Invalid type {type(value)}. Expected a string or key/value pairs.'
|
||
|
)
|
||
|
|
||
|
themes = utils.get_theme_names()
|
||
|
if theme_config['name'] is not None and theme_config['name'] not in themes:
|
||
|
raise ValidationError(
|
||
|
f"Unrecognised theme name: '{theme_config['name']}'. "
|
||
|
f"The available installed themes are: {', '.join(themes)}"
|
||
|
)
|
||
|
|
||
|
if not theme_config['name'] and 'custom_dir' not in theme_config:
|
||
|
raise ValidationError("At least one of 'name' or 'custom_dir' must be defined.")
|
||
|
|
||
|
# Ensure custom_dir is an absolute path
|
||
|
if 'custom_dir' in theme_config and not os.path.isabs(theme_config['custom_dir']):
|
||
|
config_dir = os.path.dirname(self.config_file_path)
|
||
|
theme_config['custom_dir'] = os.path.join(config_dir, theme_config['custom_dir'])
|
||
|
|
||
|
if 'custom_dir' in theme_config and not os.path.isdir(theme_config['custom_dir']):
|
||
|
raise ValidationError(
|
||
|
"The path set in custom_dir ('{path}') does not exist.".format(
|
||
|
path=theme_config['custom_dir']
|
||
|
)
|
||
|
)
|
||
|
|
||
|
if 'locale' in theme_config and not isinstance(theme_config['locale'], str):
|
||
|
raise ValidationError("'locale' must be a string.")
|
||
|
|
||
|
return theme.Theme(**theme_config)
|
||
|
|
||
|
|
||
|
class Nav(OptionallyRequired):
|
||
|
"""
|
||
|
Nav Config Option.
|
||
|
|
||
|
Validate the Nav config.
|
||
|
"""
|
||
|
|
||
|
def run_validation(self, value: object, *, top=True):
|
||
|
if isinstance(value, list):
|
||
|
for subitem in value:
|
||
|
self._validate_nav_item(subitem)
|
||
|
if top and not value:
|
||
|
value = None
|
||
|
elif isinstance(value, dict) and value and not top:
|
||
|
# TODO: this should be an error.
|
||
|
self.warnings.append(f"Expected nav to be a list, got {self._repr_item(value)}")
|
||
|
for subitem in value.values():
|
||
|
self.run_validation(subitem, top=False)
|
||
|
elif isinstance(value, str) and not top:
|
||
|
pass
|
||
|
else:
|
||
|
raise ValidationError(f"Expected nav to be a list, got {self._repr_item(value)}")
|
||
|
return value
|
||
|
|
||
|
def _validate_nav_item(self, value):
|
||
|
if isinstance(value, str):
|
||
|
pass
|
||
|
elif isinstance(value, dict):
|
||
|
if len(value) != 1:
|
||
|
raise ValidationError(
|
||
|
f"Expected nav item to be a dict of size 1, got {self._repr_item(value)}"
|
||
|
)
|
||
|
for subnav in value.values():
|
||
|
self.run_validation(subnav, top=False)
|
||
|
else:
|
||
|
raise ValidationError(
|
||
|
f"Expected nav item to be a string or dict, got {self._repr_item(value)}"
|
||
|
)
|
||
|
|
||
|
@classmethod
|
||
|
def _repr_item(cls, value) -> str:
|
||
|
if isinstance(value, dict) and value:
|
||
|
return f"dict with keys {tuple(value.keys())}"
|
||
|
elif isinstance(value, (str, type(None))):
|
||
|
return repr(value)
|
||
|
else:
|
||
|
return f"a {type(value).__name__}: {value!r}"
|
||
|
|
||
|
|
||
|
class Private(Generic[T], BaseConfigOption[T]):
|
||
|
"""A config option that can only be populated programmatically. Raises an error if set by the user."""
|
||
|
|
||
|
def run_validation(self, value: object) -> None:
|
||
|
if value is not None:
|
||
|
raise ValidationError('For internal use only.')
|
||
|
|
||
|
|
||
|
class ExtraScriptValue(Config):
|
||
|
"""An extra script to be added to the page. The `extra_javascript` config is a list of these."""
|
||
|
|
||
|
path = Type(str)
|
||
|
"""The value of the `src` tag of the script."""
|
||
|
type = Type(str, default='')
|
||
|
"""The value of the `type` tag of the script."""
|
||
|
defer = Type(bool, default=False)
|
||
|
"""Whether to add the `defer` tag to the script."""
|
||
|
async_ = Type(bool, default=False)
|
||
|
"""Whether to add the `async` tag to the script."""
|
||
|
|
||
|
def __init__(self, path: str = '', config_file_path=None):
|
||
|
super().__init__(config_file_path=config_file_path)
|
||
|
self.path = path
|
||
|
|
||
|
def __str__(self):
|
||
|
return self.path
|
||
|
|
||
|
def __fspath__(self):
|
||
|
return self.path
|
||
|
|
||
|
|
||
|
class ExtraScript(BaseConfigOption[Union[ExtraScriptValue, str]]):
|
||
|
def __init__(self):
|
||
|
super().__init__()
|
||
|
self.option_type = SubConfig[ExtraScriptValue]()
|
||
|
|
||
|
def run_validation(self, value: object) -> ExtraScriptValue | str:
|
||
|
self.option_type.warnings = self.warnings
|
||
|
if isinstance(value, str):
|
||
|
if value.endswith('.mjs'):
|
||
|
return self.option_type.run_validation({'path': value, 'type': 'module'})
|
||
|
return value
|
||
|
return self.option_type.run_validation(value)
|
||
|
|
||
|
|
||
|
class MarkdownExtensions(OptionallyRequired[List[str]]):
|
||
|
"""
|
||
|
Markdown Extensions Config Option.
|
||
|
|
||
|
A list or dict of extensions. Each list item may contain either a string or a one item dict.
|
||
|
A string must be a valid Markdown extension name with no config options defined. The key of
|
||
|
a dict item must be a valid Markdown extension name and the value must be a dict of config
|
||
|
options for that extension. Extension configs are set on the private setting passed to
|
||
|
`configkey`. The `builtins` keyword accepts a list of extensions which cannot be overridden by
|
||
|
the user. However, builtins can be duplicated to define config options for them if desired.
|
||
|
"""
|
||
|
|
||
|
def __init__(
|
||
|
self,
|
||
|
builtins: list[str] | None = None,
|
||
|
configkey: str = 'mdx_configs',
|
||
|
default: list[str] = [],
|
||
|
**kwargs,
|
||
|
) -> None:
|
||
|
super().__init__(default=default, **kwargs)
|
||
|
self.builtins = builtins or []
|
||
|
self.configkey = configkey
|
||
|
|
||
|
def validate_ext_cfg(self, ext: object, cfg: object) -> None:
|
||
|
if not isinstance(ext, str):
|
||
|
raise ValidationError(f"'{ext}' is not a valid Markdown Extension name.")
|
||
|
if not cfg:
|
||
|
return
|
||
|
if not isinstance(cfg, dict):
|
||
|
raise ValidationError(f"Invalid config options for Markdown Extension '{ext}'.")
|
||
|
self.configdata[ext] = cfg
|
||
|
|
||
|
def pre_validation(self, config, key_name):
|
||
|
# To appease validation in case it involves the `!relative` tag.
|
||
|
config._current_page = current_page = SimpleNamespace() # type: ignore[attr-defined]
|
||
|
current_page.file = SimpleNamespace()
|
||
|
current_page.file.src_path = ''
|
||
|
|
||
|
def run_validation(self, value: object) -> list[str]:
|
||
|
self.configdata: dict[str, dict] = {}
|
||
|
if not isinstance(value, (list, tuple, dict)):
|
||
|
raise ValidationError('Invalid Markdown Extensions configuration')
|
||
|
extensions = []
|
||
|
if isinstance(value, dict):
|
||
|
for ext, cfg in value.items():
|
||
|
self.validate_ext_cfg(ext, cfg)
|
||
|
extensions.append(ext)
|
||
|
else:
|
||
|
for item in value:
|
||
|
if isinstance(item, dict):
|
||
|
if len(item) > 1:
|
||
|
raise ValidationError('Invalid Markdown Extensions configuration')
|
||
|
ext, cfg = item.popitem()
|
||
|
self.validate_ext_cfg(ext, cfg)
|
||
|
extensions.append(ext)
|
||
|
elif isinstance(item, str):
|
||
|
extensions.append(item)
|
||
|
else:
|
||
|
raise ValidationError('Invalid Markdown Extensions configuration')
|
||
|
|
||
|
extensions = utils.reduce_list(self.builtins + extensions)
|
||
|
|
||
|
# Confirm that Markdown considers extensions to be valid
|
||
|
md = markdown.Markdown()
|
||
|
for ext in extensions:
|
||
|
try:
|
||
|
md.registerExtensions((ext,), self.configdata)
|
||
|
except Exception as e:
|
||
|
stack: list = []
|
||
|
for frame in reversed(traceback.extract_tb(sys.exc_info()[2])):
|
||
|
if not frame.line: # Ignore frames before <frozen importlib._bootstrap>
|
||
|
break
|
||
|
stack.insert(0, frame)
|
||
|
tb = ''.join(traceback.format_list(stack))
|
||
|
|
||
|
raise ValidationError(
|
||
|
f"Failed to load extension '{ext}'.\n{tb}{type(e).__name__}: {e}"
|
||
|
)
|
||
|
|
||
|
return extensions
|
||
|
|
||
|
def post_validation(self, config: Config, key_name: str):
|
||
|
config._current_page = None # type: ignore[attr-defined]
|
||
|
config[self.configkey] = self.configdata
|
||
|
|
||
|
|
||
|
class Plugins(OptionallyRequired[plugins.PluginCollection]):
|
||
|
"""
|
||
|
Plugins config option.
|
||
|
|
||
|
A list or dict of plugins. If a plugin defines config options those are used when
|
||
|
initializing the plugin class.
|
||
|
"""
|
||
|
|
||
|
def __init__(self, theme_key: str | None = None, **kwargs) -> None:
|
||
|
super().__init__(**kwargs)
|
||
|
self.installed_plugins = plugins.get_plugins()
|
||
|
self.theme_key = theme_key
|
||
|
self._config: Config | None = None
|
||
|
self.plugin_cache: dict[str, plugins.BasePlugin] = {}
|
||
|
|
||
|
def pre_validation(self, config, key_name):
|
||
|
self._config = config
|
||
|
|
||
|
def run_validation(self, value: object) -> plugins.PluginCollection:
|
||
|
if not isinstance(value, (list, tuple, dict)):
|
||
|
raise ValidationError('Invalid Plugins configuration. Expected a list or dict.')
|
||
|
self.plugins = plugins.PluginCollection()
|
||
|
self._instance_counter: MutableMapping[str, int] = Counter()
|
||
|
for name, cfg in self._parse_configs(value):
|
||
|
self.load_plugin_with_namespace(name, cfg)
|
||
|
return self.plugins
|
||
|
|
||
|
@classmethod
|
||
|
def _parse_configs(cls, value: list | tuple | dict) -> Iterator[tuple[str, dict]]:
|
||
|
if isinstance(value, dict):
|
||
|
for name, cfg in value.items():
|
||
|
if not isinstance(name, str):
|
||
|
raise ValidationError(f"'{name}' is not a valid plugin name.")
|
||
|
yield name, cfg
|
||
|
else:
|
||
|
for item in value:
|
||
|
if isinstance(item, dict):
|
||
|
if len(item) != 1:
|
||
|
raise ValidationError('Invalid Plugins configuration')
|
||
|
name, cfg = item.popitem()
|
||
|
else:
|
||
|
name = item
|
||
|
cfg = {}
|
||
|
if not isinstance(name, str):
|
||
|
raise ValidationError(f"'{name}' is not a valid plugin name.")
|
||
|
yield name, cfg
|
||
|
|
||
|
def load_plugin_with_namespace(self, name: str, config) -> tuple[str, plugins.BasePlugin]:
|
||
|
if '/' in name: # It's already specified with a namespace.
|
||
|
# Special case: allow to explicitly skip namespaced loading:
|
||
|
if name.startswith('/'):
|
||
|
name = name[1:]
|
||
|
else:
|
||
|
# Attempt to load with prepended namespace for the current theme.
|
||
|
if self.theme_key and self._config:
|
||
|
current_theme = self._config[self.theme_key]
|
||
|
if not isinstance(current_theme, str):
|
||
|
current_theme = current_theme['name']
|
||
|
if current_theme:
|
||
|
expanded_name = f'{current_theme}/{name}'
|
||
|
if expanded_name in self.installed_plugins:
|
||
|
name = expanded_name
|
||
|
return (name, self.load_plugin(name, config))
|
||
|
|
||
|
def load_plugin(self, name: str, config) -> plugins.BasePlugin:
|
||
|
if name not in self.installed_plugins:
|
||
|
raise ValidationError(f'The "{name}" plugin is not installed')
|
||
|
|
||
|
config = config or {} # Users may define a null (None) config
|
||
|
if not isinstance(config, dict):
|
||
|
raise ValidationError(f"Invalid config options for the '{name}' plugin.")
|
||
|
|
||
|
self._instance_counter[name] += 1
|
||
|
inst_number = self._instance_counter[name]
|
||
|
inst_name = name
|
||
|
if inst_number > 1:
|
||
|
inst_name += f' #{inst_number}'
|
||
|
|
||
|
plugin = self.plugin_cache.get(inst_name)
|
||
|
if plugin is None:
|
||
|
plugin_cls = self.installed_plugins[name].load()
|
||
|
|
||
|
if not issubclass(plugin_cls, plugins.BasePlugin):
|
||
|
raise ValidationError(
|
||
|
f'{plugin_cls.__module__}.{plugin_cls.__name__} must be a subclass of'
|
||
|
f' {plugins.BasePlugin.__module__}.{plugins.BasePlugin.__name__}'
|
||
|
)
|
||
|
|
||
|
plugin = plugin_cls()
|
||
|
|
||
|
if hasattr(plugin, 'on_startup') or hasattr(plugin, 'on_shutdown'):
|
||
|
self.plugin_cache[inst_name] = plugin
|
||
|
|
||
|
if inst_number > 1 and not getattr(plugin, 'supports_multiple_instances', False):
|
||
|
self.warnings.append(
|
||
|
f"Plugin '{name}' was specified multiple times - this is likely a mistake, "
|
||
|
"because the plugin doesn't declare `supports_multiple_instances`."
|
||
|
)
|
||
|
|
||
|
# Only if the plugin doesn't have its own "enabled" config, apply a generic one.
|
||
|
if 'enabled' in config and not any(pair[0] == 'enabled' for pair in plugin.config_scheme):
|
||
|
enabled = config.pop('enabled')
|
||
|
if not isinstance(enabled, bool):
|
||
|
raise ValidationError(
|
||
|
f"Plugin '{name}' option 'enabled': Expected boolean but received: {type(enabled)}"
|
||
|
)
|
||
|
if not enabled:
|
||
|
log.debug(f"Plugin '{inst_name}' is disabled in the config, skipping.")
|
||
|
return plugin
|
||
|
|
||
|
errors, warns = plugin.load_config(
|
||
|
config, self._config.config_file_path if self._config else None
|
||
|
)
|
||
|
for warning in warns:
|
||
|
if isinstance(warning, str):
|
||
|
self.warnings.append(f"Plugin '{inst_name}': {warning}")
|
||
|
else:
|
||
|
key, msg = warning
|
||
|
self.warnings.append(f"Plugin '{inst_name}' option '{key}': {msg}")
|
||
|
|
||
|
errors_message = '\n'.join(f"Plugin '{name}' option '{key}': {msg}" for key, msg in errors)
|
||
|
if errors_message:
|
||
|
raise ValidationError(errors_message)
|
||
|
self.plugins[inst_name] = plugin
|
||
|
return plugin
|
||
|
|
||
|
|
||
|
class Hooks(BaseConfigOption[List[types.ModuleType]]):
|
||
|
"""A list of Python scripts to be treated as instances of plugins."""
|
||
|
|
||
|
def __init__(self, plugins_key: str) -> None:
|
||
|
super().__init__()
|
||
|
self.default = []
|
||
|
self.plugins_key = plugins_key
|
||
|
|
||
|
def pre_validation(self, config: Config, key_name: str):
|
||
|
self._base_option = ListOfItems(File(exists=True))
|
||
|
self._base_option.pre_validation(config, key_name)
|
||
|
|
||
|
def run_validation(self, value: object) -> Mapping[str, Any]:
|
||
|
paths = self._base_option.validate(value)
|
||
|
self.warnings.extend(self._base_option.warnings)
|
||
|
assert isinstance(value, list)
|
||
|
|
||
|
hooks = {}
|
||
|
for name, path in zip(value, paths):
|
||
|
hooks[name] = self._load_hook(name, path)
|
||
|
return hooks
|
||
|
|
||
|
@functools.lru_cache(maxsize=None)
|
||
|
def _load_hook(self, name, path):
|
||
|
import importlib.util
|
||
|
|
||
|
spec = importlib.util.spec_from_file_location(name, path)
|
||
|
if spec is None:
|
||
|
raise ValidationError(f"Cannot import path '{path}' as a Python module")
|
||
|
module = importlib.util.module_from_spec(spec)
|
||
|
sys.modules[name] = module
|
||
|
if spec.loader is None:
|
||
|
raise ValidationError(f"Cannot import path '{path}' as a Python module")
|
||
|
|
||
|
old_sys_path = sys.path.copy()
|
||
|
sys.path.insert(0, os.path.dirname(path))
|
||
|
try:
|
||
|
spec.loader.exec_module(module)
|
||
|
finally:
|
||
|
sys.path[:] = old_sys_path
|
||
|
|
||
|
return module
|
||
|
|
||
|
def post_validation(self, config: Config, key_name: str):
|
||
|
plugins = config[self.plugins_key]
|
||
|
for name, hook in config[key_name].items():
|
||
|
plugins[name] = hook
|
||
|
|
||
|
|
||
|
class PathSpec(BaseConfigOption[pathspec.gitignore.GitIgnoreSpec]):
|
||
|
"""A path pattern based on gitignore-like syntax."""
|
||
|
|
||
|
def run_validation(self, value: object) -> pathspec.gitignore.GitIgnoreSpec:
|
||
|
if not isinstance(value, str):
|
||
|
raise ValidationError(f'Expected a multiline string, but a {type(value)} was given.')
|
||
|
try:
|
||
|
return pathspec.gitignore.GitIgnoreSpec.from_lines(lines=value.splitlines())
|
||
|
except ValueError as e:
|
||
|
raise ValidationError(str(e))
|