From 46a608794d5f4c6b9e8a194507db830ac0869a8b Mon Sep 17 00:00:00 2001
From: Patrick Smith <patrick+dds@promptworks.com>
Date: Thu, 2 Aug 2018 10:47:31 -0400
Subject: [PATCH] Update alembic/env to work with flask app

---
 alembic/env.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/alembic/env.py b/alembic/env.py
index 6a9d9749..91b96364 100644
--- a/alembic/env.py
+++ b/alembic/env.py
@@ -30,11 +30,10 @@ sys.path.append(parent_dir)
 
 from atst.app import make_config
 app_config = make_config()
-config.set_main_option('sqlalchemy.url', app_config['default']['DATABASE_URI'])
+config.set_main_option('sqlalchemy.url', app_config['DATABASE_URI'])
 
-from atst.database import make_db
+from atst.database import db
 from atst.models import *
-db = make_db(app_config)
 target_metadata = Base.metadata