dycw-utilities 0.153.7__py3-none-any.whl → 0.153.8__py3-none-any.whl

{dycw_utilities-0.153.7.dist-info → dycw_utilities-0.153.8.dist-info}/METADATA

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

{dycw_utilities-0.153.7.dist-info → dycw_utilities-0.153.8.dist-info}/RECORD

@@ -1,4 +1,4 @@
-utilities/__init__.py,sha256=9aK37ZvGGPt34vOfZ9frrf2lxPzuQCzBgDOTjJ48qjo,60
+utilities/__init__.py,sha256=fSwBHH_QjQiuW9YKDHQhOfKP46dlOwjxz1pfnRc0A_w,60
 utilities/altair.py,sha256=92E2lCdyHY4Zb-vCw6rEJIsWdKipuu-Tu2ab1ufUfAk,9079
 utilities/asyncio.py,sha256=QXkTtugXkqtYt7Do23zgYErqzdp6jwzPpV_SP9fJ1gI,16780
 utilities/atomicwrites.py,sha256=tPo6r-Rypd9u99u66B9z86YBPpnLrlHtwox_8Z7T34Y,5790
@@ -64,7 +64,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=QCRqlFUH5Iz9jkiR0IUvb5btx_ijdbYX48rlXxQTVXA,39591
+utilities/sqlalchemy.py,sha256=bSwgIOd09Pmd7D39Rv2gl6t9qHz1iYQIrPnyDUWz7pc,40411
 utilities/sqlalchemy_polars.py,sha256=Mm-sShZfqqgnzTrupMQdCfSM2akrybXHXAErTs-ofM8,14244
 utilities/statsmodels.py,sha256=koyiBHvpMcSiBfh99wFUfSggLNx7cuAw3rwyfAhoKpQ,3410
 utilities/string.py,sha256=shmBK87zZwzGyixuNuXCiUbqzfeZ9xlrFwz6JTaRvDk,582
@@ -87,8 +87,8 @@ utilities/zoneinfo.py,sha256=FBMcUQ4662Aq8SsuCL1OAhDQiyANmVjtb-C30DRrWoE,1966
 utilities/pytest_plugins/__init__.py,sha256=U4S_2y3zgLZVfMenHRaJFBW8yqh2mUBuI291LGQVOJ8,35
 utilities/pytest_plugins/pytest_randomly.py,sha256=B1qYVlExGOxTywq2r1SMi5o7btHLk2PNdY_b1p98dkE,409
 utilities/pytest_plugins/pytest_regressions.py,sha256=9v8kAXDM2ycIXJBimoiF4EgrwbUvxTycFWJiGR_GHhM,1466
-dycw_utilities-0.153.7.dist-info/METADATA,sha256=Fpe8vz0U4b-uAwUz5yo8RsIfTtVfWjVEiS5l3SG1aU8,1696
-dycw_utilities-0.153.7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-dycw_utilities-0.153.7.dist-info/entry_points.txt,sha256=BOD_SoDxwsfJYOLxhrSXhHP_T7iw-HXI9f2WVkzYxvQ,135
-dycw_utilities-0.153.7.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
-dycw_utilities-0.153.7.dist-info/RECORD,,
+dycw_utilities-0.153.8.dist-info/METADATA,sha256=kGabwftmrxbs70yKIeBiQDg0K-kZZwemtsaDauXny18,1696
+dycw_utilities-0.153.8.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+dycw_utilities-0.153.8.dist-info/entry_points.txt,sha256=BOD_SoDxwsfJYOLxhrSXhHP_T7iw-HXI9f2WVkzYxvQ,135
+dycw_utilities-0.153.8.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
+dycw_utilities-0.153.8.dist-info/RECORD,,

utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.153.7"
+__version__ = "0.153.8"

utilities/sqlalchemy.py

@@ -30,7 +30,6 @@ from typing import (
 )
 
 import sqlalchemy
-import sqlalchemy.ext.asyncio
 from sqlalchemy import (
     URL,
     Column,
@@ -63,7 +62,7 @@ from sqlalchemy.exc import (
     OperationalError,
     ProgrammingError,
 )
-from sqlalchemy.ext.asyncio import AsyncConnection, AsyncEngine
+from sqlalchemy.ext.asyncio import AsyncConnection, AsyncEngine, create_async_engine
 from sqlalchemy.orm import (
     DeclarativeBase,
     InstrumentedAttribute,
@@ -311,7 +310,7 @@ def create_engine(
         case False:
             return sqlalchemy.create_engine(url, poolclass=poolclass)
         case True:
-            return sqlalchemy.ext.asyncio.create_async_engine(url, poolclass=poolclass)
+            return create_async_engine(url, poolclass=poolclass)
         case never:
             assert_never(never)
 
@@ -319,6 +318,27 @@ def create_engine(
 ##
 
 
+async def ensure_database_created(super_: URL, database: str, /) -> None:
+    """Ensure a database is created."""
+    engine = create_async_engine(super_, isolation_level="AUTOCOMMIT")
+    async with engine.begin() as conn:
+        try:
+            _ = await conn.execute(text(f"CREATE DATABASE {database}"))
+        except (OperationalError, ProgrammingError) as error:
+            if not search('database ".*" already exists', ensure_str(one(error.args))):
+                raise
+
+
+async def ensure_database_dropped(super_: URL, database: str, /) -> None:
+    """Ensure a database is dropped."""
+    engine = create_async_engine(super_, isolation_level="AUTOCOMMIT")
+    async with engine.begin() as conn:
+        _ = await conn.execute(text(f"DROP DATABASE IF EXISTS {database}"))
+
+
+##
+
+
 async def ensure_tables_created(
     engine: AsyncEngine,
     /,
@@ -1284,6 +1304,8 @@ __all__ = [
     "columnwise_max",
     "columnwise_min",
     "create_engine",
+    "ensure_database_created",
+    "ensure_database_dropped",
     "ensure_tables_created",
     "ensure_tables_dropped",
     "enum_name",