35 lines
1,006 B
Python
35 lines
1,006 B
Python
|
""" Processing of docstrings, which means parsing for types. """
|
||
|
|
|
||
|
|
import re
|
||
|
|
|
||
|
|
import evaluate
|
||
|
|
|
||
|
|
|
||
|
|
#@evaluate.memoize_default() # TODO add
|
||
|
|
def follow_param(param):
|
||
|
|
func = param.parent_function
|
||
|
|
#print func, param, param.parent_function
|
||
|
|
param_str = search_param_in_docstr(func.docstr, str(param.get_name()))
|
||
|
|
|
||
|
|
if param_str is not None:
|
||
|
|
scope = func.get_parent_until()
|
||
|
|
return evaluate.get_scopes_for_name(scope, param_str,
|
||
|
|
search_global=True)
|
||
|
|
return []
|
||
|
|
|
||
|
|
|
||
|
|
def search_param_in_docstr(docstr, param_str):
|
||
|
|
lines = docstr.split('\n')
|
||
|
|
|
||
|
|
# look at #40 to see definitions of those params
|
||
|
|
sphinx_comp = ':type %s:' % param_str
|
||
|
|
googley_comp = re.compile('\s*%s\s+\(([^()]+)\)' % re.escape(param_str))
|
||
|
|
for l in lines:
|
||
|
|
if l.startswith(sphinx_comp):
|
||
|
|
return l.replace(sphinx_comp, '', 1).strip()
|
||
|
|
|
||
|
|
r = re.match(googley_comp, l)
|
||
|
|
if r is not None:
|
||
|
|
return r.group(1)
|
||
|
|
return None
|