Flask - 在应用程序上下文之外工作

时间:2016-03-25 02:59:54

标签: python flask flask-sqlalchemy flask-restful

我正在测试一个Flask应用程序,并且正在接收一个&34;在应用程序上下文之外工作"错误。我的文件目录如下:

api
    app.py
    __init__.py
    models
        __init__.py
        user.py
    resources
        __init__.py
        deals.py
        stores.py
    common
        __init__.py
        calculations.py
        decorators.py

我的app.py文件如下所示:

import os
from flask import Flask, jsonify, url_for, redirect, request, g, current_app
from flask_pymongo import PyMongo
from flask_restful import Api, Resource
from flask_httpauth import HTTPTokenAuth
from flask.ext.sqlalchemy import SQLAlchemy
from flask.ext.httpauth import HTTPBasicAuth

from resources.deals import Deals
from resources.stores import Stores

from models.user import User

USERDBFILE=os.path.join(os.path.join(os.path.dirname(os.path.realpath(__file__)),'database'),'db.sqlite')

#Deals database
app = Flask(__name__)
app.config["MONGO_DBNAME"] = "database"
mongo = PyMongo(app,config_prefix='MONGO')
app.db = mongo

#User database
app.config['SECRET_KEY'] = 'SECRET KEY'
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///db.sqlite'
app.config['SQLALCHEMY_COMMIT_ON_TEARDOWN'] = True
app.dbuser = SQLAlchemy(app)

#App url
app.APP_URL = "http://127.0.0.1:5000"

#Setup authorization
auth = HTTPTokenAuth(scheme='Token')

#Setup the app
api = Api(app)
api.add_resource(Deals, '/deals', '/Deals/<string:type>/<string:id>',endpoint="dealType")
api.add_resource(Stores, '/stores', '/Stores/<string:type>/<string:id>',endpoint="type")

if __name__ == "__main__":
    if not os.path.exists(USERDBFILE):
        app.dbuser.create_all()
    app.run(debug=True)

我的users.py文件如下:

from flask import current_app
import os
from flask import Flask, abort, request, jsonify, g, url_for
from flask.ext.sqlalchemy import SQLAlchemy
from flask.ext.httpauth import HTTPBasicAuth
from passlib.apps import custom_app_context as pwd_context
from itsdangerous import (TimedJSONWebSignatureSerializer
                          as Serializer, BadSignature, SignatureExpired)

class User(current_app.dbuser.Model):
    __tablename__ = 'user_api'
    id = current_app.dbuser.Column(current_app.dbuser.Integer,primary_key=True)
    date_created = current_app.dbuser.Column(current_app.dbuser.DateTime,default=current_app.dbuser.func.current_timestamp())
    date_modified = current_app.dbuser.Column(current_app.dbuser.DateTime,default=current_app.dbuser.func.current_timestamp(),
                                      onupdate=current_app.dbuser.func.current_timestamp())
    # User Name
    name = current_app.dbuser.Column(current_app.dbuser.String(128),nullable=False)
    # Identification Data: email & password
    email = current_app.dbuser.Column(current_app.dbuser.String(128),nullable=False,unique=True)
    password = current_app.dbuser.Column(current_app.dbuser.String(192),nullable=False)
    company = current_app.dbuser.Column(current_app.dbuser.String(128),nullable=False,unique=True)
    # Authorization Data: role & status
    role = current_app.dbuser.Column(current_app.dbuser.String(32),nullable=False,default='user')
    status = current_app.dbuser.Column(current_app.dbuser.Boolean,nullable=False,default=True)
    hourly_limit = current_app.dbuser.Column(current_app.dbuser.Integer,nullable=False,default=100)
    daily_limit = current_app.dbuser.Column(current_app.dbuser.Integer,nullable=False,default=2400)
    monthly_limit = current_app.dbuser.Column(current_app.dbuser.Integer,nullable=False,default=2400)
    admin = current_app.dbuser.Column(current_app.dbuser.Boolean,nullable=False,default=True)

    def hash_password(self, password):
        self.password_hash = pwd_context.encrypt(password)

    def verify_password(self, password):
        return pwd_context.verify(password, self.password_hash)

    def generate_auth_token(self, expiration=600):
        s = Serializer(current_app.config['SECRET_KEY'], expires_in=expiration)
        return s.dumps({'id': self.id})

    @staticmethod
    def verify_auth_token(token):
        s = Serializer(current_app.config['SECRET_KEY'])
        try:
            data = s.loads(token)
        except SignatureExpired:
            return None    # valid token, but expired
        except BadSignature:
            return None    # invalid token
        user = User.query.get(data['id'])
        return user

我使用

在与app.py相同的目录中运行该文件
python app.py

但它返回以下错误:

File "app.py", line 13, in <module>
    from models.user import User
  File "/Users/toby/api/api/models/user.py", line 10, in <module>
    class User(current_app.dbuser.Model):
  File "/Users/toby/api/venv/lib/python3.4/site-packages/werkzeug/local.py", line 343, in __getattr__
    return getattr(self._get_current_object(), name)
  File "/Users/toby/api/venv/lib/python3.4/site-packages/werkzeug/local.py", line 302, in _get_current_object
    return self.__local()
  File "/Users/toby/api/venv/lib/python3.4/site-packages/flask/globals.py", line 34, in _find_app
    raise RuntimeError('working outside of application context')
RuntimeError: working outside of application context

如果我将user.py文件的内容移动到app.py文件中并将继承从current_app.dbuser.Model更改为app.dbuser.Model,它似乎工作正常。有谁知道我做错了什么?

2 个答案:

答案 0 :(得分:0)

Flask-Sqlalchemy将一些sqlalchemy概念(例如会话,引擎和声明基础)绑定到烧瓶应用程序。这很方便,因为你只需要在你的uwsgi入口点(app对象)实例化一件事,但是在测试时会很痛苦 - 因为你必须实例化app对象。

编辑 - 我将离开关于下面测试的部分,但我重读了你的问题并意识到你实际上并没有尝试测试任何东西。

您在导入时无法访问'current_app'对象(当您尝试初始化sqlalchemy模型时)。相反,您必须从应用程序文件中实际导入应用程序对象。这当然意味着您必须担心循环依赖...

我有一个名为'register_routes'的方法,在我初始化导入模型并查看需要在导入时访问app对象的文件的app对象后调用。

#at the bottom of app.py
def register_models(app):
    from models import User

register_models(app)
# in models.py
from app import app

class User(app.dbuser.Model):
   ...

编辑 - 以下讨论了有关单元测试的这个问题

Flask-Testing是一个试图解决这些问题的项目,几乎可以肯定适合这个领域的初学者 - 它提供了一个继承的测试类,它将在测试用例之前设置你的烧瓶应用程序并将其拆除后。 (当你了解各种全局变量以及他们做了什么时,你可能想要摆脱这个......但是这对开始非常有用!)

如果您不想这样做,则需要先创建应用并初始化应用上下文,然后再对flask-sqlalchemy模型执行任何操作。这可能只是

app = myapp.create()
with app.test_request_context():
    # do some testing...

您可能希望在方法之间刷新它,否则全局状态将在测试用例之间泄漏。

答案 1 :(得分:-1)

基本上,flask使用了很多全局变量,如current_apprequest等,这些变量仅在烧瓶应用程序实例化并运行并处于各种状态时才存在。

您在User对象的定义中使用了current_app,一旦Python导入文件,就会对其进行评估。您需要确保在应用程序运行时仅使用此类值。

您可以移动User类的实例化,直到应用程序存在,但我认为根本问题是您使用current_app.dbuser.Boolean而不是说sqlalchemy.types.Boolean的原因?

我不是flask.ext.sqlalchemy的优秀专家,但我猜你不需要从特定的{1}}和Column加载那些定义。你有的应用程序。使用Boolean中的静态定义会阻止您从sqlalchemy类到应用程序的依赖。