Function _grab_overloads

tools/generate_proxy_methods.py:133–175

grab @overload entries for a function, assuming black-formatted code ;) so that we can do a simple regex

(fn)

def _grab_overloads(fn):
    class="st">"""grab @overload entries for a function, assuming black-formatted
    code ;) so that we can do a simple regex

    class="st">"""

    class="cm"># functions that use @util.deprecated and whatnot will have a string
    class="cm"># generated fn.  we can look at __wrapped__ but these functions don't
    class="cm"># have any overloads in any case right now so skip
    if fn.__code__.co_filename == class="st">"<string>":
        return []

    with open(fn.__code__.co_filename) as f:
        lines = [l for i, l in zip(range(fn.__code__.co_firstlineno), f)]

        lines.reverse()

    output = []

    current_ov = []
    for line in lines[1:]:
        current_ov.append(line)
        outside_block_match = re.match(rclass="st">"^\w", line)
        if outside_block_match:
            current_ov[:] = []
            break

        fn_match = re.match(rclass="st">"^    (?:    )?(?:async )?def (.*)\(", line)
        if fn_match and fn_match.group(1) != fn.__name__:
            current_ov[:] = []
            break

        ov_match = re.match(rclass="st">"^    (?:    )?@overload$", line)
        if ov_match:
            output.append(class="st">"".join(reversed(current_ov)))
            current_ov[:] = []

        if re.match(rclass="st">"^    if (?:typing\.)?TYPE_CHECKING:", line):
            output.append(line)
            current_ov[:] = []

    output.reverse()
    return output


def process_class(