限制Flask-Admin上的一个视图?

时间:2014-07-14 17:18:38

标签: python flask flask-admin flask-security flask-principal

我有一个Flask应用程序的文件views.py

views.py内,我在此处设置了Flask-Admin和相应的sqlalchemy数据库:

db = SQLAlchemy(flaskapp)

def build_db():

import random
import datetime

db.drop_all()
db.create_all()

# Create sample Users
testuser = []
testgroup = []
testlevel = []

user_list = []
for i in range(len(testuser)):
    user = User()
    user.testuser = testuser[i]
    user.testlevel = testlevel[i]
    user.testgroup = testgroup[i]

    user_list.append(user)
    db.session.add(user)

for user in user_list:
    entry = random.choice(sample_text)  # select text at random
    post = Post()
    post.user = user
    post.title = entry['title']
    post.text = entry['content']
    tmp = int(1000*random.random())  # random number between 0 and 1000:
    post.date = datetime.datetime.now() - datetime.timedelta(days=tmp)
    post.tags = random.sample(tag_list, 2)  # select a couple of tags at random
    db.session.add(post)

db.session.commit()
return

# Create models
class User(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    testuser = db.Column(db.String(100))
    testgroup = db.Column(db.String(100))
    testlevel = db.Column(db.String(100))

# Required for administrative interface. For python 3 please use __str__ instead.
def __repr__(self):
    return self.testuser

class UserInfo(db.Model):
    id = db.Column(db.Integer, primary_key=True)

    key = db.Column(db.String(64), nullable=False)
    value = db.Column(db.String(64))

    user_id = db.Column(db.Integer(), db.ForeignKey(User.id))
    user = db.relationship(User, backref='info')

def __repr__(self):
    return '%s - %s' % (self.key, self.value)

### Attempt at doing something...
class MyView(ModelView):
    @expose('/admin/userview', methods=('GET', 'POST'))
    def adminview():
        return "yes"
##ADMIN
admin = Admin(flaskapp, name="Test Aptly")
admin.add_view(MyView(User, db.session))

问题。

如何阻止访问" / admin / userview"页面取决于用户的级别(使用数据库定义" testlevel")?用户有3个品质,两个用户的例子是:

testuser          testgroup        testlevel

joe               it               admin

john              dev              basic

如何将管理页面限制为仅允许testlevel="admin"人使用? 我是否使用Flask-Security?瓶校长?如果是这样,怎么样?我一直在修补这两点,到目前为止我还没有做任何工作。

这是一个例子" / admin / userview"看起来像是:example Flask-Admin

  

首先,您可以使用各种类级属性进行配置   应该展示什么以及如何展示。例如,可以使用column_list   显示一些列或包含相关的额外列   模型。

     

例如

class UserView(ModelView):
    # Show only name and email columns in list view
    column_list = ('name', 'email')

    # Enable search functionality - it will search for terms in
    # name and email fields
    column_searchable_list = ('name', 'email')

    # Add filters for name and email columns
    column_filters = ('name', 'email')
  

或者,您可以覆盖一些ModelView方法和   实现你的自定义逻辑。

     

例如,如果您需要为其提供额外的字段   生成的表单,你可以这样做:

class UserView(ModelView):
    def scaffold_form(self):
        form_class = super(UserView, self).scaffold_form()
        form_class.extra = wtf.TextField('Extra')
        return form_class
  

检查flask.ext.admin.contrib.sqlamodel documentation以获取列表   配置属性和方法。谢谢!

1 个答案:

答案 0 :(得分:1)

我假设您已创建管理员视图的子类BaseView,并且您正在使用Flask-login。

然后覆盖视图类中的is_accessible方法,以检查当前用户的质量:

from flask.ext.admin.base import BaseView
from flask.ext.login import current_user

class MyView(BaseView):
    def is_accessible(self):
        return current_user.testlevel == 'admin'

希望这有帮助!