Finding functions in Jupyter notebooks by matching names and signatures.

find_functions.py

import inspect
import sys
import types
from typing import Dict

def global_functions() -> Dict[str, str]:
    result = {}
    module = sys.modules['__main__']
    
    for key in dir(module):
        value = getattr(module, key)
        if not isinstance(value, types.FunctionType):
            continue
        
        source = inspect.getfile(value)
        if not source.endswith('.ipynb') and not 'ipykernel' in source:
            continue

        result[key + str(inspect.signature(value))] = (
            '(this notebook)' if 'ipykernel' in source
            else source[source.rindex('/') + 1:])
    
    return result
        
def find_functions(*terms: str) -> Dict[str, str]:
    return {
        sig: source
        for sig, source in global_functions().items()
        if all(term in sig for term in terms)
    }