Fixed #4

http-samc · Jun 22, 2021 · b7c1775 · b7c1775
1 parent fc8021e
commit b7c1775
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/gen_doc/main.py b/gen_doc/main.py
@@ -114,7 +114,7 @@ def extractDocStrings(filepath: str, parent: str = None, classSections: bool = F
 
         if not isinstance(node, list):
             functionDocString = ast.get_docstring(node)
-            if fence: functionDocString = f"```Python\n{functionDocString}\n```"
+            if fence and functionDocString: functionDocString = f"```Python\n{functionDocString}\n```"
 
             if functionDocString is None and excludeDocless: continue
             retStr += f"### {relPathFormatted}.`{node.name}` [function]\n"
@@ -155,7 +155,8 @@ def extractDocStrings(filepath: str, parent: str = None, classSections: bool = F
         del node[0] # Removing top-level ClassDef, only iterating through class' nodes
         for function in node:
             functionDocString = ast.get_docstring(function)
-            if fence: functionDocString = f"```Python\n{functionDocString}\n```"
+            if fence and functionDocString: functionDocString = f"```Python\n{functionDocString}\n```"
+
 
             if functionDocString is None and excludeDocless: continue
             functionName = function.name.replace('_', '\_') # preventing MD from turning __func__ to *func*