From: CaselIT <cfederico87@gmail.com>
Date: Mon, 23 Aug 2021 22:14:58 +0000 (+0200)
Subject: append the path to write_pyi script.
X-Git-Tag: rel_1_7_0~9
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=ff5f001c007ca62df19e6c4aa5616fa4435790cc;p=thirdparty%2Fsqlalchemy%2Falembic.git

append the path to write_pyi script.

This allows running without having alembic installed

Change-Id: I1cc4f108916233e7b149485a42cae759e7e3b577
---

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"},