1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874 875 876 877 878 879 880 881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948 949 950 951 952 953 954 955 956 957 958 959 960 961 962 963 964 965 966 967 968 969 970 971 972 973 974 975 976 977 978 979 980 981 982 983 984 985 986 987 988 989 990 991 992 993 994 995 996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065 1066 1067
|
from __future__ import annotations
import functools
import ipaddress
import os
import string
import sys
import traceback
import types
import typing as t
import warnings
from collections import Counter, UserString
from typing import (
Any,
Collection,
Dict,
Generic,
Iterator,
List,
Mapping,
MutableMapping,
NamedTuple,
Tuple,
TypeVar,
Union,
overload,
)
from urllib.parse import quote as urlquote
from urllib.parse import urlsplit, urlunsplit
import markdown
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)
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`.
"""
@overload
def __init__(
self: SubConfig[SomeConfig], config_class: t.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 = {}
if (
len(config_options) == 1
and isinstance(config_options[0], type)
and issubclass(config_options[0], Config)
):
if validate is None:
validate = True
(self._make_config,) = config_options
else:
self._make_config = functools.partial(LegacyConfig, config_options)
self._do_validation = bool(validate)
def run_validation(self, value: object) -> SomeConfig:
config = self._make_config()
try:
config.load_dict(value)
failed, warnings = config.validate()
except ConfigurationError as e:
raise ValidationError(str(e))
if self._do_validation:
# Capture errors and warnings
self.warnings = [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 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: Union[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))
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 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_: t.Type[T], length: t.Optional[int] = None, **kwargs):
...
@overload
def __init__(self, type_: Tuple[t.Type[T], ...], length: t.Optional[int] = 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.Optional[T] = 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: t.Optional[str] = None,
message: t.Optional[str] = None,
removed: bool = False,
option_type: t.Optional[BaseConfigOption] = 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)
def post_validation(self, config: Config, key_name: str):
host = config[key_name].host
if key_name == 'dev_addr' and host in ['0.0.0.0', '::']:
self.warnings.append(
f"The use of the IP address '{host}' suggests a production environment "
"or the use of a proxy to connect to the MkDocs server. However, "
"the MkDocs' server is intended for local development purposes only. "
"Please use a third party production-ready server instead."
)
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) -> Union[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: t.Optional[str] = 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 = staticmethod(os.path.exists)
name = 'file or directory'
def __init__(self, exists: bool = False, **kwargs) -> None:
super().__init__(type_=str, **kwargs)
self.exists = exists
self.config_dir: t.Optional[str] = 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 config.config_file_path is None:
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']):
assert self.config_file_path is not None
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(BaseConfigOption):
"""
Private Config Option
A config option only for internal use. Raises an error if set by the user.
"""
def run_validation(self, value: object):
if value is not None:
raise ValidationError('For internal use only.')
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: t.Optional[List[str]] = 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, cfg):
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 run_validation(self, value: object):
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[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: t.Optional[str] = None, **kwargs) -> None:
super().__init__(**kwargs)
self.installed_plugins = plugins.get_plugins()
self.theme_key = theme_key
self._config: t.Optional[Config] = 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: Union[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]['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`."
)
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)
value = t.cast(List[str], value)
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)
if spec.loader is None:
raise ValidationError(f"Cannot import path '{path}' as a Python module")
spec.loader.exec_module(module)
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
|