dycw-utilities 0.150.12__tar.gz → 0.150.13__tar.gz

{dycw_utilities-0.150.12 → dycw_utilities-0.150.13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dycw-utilities
-Version: 0.150.12
+Version: 0.150.13
 Author-email: Derek Wan <d.wan@icloud.com>
 License-File: LICENSE
 Requires-Python: >=3.12

{dycw_utilities-0.150.12 → dycw_utilities-0.150.13}/pyproject.toml

@@ -232,7 +232,7 @@ dependencies = [
 name = "dycw-utilities"
 readme = "README.md"
 requires-python = ">= 3.12"
-version = "0.150.12"
+version = "0.150.13"
 
 [project.entry-points.pytest11]
 pytest-randomly = "utilities.pytest_plugins.pytest_randomly"
@@ -265,7 +265,7 @@ test = [
 # bump-my-version
 [tool.bumpversion]
 allow_dirty = true
-current_version = "0.150.12"
+current_version = "0.150.13"
 
 [[tool.bumpversion.files]]
 filename = "src/utilities/__init__.py"

{dycw_utilities-0.150.12 → dycw_utilities-0.150.13}/src/tests/test_sqlalchemy.py

@@ -67,6 +67,7 @@ from utilities.sqlalchemy import (
     _SelectedOrAll,
     _tuple_to_mapping,
     check_connect,
+    check_connect_async,
     check_engine,
     columnwise_max,
     columnwise_min,
@@ -137,9 +138,12 @@ def _upsert_lists(
 
 
 class TestCheckConnect:
-    def test_main(self, *, test_engine: Engine) -> None:
+    def test_sync(self, *, test_engine: Engine) -> None:
         assert check_connect(test_engine)
 
+    async def test_async(self, *, test_async_engine: AsyncEngine) -> None:
+        assert await check_connect_async(test_async_engine)
+
 
 class TestCheckEngine:
     async def test_main(self, *, test_async_engine: AsyncEngine) -> None:

{dycw_utilities-0.150.12 → dycw_utilities-0.150.13}/src/utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.150.12"
+__version__ = "0.150.13"

{dycw_utilities-0.150.12 → dycw_utilities-0.150.13}/src/utilities/sqlalchemy.py

@@ -115,11 +115,23 @@ CHUNK_SIZE_FRAC = 0.8
 ##
 
 
+_SELECT = text("SELECT 1")
+
+
 def check_connect(engine: Engine, /) -> bool:
     """Check if an engine can connect."""
     try:
         with engine.connect() as conn:
-            return bool(conn.execute(text("SELECT 1")).scalar_one())
+            return bool(conn.execute(_SELECT).scalar_one())
+    except (OperationalError, ProgrammingError):
+        return False
+
+
+async def check_connect_async(engine: AsyncEngine, /) -> bool:
+    """Check if an engine can connect."""
+    try:
+        async with engine.connect() as conn:
+            return bool((await conn.execute(_SELECT)).scalar_one())
     except (OperationalError, ProgrammingError):
         return False
 
@@ -1260,6 +1272,7 @@ __all__ = [
     "TablenameMixin",
     "UpsertItemsError",
     "check_connect",
+    "check_connect_async",
     "check_engine",
     "columnwise_max",
     "columnwise_min",