forked from varia/varia.website
152 lines
4.8 KiB
Python
152 lines
4.8 KiB
Python
|
"""distutils.command.check
|
||
|
|
||
|
Implements the Distutils 'check' command.
|
||
|
"""
|
||
|
import contextlib
|
||
|
|
||
|
from ..core import Command
|
||
|
from ..errors import DistutilsSetupError
|
||
|
|
||
|
with contextlib.suppress(ImportError):
|
||
|
import docutils.utils
|
||
|
import docutils.parsers.rst
|
||
|
import docutils.frontend
|
||
|
import docutils.nodes
|
||
|
|
||
|
class SilentReporter(docutils.utils.Reporter):
|
||
|
def __init__(
|
||
|
self,
|
||
|
source,
|
||
|
report_level,
|
||
|
halt_level,
|
||
|
stream=None,
|
||
|
debug=0,
|
||
|
encoding='ascii',
|
||
|
error_handler='replace',
|
||
|
):
|
||
|
self.messages = []
|
||
|
super().__init__(
|
||
|
source, report_level, halt_level, stream, debug, encoding, error_handler
|
||
|
)
|
||
|
|
||
|
def system_message(self, level, message, *children, **kwargs):
|
||
|
self.messages.append((level, message, children, kwargs))
|
||
|
return docutils.nodes.system_message(
|
||
|
message, level=level, type=self.levels[level], *children, **kwargs
|
||
|
)
|
||
|
|
||
|
|
||
|
class check(Command):
|
||
|
"""This command checks the meta-data of the package."""
|
||
|
|
||
|
description = "perform some checks on the package"
|
||
|
user_options = [
|
||
|
('metadata', 'm', 'Verify meta-data'),
|
||
|
(
|
||
|
'restructuredtext',
|
||
|
'r',
|
||
|
(
|
||
|
'Checks if long string meta-data syntax '
|
||
|
'are reStructuredText-compliant'
|
||
|
),
|
||
|
),
|
||
|
('strict', 's', 'Will exit with an error if a check fails'),
|
||
|
]
|
||
|
|
||
|
boolean_options = ['metadata', 'restructuredtext', 'strict']
|
||
|
|
||
|
def initialize_options(self):
|
||
|
"""Sets default values for options."""
|
||
|
self.restructuredtext = 0
|
||
|
self.metadata = 1
|
||
|
self.strict = 0
|
||
|
self._warnings = 0
|
||
|
|
||
|
def finalize_options(self):
|
||
|
pass
|
||
|
|
||
|
def warn(self, msg):
|
||
|
"""Counts the number of warnings that occurs."""
|
||
|
self._warnings += 1
|
||
|
return Command.warn(self, msg)
|
||
|
|
||
|
def run(self):
|
||
|
"""Runs the command."""
|
||
|
# perform the various tests
|
||
|
if self.metadata:
|
||
|
self.check_metadata()
|
||
|
if self.restructuredtext:
|
||
|
if 'docutils' in globals():
|
||
|
try:
|
||
|
self.check_restructuredtext()
|
||
|
except TypeError as exc:
|
||
|
raise DistutilsSetupError(str(exc))
|
||
|
elif self.strict:
|
||
|
raise DistutilsSetupError('The docutils package is needed.')
|
||
|
|
||
|
# let's raise an error in strict mode, if we have at least
|
||
|
# one warning
|
||
|
if self.strict and self._warnings > 0:
|
||
|
raise DistutilsSetupError('Please correct your package.')
|
||
|
|
||
|
def check_metadata(self):
|
||
|
"""Ensures that all required elements of meta-data are supplied.
|
||
|
|
||
|
Required fields:
|
||
|
name, version
|
||
|
|
||
|
Warns if any are missing.
|
||
|
"""
|
||
|
metadata = self.distribution.metadata
|
||
|
|
||
|
missing = []
|
||
|
for attr in 'name', 'version':
|
||
|
if not getattr(metadata, attr, None):
|
||
|
missing.append(attr)
|
||
|
|
||
|
if missing:
|
||
|
self.warn("missing required meta-data: %s" % ', '.join(missing))
|
||
|
|
||
|
def check_restructuredtext(self):
|
||
|
"""Checks if the long string fields are reST-compliant."""
|
||
|
data = self.distribution.get_long_description()
|
||
|
for warning in self._check_rst_data(data):
|
||
|
line = warning[-1].get('line')
|
||
|
if line is None:
|
||
|
warning = warning[1]
|
||
|
else:
|
||
|
warning = '{} (line {})'.format(warning[1], line)
|
||
|
self.warn(warning)
|
||
|
|
||
|
def _check_rst_data(self, data):
|
||
|
"""Returns warnings when the provided data doesn't compile."""
|
||
|
# the include and csv_table directives need this to be a path
|
||
|
source_path = self.distribution.script_name or 'setup.py'
|
||
|
parser = docutils.parsers.rst.Parser()
|
||
|
settings = docutils.frontend.OptionParser(
|
||
|
components=(docutils.parsers.rst.Parser,)
|
||
|
).get_default_values()
|
||
|
settings.tab_width = 4
|
||
|
settings.pep_references = None
|
||
|
settings.rfc_references = None
|
||
|
reporter = SilentReporter(
|
||
|
source_path,
|
||
|
settings.report_level,
|
||
|
settings.halt_level,
|
||
|
stream=settings.warning_stream,
|
||
|
debug=settings.debug,
|
||
|
encoding=settings.error_encoding,
|
||
|
error_handler=settings.error_encoding_error_handler,
|
||
|
)
|
||
|
|
||
|
document = docutils.nodes.document(settings, reporter, source=source_path)
|
||
|
document.note_source(source_path, -1)
|
||
|
try:
|
||
|
parser.parse(data, document)
|
||
|
except AttributeError as e:
|
||
|
reporter.messages.append(
|
||
|
(-1, 'Could not finish the parsing: %s.' % e, '', {})
|
||
|
)
|
||
|
|
||
|
return reporter.messages
|