加入收藏 | 设为首页 | 会员中心 | 我要投稿 李大同 (https://www.lidatong.com.cn/)- 科技、建站、经验、云计算、5G、大数据,站长网!
当前位置: 首页 > 编程开发 > Python > 正文

python – 如何将flask-login添加到flask-admin

发布时间:2020-12-20 13:07:25 所属栏目:Python 来源:网络整理
导读:登录在我的标准视图上运行良好,管理员工作得很好,但我似乎无法添加登录到我的管理员/ yikes !!! 这似乎很简单flask admin docs但是当我添加这一部分时 class MyView(BaseView): def is_accessible(self): return login.current_user.is_authenticated() def
登录在我的标准视图上运行良好,管理员工作得很好,但我似乎无法添加登录到我的管理员/ yikes !!!

这似乎很简单flask admin docs但是当我添加这一部分时

class MyView(BaseView):
    def is_accessible(self):
        return login.current_user.is_authenticated()

    def _handle_view(self,name,**kwargs):
        if not self.is_accessible():
            return redirect(url_for('login',next=request.url))

我没有被发送到未经身份验证的用户的登录页面.

知道我还需要做什么吗?

这在我的init文件中看起来是否正确?

from flask.ext.admin import BaseView


class MyView(BaseView):
    def is_accessible(self):
        return login.current_user.is_authenticated()

    def _handle_view(self,next=request.url))

app = Flask(__name__)
app.config.from_object('config')
migrate = Migrate(app,db)
manager = Manager(app)
manager.add_command('db',MigrateCommand)
Bootstrap(app)
from app import views,models
admin = Admin(app)


admin.add_view(ModelView(models.User,db.session))
admin.add_view(ModelView(models.rutable,db.session))
admin.add_view(ModelView(models.rustage,db.session))
admin.add_view(ModelView(models.TOC,db.session))
admin.add_view(ModelView(models.Request,db.session))
admin.add_view(ModelView(models.Staff,db.session))
admin.add_view(ModelView(models.Status,db.session))
admin.add_view(ModelView(models.Challenge,db.session))

这是我的登录视图,它连接到ldap auth

from flask.ext.admin import BaseView


def logged_in(f):
    @wraps(f)
    def decorated_function(*args,**kwargs):
        if session.get('logged_in') is not None:
            return f(*args,**kwargs)
        else:
            flash('Please log in first...','error')
            next_url = request.url
            login_url = '%s?next=%s' % (url_for('login'),next_url)
            return redirect(login_url)
    return decorated_function


@login_manager.unauthorized_handler
def unauthorized():
    print 'unauthorized'
    flash("You must be logged in.")
    return redirect(url_for("login"))

@login_manager.user_loader
def user_loader(user_id):
    """Given *user_id*,return the associated User object.
    :param unicode user_id: user_id (email) user to retrieve
    """
    g.user=current_user
    return models.User.query.get(user_id)

@app.route("/logout")
# @logged_in
def logout():
    logout_user()
    session.pop('logged_in',None)
    flash("Logged Out.")
    return redirect(url_for("login"))

@app.before_request
def before_request():
    g.user = current_user

@app.route("/login",methods=["GET","POST"])
def login():
    form = LoginForm()
    if form.validate_on_submit():
        if app.config['ENVIRONMENT']=='dev':
            try:
                print "Authentification Successful" 
                namedb=models.User.query.filter_by(name=unicode(form.username.data)).first()
                email=models.User.query.first().email         
                login_user(user_loader(unicode(email)),remember=form.remember_me.data)
                flash("Logged in successfully.")
                g.email=email
                session['logged_in'] = True
                return redirect( request.values.get('next') or url_for("main"))
            except Exception as e:
                flash("Invalid Credentials.")
                return render_template("login.html",form=form)
        else:
            try:
                if '@' in form.username.data:
                    form.username.data=re.sub(' /d+','',(re.sub("d+",form.username.data.split('@')[0]))[1:]+(re.sub("d+",form.username.data.split('@')[0]))[0:1])
                l = ldap.initialize("ldap://10.129.18.101")
                l.simple_bind_s("program%s" % form.username.data,form.password.data)
                print "Authentification Successful"
                r=l.search_s('cn=Users,dc=BHCS,dc=Internal',ldap.SCOPE_SUBTREE,'(sAMAccountName=*%s*)' % form.username.data,['mail','objectGUID','displayName'])
                email=r[0][1]['mail'][0]   
                GUID=r[0][1]['objectGUID'][0]   
                FullName=r[0][1]['displayName'][0] 
                import uuid
                guid = uuid.UUID(bytes=GUID)
                if not models.User.query.filter_by(email=unicode(email)).first(): 
                  p=models.User(name=FullName,email=email)
                  db.session.add(p)
                  db.session.commit()            
                login_user(user_loader(unicode(email)),form=form)
    return render_template("login.html",form=form)

根据要求,这是我的模型中的用户类

class User(db.Model):
    """An admin user capable of viewing reports.

    :param str email: email address of user
    :param str password: encrypted password for the user

    """
    __tablename__ = 'user'
    name=db.Column(db.String)
    email = db.Column(db.String,primary_key=True)
    authenticated = db.Column(db.Boolean,default=True)
    admin =  db.Column(db.Boolean,default=False)

    def is_active(self):
        """True,as all users are active."""
        return True

    def get_id(self):
        """Return the email address to satisfy Flask-Login's requirements."""
        return self.email

    def is_authenticated(self):
        """Return True if the user is authenticated."""
        return self.authenticated

    def is_anonymous(self):
        """False,as anonymous users aren't supported."""
        return False

解决方法

经过大量的试验和错误(更多错误)后,我终于从示例(部分理解)中复制了足够的内容以使其正常工作.似乎我需要在MyAdminIndexView()中添加一个super来使它工作.那是

我将“MyView”功能移出init并添加了这一行

admin = Admin(app,'example',index_view=views.MyAdminIndexView())

根据示例显着更改了我的视图

class MyModelView(sqla.ModelView):

    def is_accessible(self):
        return g.user.is_authenticated()

class MyAdminIndexView(admin.AdminIndexView):
    @expose('/')
    def index(self):
        if not g.user.is_authenticated():
            flash('Please log in first...',next_url)
            return redirect(login_url)
        # import pdb;pdb.set_trace()
        if g.user.admin == True:
            return super(MyAdminIndexView,self).index()
        else:
            return redirect(url_for("main"))

def logged_in(f):
    @wraps(f)
    def decorated_function(*args,None)
    flash("Logged Out.")
    return redirect(url_for("login"))

@app.before_request
def before_request():
    g.user = current_user

现在我有了它的工作.非常感谢你的例子!

(编辑:李大同)

【声明】本站内容均来自网络,其相关言论仅代表作者个人观点,不代表本站立场。若无意侵犯到您的权利,请及时与联系站长删除相关内容!

    推荐文章
      热点阅读