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

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

@@ -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 → dycw_utilities-0.150.13}/pyproject.toml

@@ -7,103 +7,232 @@ requires = ["hatchling"]
 
 # dependency groups
 [dependency-groups]
-altair = ["altair >= 5.5.0, < 5.6"]
+altair = [
+  "altair >=5.5.0, <5.6",
+]
 altair-test = [
   "polars",
   "img2pdf",
   "vl-convert-python",
 ]
-atools = ["atools >= 0.14.2, < 0.15"]
-cachetools = ["cachetools >= 6.1.0, < 6.2"]
-click = ["click >= 8.2.1, < 8.3"]
+atools = [
+  "atools >=0.14.2, <0.15",
+]
+cachetools = [
+  "cachetools >=6.1.0, <6.2",
+]
+click = [
+  "click >=8.2.1, <8.3",
+]
 core = [
-  "atomicwrites >= 1.4.1, < 1.5",
-  "typing-extensions >= 4.14.1, < 4.15",
-  "tzlocal >= 5.3.1, < 5.4",
-  "whenever >= 0.8.7, < 0.9",
-]
-cryptography = ["cryptography >= 45.0.4, < 45.1"]
-cvxpy = ["cvxpy >= 1.7.1, < 1.8"]
-dataclasses-test = ["orjson", "polars"]
+  "atomicwrites >=1.4.1, <1.5",
+  "typing-extensions >=4.14.1, <4.15",
+  "tzlocal >=5.3.1, <5.4",
+  "whenever >=0.8.7, <0.9",
+]
+cryptography = [
+  "cryptography >=45.0.4, <45.1",
+]
+cvxpy = [
+  "cvxpy >=1.7.1, <1.8",
+]
+dataclasses-test = [
+  "orjson",
+  "polars",
+]
 dev = [
-  "coloredlogs >= 15.0.1, < 15.1",
-  "coverage-conditional-plugin >= 0.9.0, < 0.10",
-  "dycw-pytest-only >= 2.1.1, < 2.2",
-  "pudb >= 2025.1, < 2025.2",
-  "pyright[nodejs] >= 1.1.401, < 1.2",
-  "pytest-cov >= 6.2.1, < 6.3",
-  "pytest-timeout >= 2.4.0, < 2.5",
-]
-dnspython = ["dnspython >= 2.7.0, < 2.8"]
-eventkit = ["eventkit >= 1.0.3, < 1.1"]
-fastapi = ["fastapi >= 0.116.0, < 0.117"]
-fastapi-test = ["httpx", "uvicorn"]
-fpdf2 = ["fpdf2 >= 2.8.3, < 2.9"]
-hashlib-test = ["orjson"]
-http-test = ["orjson"]
-hypothesis = ["hypothesis >= 6.136.1, < 6.137"]
-hypothesis-test = ["libcst", "pathvalidate", "numpy", "pytest-rerunfailures"]
-inflect = ["inflect >= 7.5.0, < 7.6"]
-jupyter-test = ["pandas", "polars"]
-libcst = ["libcst >= 1.8.0, < 1.9"]
-lightweight-charts = ["lightweight-charts >= 2.1, < 2.2"]
-lightweight-charts-test = ["polars-lts-cpu", "pyarrow"]
-math-test = ["numpy"]
-memory-profiler = ["memory-profiler >= 0.61.0, < 0.62"]
-more-itertools = ["more-itertools >= 10.7.0, < 10.8"]
-numpy = ["numpy >= 2.3.1, < 2.4"]
-operator = ["polars"]
-optuna = ["optuna >= 4.4.0, < 4.5"]
-orjson = ["orjson >= 3.11.0, < 3.12"]
-orjson-test = ["polars"]
-polars = ["polars >= 1.31.0, < 1.32"]
-polars-ols = ["polars-ols >= 0.3.5, < 0.4"]
-polars-ols-test = ["scikit-learn"]
-polars-test = ["numpy", "statsmodels"]
-pottery = ["pottery >= 3.0.1, < 3.1"]
-pottery-test = ["orjson", "pytest-rerunfailures"]
-pqdm = ["pqdm >= 0.2.0, < 0.3"]
-psutil = ["psutil >= 7.0.0, < 7.1"]
-pyinstrument = ["pyinstrument >= 5.0.3, < 5.1"]
+  "coloredlogs >=15.0.1, <15.1",
+  "coverage-conditional-plugin >=0.9.0, <0.10",
+  "dycw-pytest-only >=2.1.1, <2.2",
+  "pudb >=2025.1, <2025.2",
+  "pyright[nodejs] >=1.1.401, <1.2",
+  "pytest-cov >=6.2.1, <6.3",
+  "pytest-timeout >=2.4.0, <2.5",
+]
+dnspython = [
+  "dnspython >=2.7.0, <2.8",
+]
+eventkit = [
+  "eventkit >=1.0.3, <1.1",
+]
+fastapi = [
+  "fastapi >=0.116.0, <0.117",
+]
+fastapi-test = [
+  "httpx",
+  "uvicorn",
+]
+fpdf2 = [
+  "fpdf2 >=2.8.3, <2.9",
+]
+hashlib-test = [
+  "orjson",
+]
+http-test = [
+  "orjson",
+]
+hypothesis = [
+  "hypothesis >=6.136.1, <6.137",
+]
+hypothesis-test = [
+  "libcst",
+  "pathvalidate",
+  "numpy",
+  "pytest-rerunfailures",
+]
+inflect = [
+  "inflect >=7.5.0, <7.6",
+]
+jupyter-test = [
+  "pandas",
+  "polars",
+]
+libcst = [
+  "libcst >=1.8.0, <1.9",
+]
+lightweight-charts = [
+  "lightweight-charts >=2.1, <2.2",
+]
+lightweight-charts-test = [
+  "polars-lts-cpu",
+  "pyarrow",
+]
+math-test = [
+  "numpy",
+]
+memory-profiler = [
+  "memory-profiler >=0.61.0, <0.62",
+]
+more-itertools = [
+  "more-itertools >=10.7.0, <10.8",
+]
+numpy = [
+  "numpy >=2.3.1, <2.4",
+]
+operator = [
+  "polars",
+]
+optuna = [
+  "optuna >=4.4.0, <4.5",
+]
+orjson = [
+  "orjson >=3.11.0, <3.12",
+]
+orjson-test = [
+  "polars",
+]
+polars = [
+  "polars >=1.31.0, <1.32",
+]
+polars-ols = [
+  "polars-ols >=0.3.5, <0.4",
+]
+polars-ols-test = [
+  "scikit-learn",
+]
+polars-test = [
+  "numpy",
+  "statsmodels",
+]
+pottery = [
+  "pottery >=3.0.1, <3.1",
+]
+pottery-test = [
+  "orjson",
+  "pytest-rerunfailures",
+]
+pqdm = [
+  "pqdm >=0.2.0, <0.3",
+]
+psutil = [
+  "psutil >=7.0.0, <7.1",
+]
+pyinstrument = [
+  "pyinstrument >=5.0.3, <5.1",
+]
 pytest = [
-  "pudb >= 2025.1, < 2025.2",
-  "pytest >= 8.4.1, < 8.5",
-  "pytest-asyncio >= 1.1.0, < 1.2",
-  "pytest-randomly >= 3.16.0, < 3.17",
-  "pytest-timeout >= 2.4.0, < 2.5",
-  "pytest-xdist >= 3.8.0, < 3.9",
-]
-pytest-regressions = ["pytest-regressions >= 2.8.1, < 2.9"]
-pytest-regressions-test = ["orjson", "polars"]
-pytest-test = ["orjson", "pytest-rng", "pytest-rerunfailures"]
-redis = ["redis >= 6.2.0, < 6.3", "orjson"]
-redis-test = ["pytest-rerunfailures"]
-reprlib-test = ["rich"]
-scipy = ["scipy >= 1.16.0, < 1.17"]
-sklearn = ["scikit-learn >= 1.7.0, < 1.8"]
-slack-sdk = ["slack-sdk >= 3.36.0, < 3.37"]
-slack-sdk-test = ["aiohttp"]
-sqlalchemy = ["sqlalchemy >= 2.0.41, < 2.1", "psycopg"]
-sqlalchemy-polars = ["sqlalchemy", "polars"]
-sqlalchemy-polars-test = ["aiosqlite", "asyncpg", "greenlet"]
-sqlalchemy-test = ["aiosqlite", "asyncpg", "greenlet"]
-statsmodels = ["statsmodels >= 0.14.4, < 0.15"]
-typed-settings = ["typed-settings >= 24.6.0, < 24.7"]
-tzdata = ["tzdata >= 2025.2, < 2025.3"]
+  "pudb >=2025.1, <2025.2",
+  "pytest >=8.4.1, <8.5",
+  "pytest-asyncio >=1.1.0, <1.2",
+  "pytest-randomly >=3.16.0, <3.17",
+  "pytest-timeout >=2.4.0, <2.5",
+  "pytest-xdist >=3.8.0, <3.9",
+]
+pytest-regressions = [
+  "pytest-regressions >=2.8.1, <2.9",
+]
+pytest-regressions-test = [
+  "orjson",
+  "polars",
+]
+pytest-test = [
+  "orjson",
+  "pytest-rng",
+  "pytest-rerunfailures",
+]
+redis = [
+  "redis >=6.2.0, <6.3",
+  "orjson",
+]
+redis-test = [
+  "pytest-rerunfailures",
+]
+reprlib-test = [
+  "rich",
+]
+scipy = [
+  "scipy >=1.16.0, <1.17",
+]
+sklearn = [
+  "scikit-learn >=1.7.0, <1.8",
+]
+slack-sdk = [
+  "slack-sdk >=3.36.0, <3.37",
+]
+slack-sdk-test = [
+  "aiohttp",
+]
+sqlalchemy = [
+  "sqlalchemy >=2.0.41, <2.1",
+  "psycopg",
+]
+sqlalchemy-polars = [
+  "sqlalchemy",
+  "polars",
+]
+sqlalchemy-polars-test = [
+  "aiosqlite",
+  "asyncpg",
+  "greenlet",
+]
+sqlalchemy-test = [
+  "aiosqlite",
+  "asyncpg",
+  "greenlet",
+]
+statsmodels = [
+  "statsmodels >=0.14.4, <0.15",
+]
+typed-settings = [
+  "typed-settings >=24.6.0, <24.7",
+]
+tzdata = [
+  "tzdata >=2025.2, <2025.3",
+]
 
 # project
 [project]
 authors = [{name = "Derek Wan", email = "d.wan@icloud.com"}]
 dependencies = [
-  "atomicwrites >= 1.4.1, < 1.5",
-  "typing-extensions >= 4.14.0, < 4.15",
-  "tzlocal >= 5.3.1, < 5.4",
-  "whenever >= 0.8.7, < 0.9",
+  "atomicwrites >=1.4.1, <1.5",
+  "typing-extensions >=4.14.0, <4.15",
+  "tzlocal >=5.3.1, <5.4",
+  "whenever >=0.8.7, <0.9",
 ]
 name = "dycw-utilities"
 readme = "README.md"
 requires-python = ">= 3.12"
-version = "0.150.11"
+version = "0.150.13"
 
 [project.entry-points.pytest11]
 pytest-randomly = "utilities.pytest_plugins.pytest_randomly"
@@ -111,24 +240,24 @@ pytest-regressions = "utilities.pytest_plugins.pytest_regressions"
 
 [project.optional-dependencies]
 logging = [
-  "coloredlogs >= 15.0.1, < 15.1",
+  "coloredlogs >=15.0.1, <15.1",
 ]
 test = [
-  "dycw-pytest-only >= 2.1.1, < 2.2",
-  "hypothesis >= 6.136.1, < 6.137",
-  "pudb >= 2025.1, < 2025.2",
-  "pytest >= 8.4.1, < 8.5",
-  "pytest-asyncio >= 1.1.0, < 1.2",
-  "pytest-cov >= 6.2.1, < 6.3",
-  "pytest-instafail >= 0.5.0, < 0.6",
-  "pytest-lazy-fixtures >= 1.1.4, < 1.2",
-  "pytest-randomly >= 3.16.0, < 3.17",
-  "pytest-regressions >= 2.8.1, < 2.9",
-  "pytest-repeat >= 0.9.4, < 0.10",
-  "pytest-rerunfailures >= 15.1, < 16",
-  "pytest-rng >= 1.0.0, < 1.1",
-  "pytest-timeout >= 2.4.0, < 2.5",
-  "pytest-xdist >= 3.8.0, < 3.9",
+  "dycw-pytest-only >=2.1.1, <2.2",
+  "hypothesis >=6.136.1, <6.137",
+  "pudb >=2025.1, <2025.2",
+  "pytest >=8.4.1, <8.5",
+  "pytest-asyncio >=1.1.0, <1.2",
+  "pytest-cov >=6.2.1, <6.3",
+  "pytest-instafail >=0.5.0, <0.6",
+  "pytest-lazy-fixtures >=1.1.4, <1.2",
+  "pytest-randomly >=3.16.0, <3.17",
+  "pytest-regressions >=2.8.1, <2.9",
+  "pytest-repeat >=0.9.4, <0.10",
+  "pytest-rerunfailures >=15.1, <16",
+  "pytest-rng >=1.0.0, <1.1",
+  "pytest-timeout >=2.4.0, <2.5",
+  "pytest-xdist >=3.8.0, <3.9",
 ]
 
 [project.scripts]
@@ -136,7 +265,7 @@ test = [
 # bump-my-version
 [tool.bumpversion]
 allow_dirty = true
-current_version = "0.150.11"
+current_version = "0.150.13"
 
 [[tool.bumpversion.files]]
 filename = "src/utilities/__init__.py"

{dycw_utilities-0.150.11 → 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.11 → dycw_utilities-0.150.13}/src/utilities/__init__.py

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

{dycw_utilities-0.150.11 → 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",