diff options
| author | Federico Caselli <cfederico87@gmail.com> | 2022-10-18 19:57:19 +0200 |
|---|---|---|
| committer | Federico Caselli <cfederico87@gmail.com> | 2022-10-18 19:57:19 +0200 |
| commit | 3668889426bde1cfe31aca387a9c54fe56a66ff7 (patch) | |
| tree | 75f7c3f67caaa1f0dc926bef0eaa895018b1e1bb /tools/generate_proxy_methods.py | |
| parent | 974b1bd0fc40e11fc2886b5a9fc333feeeebf546 (diff) | |
| download | sqlalchemy-3668889426bde1cfe31aca387a9c54fe56a66ff7.tar.gz | |
Improve proxy methods with the changes from the abandoned
I5c7f172e5224fa08d0675160ac919d0b045a26eb
Change-Id: Ieb7971297f959fd635a4627bd93154e28ef0ef2b
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 |