Added info plugin, a helper for creating bug reports

This commit is contained in:
squidfunk 2022-12-18 15:41:18 +01:00
parent 76e155cb9c
commit 5ec4a45110
11 changed files with 394 additions and 3 deletions

File diff suppressed because one or more lines are too long

File diff suppressed because one or more lines are too long

File diff suppressed because one or more lines are too long

File diff suppressed because one or more lines are too long

View File

@ -246,7 +246,7 @@
{% endfor %}
{% endblock %}
{% if page.meta and page.meta.ᴴₒᴴₒᴴₒ %}
<link rel="stylesheet" href="{{ 'assets/stylesheets/extra.c7039805.min.css' | url }}">
<link rel="stylesheet" href="{{ 'assets/stylesheets/extra.195e5743.min.css' | url }}">
<script src="{{ 'assets/javascripts/extra/bundle.cfb3feee.min.js' | url }}" defer></script>
{% endif %}
</body>

View File

View File

@ -0,0 +1,190 @@
# Copyright (c) 2016-2022 Martin Donath <martin.donath@squidfunk.com>
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to
# deal in the Software without restriction, including without limitation the
# rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
# sell copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
# IN THE SOFTWARE.
import logging
import os
import platform
import sys
import warnings
# Silence deprecation warnings coming from pip
warnings.filterwarnings("ignore", category = DeprecationWarning)
from colorama import Fore, Style
from io import BytesIO
from mkdocs import utils
from mkdocs.commands.build import DuplicateFilter
from mkdocs.config import config_options as opt
from mkdocs.config.base import Config
from mkdocs.plugins import BasePlugin, event_priority
from mkdocs.structure.files import get_files
from zipfile import ZipFile, ZIP_DEFLATED
try:
from pipdeptree import (
PackageDAG,
get_installed_distributions,
render_json_tree
)
dependencies = True
except ImportError:
dependencies = False
# -----------------------------------------------------------------------------
# Class
# -----------------------------------------------------------------------------
# Info plugin configuration scheme
class InfoPluginConfig(Config):
enabled = opt.Type(bool, default = True),
# Options for archive
archive = opt.Type(bool, default = True),
archive_name = opt.Type(str, default = "example"),
# -----------------------------------------------------------------------------
# Info plugin
class InfoPlugin(BasePlugin[InfoPluginConfig]):
# Initialize plugin (run earliest)
@event_priority(100)
def on_config(self, config):
if not self.config.enabled:
return
# Check if required dependencies are installed
if not dependencies:
log.error("Required dependencies of \"info\" plugin not found.")
print(Fore.RED)
print(" pip install \"mkdocs-material[info]\"")
print(Style.RESET_ALL)
sys.exit(1)
# Print info that we're creating a bug report
log.info("Started creating archive for bug report")
# Check that there are no overrides in place - we need to use a little
# hack to detect whether the custom_dir setting was used without parsing
# mkdocs.yml again - we check at which position the directory provided
# by the theme resides, and if it's not the first one, abort.
base = utils.get_theme_dir(config.theme.name)
if config.theme.dirs.index(base):
log.error("Please remove 'custom_dir' setting.")
self._help_and_exit()
# Check that there are no hooks in place
if config.hooks:
log.error("Please remove 'hooks' setting.")
self._help_and_exit()
# Create in-memory archive
archive = BytesIO()
archive_name = self.config.archive_name
# Create self-contained example from project
files = []
with ZipFile(archive, "a", ZIP_DEFLATED, False) as f:
for path in ["mkdocs.yml", "requirements.txt"]:
if os.path.isfile(path):
f.write(path, os.path.join(archive_name, path))
# Append all files visible to MkDocs
for file in get_files(config):
path = os.path.relpath(file.abs_src_path, os.path.curdir)
f.write(path, os.path.join(archive_name, path))
# Add dependency tree as returned by pipdeptree
f.writestr(
os.path.join(archive_name, "dependencies.json"),
render_json_tree(PackageDAG.from_pkgs(
get_installed_distributions(local_only = True)
), 2)
)
# Retrieve list of processed files
for a in f.filelist:
files.append("".join([
Fore.LIGHTBLACK_EX, a.filename, " ",
Fore.GREEN, _size(a.compress_size)
]))
# Finally, write archive to disk
buffer = archive.getbuffer()
with open(f"{archive_name}.zip", "wb") as f:
f.write(archive.getvalue())
# Print summary and archive name
log.info("Archive successfully created:")
print(Style.NORMAL)
print("".join([
" ", f.name, " ",
Fore.GREEN, _size(buffer.nbytes)
]))
print(Style.RESET_ALL)
# Print sorted list of archived files
files.sort()
for file in files:
print(f" {file}")
# Aaaaaand done.
print(Style.RESET_ALL)
sys.exit(1)
# -------------------------------------------------------------------------
# Print explanation and exit
def _help_and_exit(self):
print(Fore.RED)
print(" When reporting issues, you must remove all customizations")
print(" and check if the problem persists. If not, the problem is")
print(" caused by your overrides. Please understand that we can't")
print(" provide support for customizations. Please remove:")
print(Style.NORMAL)
print(" - theme.custom_dir")
print(" - hooks")
print(Fore.YELLOW)
print(" Additionally, please remove all third-party JavaScript or")
print(" CSS not explicitly mentioned in our documentation:")
print(Style.NORMAL)
print(" - extra_css")
print(" - extra_javascript")
print(Style.RESET_ALL)
sys.exit(1)
# -----------------------------------------------------------------------------
# Helper functions
# -----------------------------------------------------------------------------
# Print human-readable size
def _size(value):
for unit in ["B", "kB", "MB", "GB", "TB", "PB", "EB", "ZB"]:
if abs(value) < 1000.0:
return f"{value:3.1f} {unit}"
value /= 1000.0
# -----------------------------------------------------------------------------
# Data
# -----------------------------------------------------------------------------
# Set up logging
log = logging.getLogger("mkdocs")
log.addFilter(DuplicateFilter())

View File

@ -27,5 +27,6 @@ pygments>=2.12
pymdown-extensions>=9.6
# Requirements for plugins
colorama>=0.4.6
regex>=2022.4.24
requests>=2.26

View File

@ -42,7 +42,11 @@ setup(
version = package["version"],
url = package["homepage"],
project_urls = {
"Changelog": "https://squidfunk.github.io/mkdocs-material/changelog/",
"Documentation": "https://squidfunk.github.io/mkdocs-material/",
"Source": "https://github.com/squidfunk/mkdocs-material",
"Issues": "https://github.com/squidfunk/mkdocs-material/issues",
"Funding": "https://github.com/sponsors/squidfunk"
},
license = package["license"],
description = package["description"],
@ -64,12 +68,18 @@ setup(
packages = find_packages(exclude = ["src", "src.*"]),
include_package_data = True,
install_requires = install_requires,
extras_require = {
"info": [
"pipdeptree>=2.3.3"
]
},
python_requires='>=3.7',
entry_points = {
"mkdocs.themes": [
"material = material",
],
"mkdocs.plugins": [
"material/info = material.plugins.info.plugin:InfoPlugin",
"material/offline = material.plugins.offline.plugin:OfflinePlugin",
"material/search = material.plugins.search.plugin:SearchPlugin",
"material/social = material.plugins.social.plugin:SocialPlugin",

View File

190
src/plugins/info/plugin.py Normal file
View File

@ -0,0 +1,190 @@
# Copyright (c) 2016-2022 Martin Donath <martin.donath@squidfunk.com>
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to
# deal in the Software without restriction, including without limitation the
# rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
# sell copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
# IN THE SOFTWARE.
import logging
import os
import platform
import sys
import warnings
# Silence deprecation warnings coming from pip
warnings.filterwarnings("ignore", category = DeprecationWarning)
from colorama import Fore, Style
from io import BytesIO
from mkdocs import utils
from mkdocs.commands.build import DuplicateFilter
from mkdocs.config import config_options as opt
from mkdocs.config.base import Config
from mkdocs.plugins import BasePlugin, event_priority
from mkdocs.structure.files import get_files
from zipfile import ZipFile, ZIP_DEFLATED
try:
from pipdeptree import (
PackageDAG,
get_installed_distributions,
render_json_tree
)
dependencies = True
except ImportError:
dependencies = False
# -----------------------------------------------------------------------------
# Class
# -----------------------------------------------------------------------------
# Info plugin configuration scheme
class InfoPluginConfig(Config):
enabled = opt.Type(bool, default = True),
# Options for archive
archive = opt.Type(bool, default = True),
archive_name = opt.Type(str, default = "example"),
# -----------------------------------------------------------------------------
# Info plugin
class InfoPlugin(BasePlugin[InfoPluginConfig]):
# Initialize plugin (run earliest)
@event_priority(100)
def on_config(self, config):
if not self.config.enabled:
return
# Check if required dependencies are installed
if not dependencies:
log.error("Required dependencies of \"info\" plugin not found.")
print(Fore.RED)
print(" pip install \"mkdocs-material[info]\"")
print(Style.RESET_ALL)
sys.exit(1)
# Print info that we're creating a bug report
log.info("Started creating archive for bug report")
# Check that there are no overrides in place - we need to use a little
# hack to detect whether the custom_dir setting was used without parsing
# mkdocs.yml again - we check at which position the directory provided
# by the theme resides, and if it's not the first one, abort.
base = utils.get_theme_dir(config.theme.name)
if config.theme.dirs.index(base):
log.error("Please remove 'custom_dir' setting.")
self._help_and_exit()
# Check that there are no hooks in place
if config.hooks:
log.error("Please remove 'hooks' setting.")
self._help_and_exit()
# Create in-memory archive
archive = BytesIO()
archive_name = self.config.archive_name
# Create self-contained example from project
files = []
with ZipFile(archive, "a", ZIP_DEFLATED, False) as f:
for path in ["mkdocs.yml", "requirements.txt"]:
if os.path.isfile(path):
f.write(path, os.path.join(archive_name, path))
# Append all files visible to MkDocs
for file in get_files(config):
path = os.path.relpath(file.abs_src_path, os.path.curdir)
f.write(path, os.path.join(archive_name, path))
# Add dependency tree as returned by pipdeptree
f.writestr(
os.path.join(archive_name, "dependencies.json"),
render_json_tree(PackageDAG.from_pkgs(
get_installed_distributions(local_only = True)
), 2)
)
# Retrieve list of processed files
for a in f.filelist:
files.append("".join([
Fore.LIGHTBLACK_EX, a.filename, " ",
Fore.GREEN, _size(a.compress_size)
]))
# Finally, write archive to disk
buffer = archive.getbuffer()
with open(f"{archive_name}.zip", "wb") as f:
f.write(archive.getvalue())
# Print summary and archive name
log.info("Archive successfully created:")
print(Style.NORMAL)
print("".join([
" ", f.name, " ",
Fore.GREEN, _size(buffer.nbytes)
]))
print(Style.RESET_ALL)
# Print sorted list of archived files
files.sort()
for file in files:
print(f" {file}")
# Aaaaaand done.
print(Style.RESET_ALL)
sys.exit(1)
# -------------------------------------------------------------------------
# Print explanation and exit
def _help_and_exit(self):
print(Fore.RED)
print(" When reporting issues, you must remove all customizations")
print(" and check if the problem persists. If not, the problem is")
print(" caused by your overrides. Please understand that we can't")
print(" provide support for customizations. Please remove:")
print(Style.NORMAL)
print(" - theme.custom_dir")
print(" - hooks")
print(Fore.YELLOW)
print(" Additionally, please remove all third-party JavaScript or")
print(" CSS not explicitly mentioned in our documentation:")
print(Style.NORMAL)
print(" - extra_css")
print(" - extra_javascript")
print(Style.RESET_ALL)
sys.exit(1)
# -----------------------------------------------------------------------------
# Helper functions
# -----------------------------------------------------------------------------
# Print human-readable size
def _size(value):
for unit in ["B", "kB", "MB", "GB", "TB", "PB", "EB", "ZB"]:
if abs(value) < 1000.0:
return f"{value:3.1f} {unit}"
value /= 1000.0
# -----------------------------------------------------------------------------
# Data
# -----------------------------------------------------------------------------
# Set up logging
log = logging.getLogger("mkdocs")
log.addFilter(DuplicateFilter())