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 234b06f17f0b41797922d5bb9854ef6650a19efe..2d9efd5939798360b10a96f7eb1f731fa0d87fcf 100644 (file)
--- 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"},