support cst.If at module level

src/docc/__init__.py

@@ -17,5 +17,5 @@
 The documentation compiler.
 """
 
-__version__ = "0.6.0"
+__version__ = "0.6.1"
 "Current version of docc"

src/docc/plugins/python/cst.py

@@ -873,6 +873,15 @@ def enter(self, node: Node) -> Visit:
             visit = self.enter_assign(node, cst_node)
         elif isinstance(cst_node, cst.SimpleStatementLine):
             visit = Visit.TraverseChildren
+        elif isinstance(cst_node, cst.IndentedBlock):
+            visit = Visit.TraverseChildren
+        elif isinstance(cst_node, cst.If):
+            node.find_child(cst_node.body).visit(self)
+            if cst_node.orelse is not None:
+                node.find_child(cst_node.orelse).visit(self)
+            visit = Visit.SkipChildren
+        elif isinstance(cst_node, cst.Else):
+            visit = Visit.TraverseChildren
         elif isinstance(cst_node, cst.Expr):
             visit = Visit.SkipChildren
         elif isinstance(cst_node, (cst.Import, cst.ImportFrom)):
@@ -917,6 +926,12 @@ def exit(self, node: Node) -> None:
             self.exit_assign()
         elif isinstance(cst_node, cst.SimpleStatementLine):
             pass
+        elif isinstance(cst_node, cst.IndentedBlock):
+            pass
+        elif isinstance(cst_node, cst.If):
+            pass
+        elif isinstance(cst_node, cst.Else):
+            pass
         elif isinstance(cst_node, cst.Expr):
             pass
         elif isinstance(cst_node, (cst.ImportFrom, cst.Import)):

whitelist.txt

@@ -40,6 +40,7 @@ matcher
 merchantability
 metaclass
 modulefinder
+orelse
 param
 params
 pathname2url