Skip to content

Commit 49f6bc3

Browse files
committed
Updates database template with database check and creation
1 parent 13ffd2d commit 49f6bc3

File tree

2 files changed

+7
-0
lines changed

2 files changed

+7
-0
lines changed

pyblaze/cli/templates/database_template.txt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,13 @@
11
from sqlalchemy import create_engine
22
from sqlalchemy.orm import declarative_base, scoped_session, sessionmaker
3+
from sqlalchemy_utils import database_exists, create_database
4+
35

46
engine = create_engine("{{database_url}}")
7+
8+
if not database_exists(engine.url):
9+
create_database(engine.url)
10+
511
db_session = scoped_session(
612
sessionmaker(autocommit=False, autoflush=False, bind=engine)
713
)

pyproject.toml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@ dependencies = [
1616
"sqlalchemy",
1717
"bcrypt",
1818
"inflect",
19+
"sqlalchemy-utils"
1920
]
2021

2122
[project.scripts]

0 commit comments

Comments
 (0)