edgedb · msullivan · Nov 27, 2024 · Nov 26, 2024 · Nov 27, 2024
diff --git a/edgedb/__init__.py b/edgedb/__init__.py
@@ -1,4 +1,7 @@
 # Auto-generated shim
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from gel import *  # noqa
 import gel as _mod
 import sys as _sys
 _cur = _sys.modules['edgedb']

diff --git a/edgedb/_taskgroup.py b/edgedb/_taskgroup.py
@@ -1,4 +1,7 @@
 # Auto-generated shim
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from gel._taskgroup import *  # noqa
 import gel._taskgroup as _mod
 import sys as _sys
 _cur = _sys.modules['edgedb._taskgroup']

diff --git a/edgedb/_version.py b/edgedb/_version.py
@@ -1,4 +1,7 @@
 # Auto-generated shim
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from gel._version import *  # noqa
 import gel._version as _mod
 import sys as _sys
 _cur = _sys.modules['edgedb._version']

diff --git a/edgedb/abstract.py b/edgedb/abstract.py
@@ -1,4 +1,7 @@
 # Auto-generated shim
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from gel.abstract import *  # noqa
 import gel.abstract as _mod
 import sys as _sys
 _cur = _sys.modules['edgedb.abstract']

diff --git a/edgedb/ai/__init__.py b/edgedb/ai/__init__.py
@@ -1,4 +1,7 @@
 # Auto-generated shim
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from gel.ai import *  # noqa
 import gel.ai as _mod
 import sys as _sys
 _cur = _sys.modules['edgedb.ai']

diff --git a/edgedb/ai/core.py b/edgedb/ai/core.py
@@ -1,4 +1,7 @@
 # Auto-generated shim
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from gel.ai.core import *  # noqa
 import gel.ai.core as _mod
 import sys as _sys
 _cur = _sys.modules['edgedb.ai.core']

diff --git a/edgedb/ai/types.py b/edgedb/ai/types.py
@@ -1,4 +1,7 @@
 # Auto-generated shim
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from gel.ai.types import *  # noqa
 import gel.ai.types as _mod
 import sys as _sys
 _cur = _sys.modules['edgedb.ai.types']

diff --git a/edgedb/asyncio_client.py b/edgedb/asyncio_client.py
@@ -1,4 +1,7 @@
 # Auto-generated shim
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from gel.asyncio_client import *  # noqa
 import gel.asyncio_client as _mod
 import sys as _sys
 _cur = _sys.modules['edgedb.asyncio_client']

diff --git a/edgedb/base_client.py b/edgedb/base_client.py
@@ -1,4 +1,7 @@
 # Auto-generated shim
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from gel.base_client import *  # noqa
 import gel.base_client as _mod
 import sys as _sys
 _cur = _sys.modules['edgedb.base_client']

diff --git a/edgedb/blocking_client.py b/edgedb/blocking_client.py
@@ -1,4 +1,7 @@
 # Auto-generated shim
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from gel.blocking_client import *  # noqa
 import gel.blocking_client as _mod
 import sys as _sys
 _cur = _sys.modules['edgedb.blocking_client']

diff --git a/edgedb/codegen.py b/edgedb/codegen.py
@@ -1,4 +1,7 @@
 # Auto-generated shim
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from gel.codegen import *  # noqa
 import gel.codegen as _mod
 import sys as _sys
 _cur = _sys.modules['edgedb.codegen']

diff --git a/edgedb/color.py b/edgedb/color.py
@@ -1,4 +1,7 @@
 # Auto-generated shim
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from gel.color import *  # noqa
 import gel.color as _mod
 import sys as _sys
 _cur = _sys.modules['edgedb.color']

diff --git a/edgedb/con_utils.py b/edgedb/con_utils.py
@@ -1,4 +1,7 @@
 # Auto-generated shim
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from gel.con_utils import *  # noqa
 import gel.con_utils as _mod
 import sys as _sys
 _cur = _sys.modules['edgedb.con_utils']

diff --git a/edgedb/credentials.py b/edgedb/credentials.py
@@ -1,4 +1,7 @@
 # Auto-generated shim
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from gel.credentials import *  # noqa
 import gel.credentials as _mod
 import sys as _sys
 _cur = _sys.modules['edgedb.credentials']

diff --git a/edgedb/datatypes/__init__.py b/edgedb/datatypes/__init__.py
@@ -1,4 +1,7 @@
 # Auto-generated shim
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from gel.datatypes import *  # noqa
 import gel.datatypes as _mod
 import sys as _sys
 _cur = _sys.modules['edgedb.datatypes']

diff --git a/edgedb/datatypes/datatypes.py b/edgedb/datatypes/datatypes.py
@@ -1,4 +1,7 @@
 # Auto-generated shim
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from gel.datatypes.datatypes import *  # noqa
 import gel.datatypes.datatypes as _mod
 import sys as _sys
 _cur = _sys.modules['edgedb.datatypes.datatypes']

diff --git a/edgedb/datatypes/range.py b/edgedb/datatypes/range.py
@@ -1,4 +1,7 @@
 # Auto-generated shim
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from gel.datatypes.range import *  # noqa
 import gel.datatypes.range as _mod
 import sys as _sys
 _cur = _sys.modules['edgedb.datatypes.range']

diff --git a/edgedb/describe.py b/edgedb/describe.py
@@ -1,4 +1,7 @@
 # Auto-generated shim
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from gel.describe import *  # noqa
 import gel.describe as _mod
 import sys as _sys
 _cur = _sys.modules['edgedb.describe']

diff --git a/edgedb/enums.py b/edgedb/enums.py
@@ -1,4 +1,7 @@
 # Auto-generated shim
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from gel.enums import *  # noqa
 import gel.enums as _mod
 import sys as _sys
 _cur = _sys.modules['edgedb.enums']

diff --git a/edgedb/errors/__init__.py b/edgedb/errors/__init__.py
@@ -1,4 +1,7 @@
 # Auto-generated shim
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from gel.errors import *  # noqa
 import gel.errors as _mod
 import sys as _sys
 _cur = _sys.modules['edgedb.errors']

diff --git a/edgedb/errors/_base.py b/edgedb/errors/_base.py
@@ -1,4 +1,7 @@
 # Auto-generated shim
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from gel.errors._base import *  # noqa
 import gel.errors._base as _mod
 import sys as _sys
 _cur = _sys.modules['edgedb.errors._base']

diff --git a/edgedb/errors/tags.py b/edgedb/errors/tags.py
@@ -1,4 +1,7 @@
 # Auto-generated shim
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from gel.errors.tags import *  # noqa
 import gel.errors.tags as _mod
 import sys as _sys
 _cur = _sys.modules['edgedb.errors.tags']

diff --git a/edgedb/introspect.py b/edgedb/introspect.py
@@ -1,4 +1,7 @@
 # Auto-generated shim
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from gel.introspect import *  # noqa
 import gel.introspect as _mod
 import sys as _sys
 _cur = _sys.modules['edgedb.introspect']

diff --git a/edgedb/options.py b/edgedb/options.py
@@ -1,4 +1,7 @@
 # Auto-generated shim
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from gel.options import *  # noqa
 import gel.options as _mod
 import sys as _sys
 _cur = _sys.modules['edgedb.options']

diff --git a/edgedb/pgproto/__init__.py b/edgedb/pgproto/__init__.py
@@ -1,4 +1,7 @@
 # Auto-generated shim
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from gel.pgproto import *  # noqa
 import gel.pgproto as _mod
 import sys as _sys
 _cur = _sys.modules['edgedb.pgproto']

diff --git a/edgedb/pgproto/pgproto.py b/edgedb/pgproto/pgproto.py
@@ -1,4 +1,7 @@
 # Auto-generated shim
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from gel.pgproto.pgproto import *  # noqa
 import gel.pgproto.pgproto as _mod
 import sys as _sys
 _cur = _sys.modules['edgedb.pgproto.pgproto']

diff --git a/edgedb/pgproto/types.py b/edgedb/pgproto/types.py
@@ -1,4 +1,7 @@
 # Auto-generated shim
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from gel.pgproto.types import *  # noqa
 import gel.pgproto.types as _mod
 import sys as _sys
 _cur = _sys.modules['edgedb.pgproto.types']

diff --git a/edgedb/platform.py b/edgedb/platform.py
@@ -1,4 +1,7 @@
 # Auto-generated shim
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from gel.platform import *  # noqa
 import gel.platform as _mod
 import sys as _sys
 _cur = _sys.modules['edgedb.platform']

diff --git a/edgedb/protocol/__init__.py b/edgedb/protocol/__init__.py
@@ -1,4 +1,7 @@
 # Auto-generated shim
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from gel.protocol import *  # noqa
 import gel.protocol as _mod
 import sys as _sys
 _cur = _sys.modules['edgedb.protocol']

diff --git a/edgedb/protocol/asyncio_proto.py b/edgedb/protocol/asyncio_proto.py
@@ -1,4 +1,7 @@
 # Auto-generated shim
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from gel.protocol.asyncio_proto import *  # noqa
 import gel.protocol.asyncio_proto as _mod
 import sys as _sys
 _cur = _sys.modules['edgedb.protocol.asyncio_proto']

diff --git a/edgedb/protocol/blocking_proto.py b/edgedb/protocol/blocking_proto.py
@@ -1,4 +1,7 @@
 # Auto-generated shim
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from gel.protocol.blocking_proto import *  # noqa
 import gel.protocol.blocking_proto as _mod
 import sys as _sys
 _cur = _sys.modules['edgedb.protocol.blocking_proto']

diff --git a/edgedb/protocol/protocol.py b/edgedb/protocol/protocol.py
@@ -1,4 +1,7 @@
 # Auto-generated shim
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from gel.protocol.protocol import *  # noqa
 import gel.protocol.protocol as _mod
 import sys as _sys
 _cur = _sys.modules['edgedb.protocol.protocol']

diff --git a/edgedb/scram/__init__.py b/edgedb/scram/__init__.py
@@ -1,4 +1,7 @@
 # Auto-generated shim
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from gel.scram import *  # noqa
 import gel.scram as _mod
 import sys as _sys
 _cur = _sys.modules['edgedb.scram']

diff --git a/edgedb/scram/saslprep.py b/edgedb/scram/saslprep.py
@@ -1,4 +1,7 @@
 # Auto-generated shim
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from gel.scram.saslprep import *  # noqa
 import gel.scram.saslprep as _mod
 import sys as _sys
 _cur = _sys.modules['edgedb.scram.saslprep']

diff --git a/edgedb/transaction.py b/edgedb/transaction.py
@@ -1,4 +1,7 @@
 # Auto-generated shim
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from gel.transaction import *  # noqa
 import gel.transaction as _mod
 import sys as _sys
 _cur = _sys.modules['edgedb.transaction']

diff --git a/gel/__init__.py b/gel/__init__.py
@@ -59,6 +59,7 @@
     "EnumValue",
     "Executor",
     "IsolationLevel",
+    "MultiRange",
     "NamedTuple",
     "Object",
     "Range",

diff --git a/gel/abstract.py b/gel/abstract.py
@@ -33,6 +33,7 @@
     "QueryOptions",
     "QueryContext",
     "Executor",
+    "ExecuteContext",
     "AsyncIOExecutor",
     "ReadOnlyExecutor",
     "AsyncIOReadOnlyExecutor",

diff --git a/tools/make_import_shims.py b/tools/make_import_shims.py
@@ -22,6 +22,9 @@ def main():
         with open(fname, 'w') as f:
             f.write(f'''\
 # Auto-generated shim
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from {mod} import *  # noqa
 import {mod} as _mod
 import sys as _sys
 _cur = _sys.modules['{nmod}']