2017-01-23 50 views
0

是否可以在Flask中设置多个登录管理器?我有3个网页包含不同内容的需要密码保护:如何在Flask中设置多个登录管理器?

mydomain.com/page1 
mydomain.com/page2 
mydomain.com/page3 

任何登录和访问第1页应该访问第2页第三页或(反之亦然)。

我想一种方法是在我的jinja2模板中指定username,并根据它显示页面结果。例如,我可以添加:

{% if current_user.is_authenticated and current_user.username == 'abc' %} 
<!-- show HTML for user 'abc' only --> 

但是使用这种方法,页面2和页面3仍然可以访问。他们只是不会显示任何内容,因为在jinja2模板中没有为这些用户定义任何内容。

以下是我目前拥有的LoginManager()设置:

lm = LoginManager() 
lm.init_app(app) 
lm.login_view = 'login' 

class User(): 
    def __init__(self, username): 
     self.username = username 

    def is_authenticated(self): 
     return True 

    def is_active(self): 
     return True 

    def is_anonymous(self): 
     return False 

    def get_id(self): 
     return self.username 

    @staticmethod 
    def validate_login(password_hash, password): 
     return check_password_hash(password_hash, password) 

@lm.user_loader 
def load_user(id): 
    user = myUsers.find_one({"User": id}) 
    if not user: 
     return None 
    return User(user['User']) 

@app.route('/login', methods=['GET', 'POST']) 
def login(): 
    form = LoginForm() 
    if current_user.is_authenticated: 
     return redirect(request.args.get("next") or url_for("dashboard")) 
     return redirect(url_for("dashboard")) 
    else: 
     if request.method == 'POST' and form.validate_on_submit(): 
      user = myUsers.find_one({"User": form.username.data}) 
      if user and User.validate_login(user['Password'], form.password.data): 
       user_obj = User(user['User']) 
       login_user(user_obj) 
       return redirect(request.args.get("next") or url_for("dashboard")) 
      flash("Wrong username or password. Please try again.", category='error') 
     return render_template('/login.html', title='Login', form=form) 
+0

添加[Flask-Principle](https://pythonhosted.org/Flask-Principal/)以使用Flask-Login。请参阅文档中的[细粒度资源保护](https://pythonhosted.org/Flask-Principal/#granular-resource-protection)。 – pjcunningham

+0

@pjcunningham谢谢。我会看看。 – mapr

回答

0

这篇文章并没有直接回答你的问题,但我希望它可以提供另一个角度来看,您的问题。我的解决方案是在您的User类中实现自定义view decoratorrole字段以限制用户访问。

+0

谢谢。我已经开始尝试'角色'了。基本上它是另一个被称为User类的一部分。然后在我的jinja2模板中,我显示了基于角色的内容,如'{%if current_user.is_authenticated and current_user.role =='admin'%} <! - 在此处添加内容 - > {%endif%}'。尽管如此,宁愿在服务器端做到这一点。 – mapr

+0

查看装饰是服务器端。在这个例子中,你可以看到如果用户没有登录,视图将不会被渲染。 – dvnguyen