Flask-Security：如何在主.py文件之外设置数据库？

Question

我正在使用 Flask-Security 1.7.4 和 Flask 0.10.1。

如果我的数据库结构保存在 Python 文件 app.py 中，我的网站运行得很好：

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

app = Flask(__name__)

#Create database connection object
db = SQLAlchemy(app)

#Define models
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)
    email = db.Column(db.String(255), unique=True)
    companyid = db.Column(db.String(255), unique=True)
    lastname = db.Column(db.String(255))
    firstname = db.Column(db.String(255))
    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'))

# Setup Flask-Security
user_datastore = SQLAlchemyUserDatastore(db, User, Role)
security = Security(app, user_datastore)

不过，我想把代码分开，这样我可以有：

app_folder
|__ app.py  # Main file for the app
|__ security.py  # File that contain the database for the login/roles

所以我尝试了：

在 security.py 中：

db = SQLAlchemy()  

# Define models
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)
    email = db.Column(db.String(255), unique=True)
    companyid = db.Column(db.String(255), unique=True)
    lastname = db.Column(db.String(255))
    firstname = db.Column(db.String(255))
    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'))

user_datastore = SQLAlchemyUserDatastore(db, User, Role)

而在 app.py 中：

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

app = Flask(__name__)

#Initiate the database connection object
db.init_app(app)

# Setup Flask-Security
security = Security(app, user_datastore)

但是我遇到了以下错误：

File "C:\Work\pythonVirtualEnv\env02\lib\site-packages\sqlalchemy\engine\default.py", line 436, in do_execute
    cursor.execute(statement, parameters)
OperationalError: (OperationalError) no such table: users u'SELECT users.user_id AS users_user_id, users.username AS users_username, users.password AS users_password, users.email AS users_email, users.registered_on AS users_registered_on \nFROM users \nWHERE users.password = ? AND users.email = ?\n LIMIT ? OFFSET ?' (u'test', u'demo@demo.com', 1, 0)

问题是，如果我把所有代码都放在 app.py 中，它就能完美运行。 当然，数据库是存在的。

你能告诉我我哪里做错了吗？

谢谢 :)

Answer 1

你应该使用一种不同的结构（而且是推荐的结构）：

proj_folder
- run.py
- config.py
|__ app  # Main folder for the app
|__ __init__.py
|__ models.py
|__ security.py
|__ views.py

在模型文件里，你放置应用程序的模型；

在安全文件里，你放置安全相关的模型；

在视图文件里，你创建视图并暴露你的网址。

然后在 init.py 文件中：

import logging
from flask import Flask
from flask.ext.appbuilder import SQLA
from flask.ext.appbuilder import AppBuilder
from sqlalchemy import SQLAlchemy

logging.basicConfig(format='%(asctime)s:%(levelname)s:%(name)s:%(message)s')
logging.getLogger().setLevel(logging.DEBUG)

app = Flask(__name__)
app.config.from_object('config')
db = SQLAlchemy(app)

from app import models, security, views

在 run.py 文件中：

from app import app

app.run(host='0.0.0.0', port=8080, debug=True)

希望这些对你有帮助。

可以看看这个链接：https://github.com/dpgaspar/Flask-AppBuilder-Skeleton