diff options
author | CaselIT <cfederico87@gmail.com> | 2021-08-24 00:14:58 +0200 |
---|---|---|
committer | CaselIT <cfederico87@gmail.com> | 2021-08-24 00:14:58 +0200 |
commit | ff5f001c007ca62df19e6c4aa5616fa4435790cc (patch) | |
tree | 662b2df8e6f1c45db36de1e22c4030869b13ed44 | |
parent | 7fd48061bbf893002b2d7a624b313b09ee0a9700 (diff) | |
download | alembic-ff5f001c007ca62df19e6c4aa5616fa4435790cc.tar.gz |
append the path to write_pyi script.
This allows running without having alembic installed
Change-Id: I1cc4f108916233e7b149485a42cae759e7e3b577
-rw-r--r-- | tools/write_pyi.py | 13 |
1 files changed, 8 insertions, 5 deletions
diff --git a/tools/write_pyi.py b/tools/write_pyi.py index 234b06f..2d9efd5 100644 --- a/tools/write_pyi.py +++ b/tools/write_pyi.py @@ -7,11 +7,14 @@ import textwrap from mako.pygen import PythonPrinter -from alembic.operations.base import Operations -from alembic.runtime.environment import EnvironmentContext -from alembic.script.write_hooks import console_scripts -from alembic.util.compat import inspect_formatargspec -from alembic.util.compat import inspect_getfullargspec +sys.path.append(str(Path(__file__).parent.parent)) + +if True: # avoid flake/zimports missing with the order + from alembic.operations.base import Operations + from alembic.runtime.environment import EnvironmentContext + from alembic.script.write_hooks import console_scripts + from alembic.util.compat import inspect_formatargspec + from alembic.util.compat import inspect_getfullargspec IGNORE_ITEMS = { "op": {"context", "create_module_class_proxy"}, |