From 74118c8ded9af90fa6c9e856c733e213922a67a4 Mon Sep 17 00:00:00 2001 From: David Lenwell Date: Mon, 21 Oct 2013 01:22:17 -0700 Subject: [PATCH] switched to new models --- refstack/web.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/refstack/web.py b/refstack/web.py index d8ca8d35..7da9cd78 100755 --- a/refstack/web.py +++ b/refstack/web.py @@ -27,8 +27,8 @@ from wtforms import Form, BooleanField, TextField, \ PasswordField, validators from flask_mail import Mail -from app import app -from models import Vendor, Cloud, User, db +from refstack.app import app +from refstack.models import * mail = Mail(app) @@ -43,9 +43,9 @@ class SecureView(ModelView): """ """ return g.user is not None -admin.add_view(SecureView(Vendor, db.session)) -admin.add_view(SecureView(Cloud, db.session)) -admin.add_view(SecureView(User, db.session)) +admin.add_view(SecureView(Vendor, db)) +admin.add_view(SecureView(Cloud, db)) +admin.add_view(SecureView(User, db)) @app.before_request