From ff5f001c007ca62df19e6c4aa5616fa4435790cc Mon Sep 17 00:00:00 2001 From: CaselIT Date: Tue, 24 Aug 2021 00:14:58 +0200 Subject: [PATCH] append the path to write_pyi script. This allows running without having alembic installed Change-Id: I1cc4f108916233e7b149485a42cae759e7e3b577 --- tools/write_pyi.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/tools/write_pyi.py b/tools/write_pyi.py index 234b06f1..2d9efd59 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"}, -- 2.47.2