diff options
Diffstat (limited to 'tools/generate_proxy_methods.py')
| -rw-r--r-- | tools/generate_proxy_methods.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/tools/generate_proxy_methods.py b/tools/generate_proxy_methods.py index fd2070784..85f8b2320 100644 --- a/tools/generate_proxy_methods.py +++ b/tools/generate_proxy_methods.py @@ -154,16 +154,19 @@ def _grab_overloads(fn): current_ov[:] = [] break - fn_match = re.match(r"^ (?:async )?def (.*)\($", line) + fn_match = re.match(r"^ (?: )?(?:async )?def (.*)\($", line) if fn_match and fn_match.group(1) != fn.__name__: current_ov[:] = [] break - ov_match = re.match(r"^ @overload$", line) + ov_match = re.match(r"^ (?: )?@overload$", line) if ov_match: output.append("".join(reversed(current_ov))) current_ov[:] = [] + if re.match(r"^ if (?:typing\.)?TYPE_CHECKING:", line): + output.append(line) + output.reverse() return output |