Create database if it does not exist

Author: defnull

bbblb/model.py

@@ -41,6 +41,8 @@
 from sqlalchemy.orm import relationship
 from sqlalchemy.exc import NoResultFound, IntegrityError, OperationalError  # noqa: F401
 
+import sqlalchemy_utils
+
 LOG = logging.getLogger(__name__)
 
 P = typing.ParamSpec("P")
@@ -62,15 +64,19 @@ async def init_engine(db: str, echo=False):
     global async_engine, AsyncSessionMaker, ScopedSession
 
     if db.startswith("sqlite://"):
-        db = db.replace("sqlite://", "sqlite+aiosqlite://")
+        adb = db.replace("sqlite://", "sqlite+aiosqlite://")
     elif db.startswith("postgresql://"):
-        db = db.replace("postgresql://", "postgresql+asyncpg://")
+        adb = db.replace("postgresql://", "postgresql+asyncpg://")
     else:
         raise ValueError(
             f"Unsupported database dialect: {db} (must be sqlite:// or postgresql://)"
         )
 
-    async_engine = create_async_engine(db, echo=echo)
+    if not await asyncio.to_thread(sqlalchemy_utils.database_exists, db):
+        LOG.warning("Database does not exist, trying to create it...")
+        await asyncio.to_thread(sqlalchemy_utils.create_database, db)
+
+    async_engine = create_async_engine(adb, echo=echo)
     AsyncSessionMaker = async_sessionmaker(async_engine, expire_on_commit=False)
 
     ScopedSession = async_scoped_session(

pyproject.toml

@@ -19,6 +19,7 @@ dependencies = [
     "lxml>=6.0.2",
     "pyjwt>=2.10.1",
     "sqlalchemy>=2.0.44",
+    "sqlalchemy-utils>=0.42.0",
     "starlette>=0.48.0",
 ]