dycw-utilities 0.150.11__py3-none-any.whl → 0.150.13__py3-none-any.whl

{dycw_utilities-0.150.11.dist-info → dycw_utilities-0.150.13.dist-info}/METADATA

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

{dycw_utilities-0.150.11.dist-info → dycw_utilities-0.150.13.dist-info}/RECORD

@@ -1,4 +1,4 @@
-utilities/__init__.py,sha256=-W11lQr4jP-tn91Yfbt2z87iYMIXPcyPiNogVGYlrG0,61
+utilities/__init__.py,sha256=Dso32ACb66Psk-daIQDsHv_bmVx1YtOxzv9SWRzvkOk,61
 utilities/altair.py,sha256=92E2lCdyHY4Zb-vCw6rEJIsWdKipuu-Tu2ab1ufUfAk,9079
 utilities/asyncio.py,sha256=2m2a2C-Qgc6OHTTHL332-t66A7xDITt_SORT7a1DJWo,16792
 utilities/atomicwrites.py,sha256=xcOWenTBRS0oat3kg7Sqe51AohNThMQ2ixPL7QCG8hw,5795
@@ -66,7 +66,7 @@ utilities/sentinel.py,sha256=3jIwgpMekWgDAxPDA_hXMP2St43cPhciKN3LWiZ7kv0,1248
 utilities/shelve.py,sha256=4OzjQI6kGuUbJciqf535rwnao-_IBv66gsT6tRGiUt0,759
 utilities/slack_sdk.py,sha256=ppFBvKgfg5IRWiIoKPtpTyzBtBF4XmwEvU3I5wLJikM,2140
 utilities/socket.py,sha256=K77vfREvzoVTrpYKo6MZakol0EYu2q1sWJnnZqL0So0,118
-utilities/sqlalchemy.py,sha256=u2ZVvr_4Yrqsur6FlLHUuW0mPaoNrHvaJ2wj7o_WG3A,39264
+utilities/sqlalchemy.py,sha256=PVQXLRXPvULmGpKaa21lM21fUDzEIcqx_-d5oQFdD8A,39610
 utilities/sqlalchemy_polars.py,sha256=18AoEbeNJUKF3-5hroNy9J5LQwS_QJAXbMfKc9sChtk,14250
 utilities/statsmodels.py,sha256=koyiBHvpMcSiBfh99wFUfSggLNx7cuAw3rwyfAhoKpQ,3410
 utilities/string.py,sha256=MB0X6UPTUc06JdAdj-PctZ238IXeCjE5dAJibNw6ZrU,587
@@ -89,8 +89,8 @@ utilities/zoneinfo.py,sha256=oEH-nL3t4h9uawyZqWDtNtDAl6M-CLpLYGI_nI6DulM,1971
 utilities/pytest_plugins/__init__.py,sha256=U4S_2y3zgLZVfMenHRaJFBW8yqh2mUBuI291LGQVOJ8,35
 utilities/pytest_plugins/pytest_randomly.py,sha256=NXzCcGKbpgYouz5yehKb4jmxmi2SexKKpgF4M65bi10,414
 utilities/pytest_plugins/pytest_regressions.py,sha256=Iwhfv_OJH7UCPZCfoh7ugZ2Xjqjil-BBBsOb8sDwiGI,1471
-dycw_utilities-0.150.11.dist-info/METADATA,sha256=i9tIQJnes-TSj9Z8tHap_MgTtlzmZuXvVAPcoG_JY-Q,1697
-dycw_utilities-0.150.11.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-dycw_utilities-0.150.11.dist-info/entry_points.txt,sha256=BOD_SoDxwsfJYOLxhrSXhHP_T7iw-HXI9f2WVkzYxvQ,135
-dycw_utilities-0.150.11.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
-dycw_utilities-0.150.11.dist-info/RECORD,,
+dycw_utilities-0.150.13.dist-info/METADATA,sha256=WnT6T9ZLZ13ELERVuzIbh3HGGjn9pDj0_JVykzsUezs,1697
+dycw_utilities-0.150.13.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+dycw_utilities-0.150.13.dist-info/entry_points.txt,sha256=BOD_SoDxwsfJYOLxhrSXhHP_T7iw-HXI9f2WVkzYxvQ,135
+dycw_utilities-0.150.13.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
+dycw_utilities-0.150.13.dist-info/RECORD,,

utilities/__init__.py

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

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