From 9b0b7b0ec7c0d5ceac281521fa1780a4513f6672 Mon Sep 17 00:00:00 2001
From: Jagraj Aulakh <aulakhjagraj@gmail.com>
Date: Fri, 17 Mar 2023 01:49:15 -0400
Subject: [PATCH] #7 Fixed boot script to properly upgrade DB.

---
 backend/boot.sh | 5 ++---
 backend/main.py | 3 ++-
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/backend/boot.sh b/backend/boot.sh
index cabacd4..0743b5d 100755
--- a/backend/boot.sh
+++ b/backend/boot.sh
@@ -1,11 +1,10 @@
 #!/bin/bash
 
-source ./venv/bin/activate
+. ./venv/bin/activate
 
 pip install -r ./requirements.txt
 
-flask db update
-
+flask db upgrade
 
 exec gunicorn -b :5000 --access-logfile - --error-logfile - main:app
 
diff --git a/backend/main.py b/backend/main.py
index 15d9e6f..1cbaba5 100644
--- a/backend/main.py
+++ b/backend/main.py
@@ -2,10 +2,11 @@ import dotenv
 dotenv.load_dotenv()
 
 from app import create_app, db
+from app.models import User
 
 app = create_app()
 
 @app.shell_context_processor
 def make_shell_context():
-    return {}
+    return {'db': db, 'User': User}