Updates database template with database check and creation

cicekhayri · Dec 13, 2023 · 49f6bc3 · 49f6bc3
1 parent 13ffd2d
commit 49f6bc3
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/pyblaze/cli/templates/database_template.txt b/pyblaze/cli/templates/database_template.txt
@@ -1,7 +1,13 @@
 from sqlalchemy import create_engine
 from sqlalchemy.orm import declarative_base, scoped_session, sessionmaker
+from sqlalchemy_utils import database_exists, create_database
+
 
 engine = create_engine("{{database_url}}")
+
+if not database_exists(engine.url):
+    create_database(engine.url)
+
 db_session = scoped_session(
     sessionmaker(autocommit=False, autoflush=False, bind=engine)
 )

diff --git a/pyproject.toml b/pyproject.toml
@@ -16,6 +16,7 @@ dependencies = [
     "sqlalchemy",
     "bcrypt",
     "inflect",
+    "sqlalchemy-utils"
 ]
 
 [project.scripts]