Create database if it does not exist

Author: defnull

bbblb/model.py

@@ -10,6 +10,7 @@
 import typing
 import uuid
 from uuid import UUID
+import sqlalchemy
 from sqlalchemy.ext.asyncio import create_async_engine
 from contextlib import asynccontextmanager
 
@@ -41,6 +42,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")
@@ -70,6 +73,10 @@ async def init_engine(db: str, echo=False):
             f"Unsupported database dialect: {db} (must be sqlite:// or postgresql://)"
         )
 
+    if not sqlalchemy_utils.database_exists(db):
+        LOG.warning("Database does not exist, trying to create it...")
+        sqlalchemy_utils.create_database(db)
+
     async_engine = create_async_engine(db, echo=echo)
     AsyncSessionMaker = async_sessionmaker(async_engine, expire_on_commit=False)
 
@@ -79,6 +86,12 @@ async def init_engine(db: str, echo=False):
     )
 
     async with async_engine.begin() as conn:
+
+        if "postgres" in async_engine.url.drivername:
+            rows = (await conn.execute(sqlalchemy.text("SELECT datname FROM pg_database"))).all()
+            if async_engine.url.database not in [row[0] for row in rows]:
+                await conn.execute(sqlalchemy.text(f"CREATE DATABASE {async_engine.url.database} ENCODING 'utf-8"))
+
         # Creating tables is not transactional in some databases, so we just try
         # our luck and if that fails, we sleep a couple of ms and try again.
         try: