Source code for pyquickhelper.sphinxext.sphinx_docassert_extension

# -*- coding: utf-8 -*-
Defines a :epkg:`sphinx` extension which if all parameters are documented.

import inspect
from docutils import nodes
import sphinx
from sphinx.util import logging
from sphinx.util.docfields import DocFieldTransformer, _is_single_paragraph
from .import_object_helper import import_any_object

[docs]def check_typed_make_field(self, types, # type: Dict[unicode, List[nodes.Node]] domain, # type: unicode items, # type: Tuple env=None, # type: BuildEnvironment # type inspect.Parameters (to check that all parameters are documented) parameters=None, function_name=None, # str docname=None, # str kind=None # str ): """ Overwrites function `make_field <>`_. Processes one argument of a function. :param self: from original function :param types: from original function :param domain: from original function :param items: from original function :param env: from original function :param parameters: list of known arguments for the function or method :param function_name: function name these arguments belong to :param docname: document which contains the object :param kind: tells which kind of object *function_name* is (function, method or class) Example of warnings it raises: :: [docassert] 'onefunction' has no parameter 'a' (in '...project_name\\subproject\\'). [docassert] 'onefunction' has undocumented parameters 'a, b' (...project_name\\subproject\\'). :githublink:`%|py|47` """ if parameters is None: parameters = None check_params = {} else: parameters = list(parameters) if kind == "method": parameters = parameters[1:] def kg(p): "local function" return p if isinstance(p, str) else check_params = {kg(p): 0 for p in parameters} logger = logging.getLogger("docassert") def check_item(fieldarg, content, logger): "local function" if fieldarg not in check_params: if function_name is not None: logger.warning("[docassert] '{0}' has no parameter '{1}' (in '{2}').".format( function_name, fieldarg, docname)) else: check_params[fieldarg] += 1 if check_params[fieldarg] > 1: logger.warning("[docassert] '{1}' of '{0}' is duplicated (in '{2}').".format( function_name, fieldarg, docname)) if isinstance(items, list): for fieldarg, content in items: check_item(fieldarg, content, logger) mini = None if len(check_params) == 0 else min(check_params.values()) if mini == 0: check_params = list(check_params.items()) nodoc = list(sorted(k for k, v in check_params if v == 0)) if len(nodoc) > 0: if len(nodoc) == 1 and nodoc[0] == 'self': # Behavior should be improved. pass else: logger.warning("[docassert] '{0}' has undocumented parameters '{1}' (in '{2}').".format( function_name, ", ".join(nodoc), docname)) else: # Documentation related to the return. pass
[docs]class OverrideDocFieldTransformer: """ Overrides one function with assigning it to a method :githublink:`%|py|96` """
[docs] def __init__(self, replaced): """ Constructor :param replaced: should be *DocFieldTransformer.transform* :githublink:`%|py|103` """ self.replaced = replaced
[docs] def override_transform(self, other_self, node): """ Transform a single field list *node*. Overwrite function `transform <>`_. It only adds extra verification and returns results from the replaced function. :param other_self: the builder :param node: node the replaced function changes or replace The function parses the original function and checks that the list of arguments declared by the function is the same the list of documented arguments. :githublink:`%|py|117` """ typemap = other_self.typemap entries = [] groupindices = {} # type: Dict[unicode, int] types = {} # type: Dict[unicode, Dict] # step 1: traverse all fields and collect field types and content for field in node: fieldname, fieldbody = field try: # split into field type and argument fieldtype, fieldarg = fieldname.astext().split(None, 1) except ValueError: # maybe an argument-less field type? fieldtype, fieldarg = fieldname.astext(), '' if fieldtype != "param": continue typedesc, is_typefield = typemap.get(fieldtype, (None, None)) # sort out unknown fields if typedesc is None or typedesc.has_arg != bool(fieldarg): # either the field name is unknown, or the argument doesn't # match the spec; capitalize field name and be done with it new_fieldname = fieldtype[0:1].upper() + fieldtype[1:] if fieldarg: new_fieldname += ' ' + fieldarg fieldname[0] = nodes.Text(new_fieldname) entries.append(field) continue typename = # collect the content, trying not to keep unnecessary paragraphs if _is_single_paragraph(fieldbody): content = fieldbody.children[0].children else: content = fieldbody.children # if the field specifies a type, put it in the types collection if is_typefield: # filter out only inline nodes; others will result in invalid # markup being written out content = [n for n in content if isinstance( n, (nodes.Inline, nodes.Text))] if content: types.setdefault(typename, {})[fieldarg] = content continue # also support syntax like ``:param type name:`` if typedesc.is_typed: try: argtype, argname = fieldarg.split(None, 1) except ValueError: pass else: types.setdefault(typename, {})[argname] = [ nodes.Text(argtype)] fieldarg = argname translatable_content = nodes.inline( fieldbody.rawsource, translatable=True) translatable_content.document = fieldbody.parent.document translatable_content.source = fieldbody.parent.source translatable_content.line = fieldbody.parent.line translatable_content += content # Import object, get the list of parameters docs = fieldbody.parent.source.split(":docstring of")[-1].strip() myfunc = None funckind = None function_name = None excs = [] try: myfunc, function_name, funckind = import_any_object(docs) except ImportError as e: excs.append(e) if myfunc is None: if len(excs) > 0: reasons = "\n".join(" {0}".format(e) for e in excs) else: reasons = "unknown" logger = logging.getLogger("docassert") logger.warning( "[docassert] unable to import object '{0}', reasons:\n{1}".format(docs, reasons)) myfunc = None if myfunc is None: signature = None parameters = None else: try: signature = inspect.signature(myfunc) parameters = signature.parameters except (TypeError, ValueError): logger = logging.getLogger("docassert") logger.warning( "[docassert] unable to get signature of '{0}'.".format(docs)) signature = None parameters = None # grouped entries need to be collected in one entry, while others # get one entry per field if typedesc.is_grouped: if typename in groupindices: group = entries[groupindices[typename]] else: groupindices[typename] = len(entries) group = [typedesc, []] entries.append(group) entry = typedesc.make_entry(fieldarg, [translatable_content]) group[1].append(entry) else: entry = typedesc.make_entry(fieldarg, [translatable_content]) entries.append([typedesc, entry]) # step 2: all entries are collected, check the parameters list. try: env = other_self.directive.state.document.settings.env except AttributeError as e: logger = logging.getLogger("docassert") logger.warning("[docassert] {0}".format(e)) env = None docname = fieldbody.parent.source.split(':docstring')[0] for entry in entries: if isinstance(entry, nodes.field): # raise NotImplementedError() logger = logging.getLogger("docassert") logger.warning( "[docassert] unable to checl [nodes.field] {0}".format(entry)) else: fieldtype, content = entry fieldtypes = types.get(, {}) check_typed_make_field(other_self, fieldtypes, other_self.directive.domain, content, env=env, parameters=parameters, function_name=function_name, docname=docname, kind=funckind) return self.replaced(other_self, node)
[docs]def setup_docassert(app): """ Setup for ``docassert`` extension (sphinx). This changes ``DocFieldTransformer.transform`` and replaces it by a function which calls the current function and does extra checking on the list of parameters. .. warning:: This class does not handle methods if the parameter name for the class is different from *self*. Classes included in other classes are not properly handled. :githublink:`%|py|271` """ inst = OverrideDocFieldTransformer(DocFieldTransformer.transform) def local_transform(me, node): "local function" return inst.override_transform(me, node) DocFieldTransformer.transform = local_transform return {'version': sphinx.__display_version__, 'parallel_read_safe': True}
[docs]def setup(app): "see @see fn setup_docassert" return setup_docassert(app)