Skip to content

Commit 13141fd

Browse files
committed
Add SQL commands for new columns
1 parent c9c0b0e commit 13141fd

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

app.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -451,6 +451,22 @@ def init_db() -> None: # initialize database
451451
"ALTER TABLE user ADD COLUMN days_completed INT NOT NULL DEFAULT 1"
452452
)
453453
) # create days completed column
454+
if "combo_multiplier" not in [
455+
column["name"] for column in db.inspect(db.engine).get_columns("user")
456+
]: # check if combo multiplier column is not in the user table
457+
db.session.execute(
458+
text(
459+
"ALTER TABLE user ADD COLUMN combo_multiplier INT NOT NULL DEFAULT 0"
460+
)
461+
) # create combo multiplier column
462+
if "last_task_completed" not in [
463+
column["name"] for column in db.inspect(db.engine).get_columns("user")
464+
]: # check if last task completed column is not in the user table
465+
db.session.execute(
466+
text(
467+
"ALTER TABLE user ADD COLUMN last_task_completed INT NOT NULL DEFAULT -1"
468+
)
469+
) # create last task completed column
454470
if User.query.count() == 0: # if there are no users
455471
new_user = User(username="Player") # create new user
456472
db.session.add(new_user) # add new user to the database

0 commit comments

Comments
 (0)