add scanfilter parameter, move _iter_modules out of method scope, addresses #99

Author: poke1024

pdoc/__init__.py

@@ -479,6 +479,23 @@ def __lt__(self, other):
         return self.refname < other.refname
 
 
+def _iter_modules(paths):
+    """
+    Custom implementation of `pkgutil.iter_modules()`
+    because that one doesn't play well with namespace packages.
+    See: https://github.com/pypa/setuptools/issues/83
+    """
+    from os.path import isdir, join, splitext
+    for pth in paths:
+        for file in os.listdir(pth):
+            if file.startswith(('.', '__pycache__', '__init__.py')):
+                continue
+            if file.endswith(_SOURCE_SUFFIXES):
+                yield splitext(file)[0]
+            if isdir(join(pth, file)) and '.' not in file:
+                yield file
+
+
 class Module(Doc):
     """
     Representation of a module's documentation.
@@ -491,7 +508,8 @@ class Module(Doc):
     __slots__ = ('supermodule', 'doc', '_context', '_is_inheritance_linked')
 
     def __init__(self, module: Union[ModuleType, str], *, docfilter: Callable[[Doc], bool] = None,
-                 supermodule: 'Module' = None, context: Context = None):
+                 supermodule: 'Module' = None, context: Context = None,
+                 scanfilter: Callable[[str], bool] = None):
         """
         Creates a `Module` documentation object given the actual
         module Python object.
@@ -563,23 +581,7 @@ def is_from_this_module(obj):
         # If the module is a package, scan the directory for submodules
         if self.is_package:
 
-            def iter_modules(paths):
-                """
-                Custom implementation of `pkgutil.iter_modules()`
-                because that one doesn't play well with namespace packages.
-                See: https://github.com/pypa/setuptools/issues/83
-                """
-                from os.path import isdir, join, splitext
-                for pth in paths:
-                    for file in os.listdir(pth):
-                        if file.startswith(('.', '__pycache__', '__init__.py')):
-                            continue
-                        if file.endswith(_SOURCE_SUFFIXES):
-                            yield splitext(file)[0]
-                        if isdir(join(pth, file)) and '.' not in file:
-                            yield file
-
-            for root in iter_modules(self.obj.__path__):
+            for root in _iter_modules(self.obj.__path__):
                 # Ignore if this module was already doc'd.
                 if root in self.doc:
                     continue
@@ -590,9 +592,13 @@ def iter_modules(paths):
 
                 assert self.refname == self.name
                 fullname = "%s.%s" % (self.name, root)
+
+                if scanfilter and not scanfilter(fullname):
+                    continue
+
                 self.doc[root] = m = Module(import_module(fullname),
                                             docfilter=docfilter, supermodule=self,
-                                            context=self._context)
+                                            context=self._context, scanfilter=scanfilter)
                 # Skip empty namespace packages because they may
                 # as well be other auxiliary directories
                 if m.is_namespace and not m.doc: