haldis/app/admin.py

29 lines
713 B
Python
Raw Normal View History

2015-03-26 21:49:14 +01:00
from flask import url_for, redirect
from flask.ext.admin import Admin, BaseView, expose
from flask.ext.admin.contrib.sqla import ModelView
from flask.ext import login
from app import app, db
from models import User
from utils import send_command
class ModelBaseView(ModelView):
def is_accessible(self):
if login.current_user.is_anonymous():
return False
return login.current_user.is_admin()
class UserAdminModel(ModelBaseView):
column_searchable_list = ('username',)
inline_models = None
form_columns = ('username', 'admin')
admin = Admin(app, name='FoodBot', url='/foodbot/admin', template_mode='bootstrap3')
admin.add_view(UserAdminModel(User, db.session))