diff options
| author | CaselIT <cfederico87@gmail.com> | 2022-09-11 22:38:46 +0200 |
|---|---|---|
| committer | CaselIT <cfederico87@gmail.com> | 2022-09-12 21:00:52 +0200 |
| commit | 0e83fddb6c110bf1658564c248ffad9163a365fa (patch) | |
| tree | 9b61207cd621fb526cda1daa55eaf94bd50aeb25 /tools | |
| parent | 747ec301529cf2a08d56d3596aedbf54a93b8742 (diff) | |
| download | alembic-0e83fddb6c110bf1658564c248ffad9163a365fa.tar.gz | |
Improve typing
Change-Id: I9fc86c4a92e1b76d19c9e891ff08ce8a46ad4e35
Diffstat (limited to 'tools')
| -rw-r--r-- | tools/write_pyi.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/tools/write_pyi.py b/tools/write_pyi.py index cf42d1b..52fac3c 100644 --- a/tools/write_pyi.py +++ b/tools/write_pyi.py @@ -29,6 +29,7 @@ IGNORE_ITEMS = { } TRIM_MODULE = [ "alembic.runtime.migration.", + "alembic.operations.base.", "alembic.operations.ops.", "sqlalchemy.engine.base.", "sqlalchemy.sql.schema.", @@ -85,6 +86,8 @@ def generate_pyi_for_proxy( module = sys.modules[cls.__module__] env = { + **typing.__dict__, + **sa.sql.schema.__dict__, **sa.__dict__, **sa.types.__dict__, **ops.__dict__, @@ -141,7 +144,7 @@ def _generate_stub_for_meth(cls, name, printer, env, is_context_manager): annotations = typing.get_type_hints(fn, env) spec.annotations.update(annotations) except NameError as e: - pass + print(f"{cls.__name__}.{name} NameError: {e}", file=sys.stderr) name_args = spec[0] assert name_args[0:1] == ["self"] or name_args[0:1] == ["cls"] |