Function _fix_default_values

bpython/inspection.py:176–215

Functions taking default arguments that are references to other objects will cause breakage, so we swap out the object itself with the name it was referenced with in the source by parsing the source itself!

(f: Callable, argspec: ArgSpec)

def _fix_default_values(f: Callable, argspec: ArgSpec) -> ArgSpec:
    """Functions taking default arguments that are references to other objects
    will cause breakage, so we swap out the object itself with the name it was
    referenced with in the source by parsing the source itself!"""

    if argspec.defaults is None and argspec.kwonly_defaults is None:
        # No keyword args, no need to do anything
        return argspec

    try:
        src, _ = inspect.getsourcelines(f)
    except (OSError, IndexError):
        # IndexError is raised in inspect.findsource(), can happen in
        # some situations. See issue #94.
        return argspec
    except TypeError:
        # No source code is available, so replace the default values with what we have.
        if argspec.defaults is not None:
            argspec.defaults = [_Repr(str(value)) for value in argspec.defaults]
        if argspec.kwonly_defaults is not None:
            argspec.kwonly_defaults = {
                key: _Repr(str(value))
                for key, value in argspec.kwonly_defaults.items()
            }
        return argspec

    kwparsed = parsekeywordpairs("".join(src))

    if argspec.defaults is not None:
        values = list(argspec.defaults)
        keys = argspec.args[-len(values) :]
        for i, key in enumerate(keys):
            values[i] = _Repr(kwparsed[key])

        argspec.defaults = values
    if argspec.kwonly_defaults is not None:
        for key in argspec.kwonly_defaults.keys():
            argspec.kwonly_defaults[key] = _Repr(kwparsed[key])

    return argspec


_getpydocspec_re = LazyReCompile(