switched to new models

This commit is contained in:
David Lenwell
2013-10-21 01:22:17 -07:00
parent 66a53ef2fc
commit 74118c8ded

View File

@@ -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