fixes #671

Author: jph00

fastcore/docments.py

@@ -48,16 +48,18 @@ def get_dataclass_source(s):
 # %% ../nbs/06_docments.ipynb
 def get_source(s):
     "Get source code for string, function object or dataclass `s`"
-    return getsource(s) if isfunction(s) or ismethod(s) else get_dataclass_source(s) if isdataclass(s) else s
+    if isinstance(s,str): return s
+    return getsource(s) if isfunction(s) or ismethod(s) else get_dataclass_source(s) if isdataclass(s) else None
 
 # %% ../nbs/06_docments.ipynb
 def _parses(s):
     "Parse Python code in string, function object or dataclass `s`"
-    return parse(dedent(get_source(s)))
+    return parse(dedent(get_source(s) or ''))
 
 def _tokens(s):
     "Tokenize Python code in string or function object `s`"
     s = get_source(s)
+    if not s: return []
     return tokenize(BytesIO(s.encode('utf-8')).readline)
 
 _clean_re = re.compile(r'^\s*#(.*)\s*$')

nbs/06_docments.ipynb

@@ -213,7 +213,8 @@
     "#|export\n",
     "def get_source(s):\n",
     "    \"Get source code for string, function object or dataclass `s`\"\n",
-    "    return getsource(s) if isfunction(s) or ismethod(s) else get_dataclass_source(s) if isdataclass(s) else s"
+    "    if isinstance(s,str): return s\n",
+    "    return getsource(s) if isfunction(s) or ismethod(s) else get_dataclass_source(s) if isdataclass(s) else None"
    ]
   },
   {
@@ -225,11 +226,12 @@
     "#|export\n",
     "def _parses(s):\n",
     "    \"Parse Python code in string, function object or dataclass `s`\"\n",
-    "    return parse(dedent(get_source(s)))\n",
+    "    return parse(dedent(get_source(s) or ''))\n",
     "\n",
     "def _tokens(s):\n",
     "    \"Tokenize Python code in string or function object `s`\"\n",
     "    s = get_source(s)\n",
+    "    if not s: return []\n",
     "    return tokenize(BytesIO(s.encode('utf-8')).readline)\n",
     "\n",
     "_clean_re = re.compile(r'^\\s*#(.*)\\s*$')\n",