2013-02-10 56 views
2

在编写我自己的登录/注册系统后,我发现约Flask-Security,并决定将其集成到我的应用程序。我所有的身份验证和恢复代码都没有了,但我无法弄清楚为什么我在启动应用程序时遇到了简单的错误。烧瓶安全无法正确初始化

这里是最精简的例子,我能想出:

from flask import Flask, render_template_string 

from flask.ext.sqlalchemy import SQLAlchemy 
from flask.ext.security import Security, SQLAlchemyUserDatastore, UserMixin, RoleMixin, current_user 

app = Flask('test') 
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:////tmp/database.db' 

db = SQLAlchemy(app) 

roles_users = db.Table('roles_users', 
    db.Column('user_id', db.Integer, db.ForeignKey('user.id')), 
    db.Column('role_id', db.Integer, db.ForeignKey('role.id')) 
) 

class Role(db.Model, RoleMixin): 
    id = db.Column(db.Integer(), primary_key=True) 
    name = db.Column(db.String(80), unique=True) 
    description = db.Column(db.String(255)) 

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

    username = db.Column(db.String(50), unique=True) 
    email = db.Column(db.String(255), unique=True) 
    password = db.Column(db.String(255)) 

    active = db.Column(db.Boolean) 
    confirmed_at = db.Column(db.DateTime) 
    roles = db.relationship('Role', secondary=roles_users, backref=db.backref('users', lazy='dynamic')) 

    def is_authenticated(self): 
     return True 

    def is_active(self): 
     return True 

    def is_anonymous(self): 
     return False 

    def get_id(self): 
     return unicode(self.id) 

    def __repr__(self): 
     return '<User {self.username}>'.format(self=self) 

user_datastore = SQLAlchemyUserDatastore(db, User, Role) 
security = Security(app) 

@app.before_first_request 
def create_db(): 
    db.create_all() 

@app.route('/') 
def index(): 
    return render_template_string(''' 
     {% if current_user.is_authenticated() %} 
      You're logged in. 
     {% else %} 
      You're not logged in. 
     {% endif %} 
    ''') 

if __name__ == '__main__': 
    app.run(debug=True) 

在访问http://localhost:5000/,我得到这个错误:

Traceback (most recent call last): 
    File "/usr/lib/python2.7/site-packages/flask/app.py", line 1701, in __call__ 
    return self.wsgi_app(environ, start_response) 
    File "/usr/lib/python2.7/site-packages/flask/app.py", line 1689, in wsgi_app 
    response = self.make_response(self.handle_exception(e)) 
    File "/usr/lib/python2.7/site-packages/flask/app.py", line 1687, in wsgi_app 
    response = self.full_dispatch_request() 
    File "/usr/lib/python2.7/site-packages/flask/app.py", line 1360, in full_dispatch_request 
    rv = self.handle_user_exception(e) 
    File "/usr/lib/python2.7/site-packages/flask/app.py", line 1358, in full_dispatch_request 
    rv = self.dispatch_request() 
    File "/usr/lib/python2.7/site-packages/flask/app.py", line 1344, in dispatch_request 
    return self.view_functions[rule.endpoint](**req.view_args) 
    File "/tmp/app.py", line 62, in index 
    ''') 
    File "/usr/lib/python2.7/site-packages/flask/templating.py", line 140, in render_template_string 
    context, ctx.app) 
    File "/usr/lib/python2.7/site-packages/flask/templating.py", line 107, in _render 
    rv = template.render(context) 
    File "/usr/lib/python2.7/site-packages/jinja2/environment.py", line 894, in render 
    return self.environment.handle_exception(exc_info, True) 
    File "<template>", line 2, in top-level template code 

    File "/usr/lib/python2.7/site-packages/jinja2/environment.py", line 372, in getattr 
    return getattr(obj, attribute) 
UndefinedError: 'current_user' is undefined 

我看着都Flask-Security-Example的来源和Flask-Social-Example,但我无法弄清楚我要去哪里错了。

非常感谢所有帮助。


作为一个小问题,为什么这个表是这样创建的(更重要的是)它是为了什么?

roles_users = db.Table('roles_users', 
    db.Column('user_id', db.Integer, db.ForeignKey('user.id')), 
    db.Column('role_id', db.Integer, db.ForeignKey('role.id')) 
) 
+0

的'roles_user'表“角色”和“用户”之间的映射表 - 只是每个主键。据我所知,你只创建'Table'而不是使用这种表格的完整模型,因为你永远不会使用该表格作为模型本身 - 它只是用来映射你的模型之间的关系* 关心。因此,您只需创建一个表即可节省创建完整模型所涉及的任何开销。 – 2013-02-12 01:33:00

+0

@SeanVieira:谢谢Sean,现在有道理。 – Blender 2013-02-19 23:12:58

回答

2

看起来像我发现我的问题(捂脸)

security = Security(app) 

我应该在数据存储路过作为第二个参数:

security = Security(app, user_datastore)