在蓝图应用程序中访问 Flask 全局变量

Question

我的源代码有这样的结构：

main.py:

from flask import Flask, g
app = Flask(__name__)
with app.app_context():
    g.my_db = PostgreSQL()
    app.register_blueprint(my_app, url_prefix="/my_app")

my_app.py:

from flask import Blueprint, g
my_app = Blueprint("my_app", __name__)
@my_app.route("/")
def index():
    return g.my_db.fetch_all()   <<< ERROR

但它显示此错误：

AttributeError: '_AppCtxGlobals' object has no attribute 'my_db'

即使我尝试在应用程序上下文之外使用g，它也会显示此错误：

RuntimeError: Working outside of application context.

那么如何在 Flask 中设置和访问全局变量呢？

Answer 1

g 在您尝试使用它的方式上并不持久。每次需要时编写一个函数来创建一个连接。最好使用像 Flask-SQLAlchemy 这样的数据库扩展来为您管理连接。

db.py:

import <postgresql dependencies>

def get_db():
   db = PostgreSQL()
   # config here
   return db

main.py:

from flask import Flask

app = Flask(__name__)
app.register_blueprint(my_app, url_prefix="/my_app")

my_app.py:

from flask import Blueprint, g
from db import get_db

my_app = Blueprint("my_app", __name__)

@my_app.route("/")
def index():
    db = get_db()
    data = db.fetch_all()
    db.close()
    return data

Answer 2

发生这种情况是因为当上下文 (with app.app_context()) 结束 (doc) 时数据丢失。

在上下文中，一切正常：

from flask import Flask, g
app = Flask(__name__)
with app.app_context():
    g.my_db = 'database ok'
    print(g.my_db)

# >>> this prints 'database ok'

但是在外面，你不能访问该属性：

from flask import Flask, g
app = Flask(__name__)
with app.app_context():
    g.my_db = 'database ok'

print(g.my_db)

# >>> this throws RuntimeError: Working outside of application context

即使你创建了一个新的上下文：

from flask import Flask, g
app = Flask(__name__)
with app.app_context():
    g.my_db = 'database ok'

with app.app_context():
    print(g.my_db)

>>> this throws AttributeError: '_AppCtxGlobals' object has no attribute 'my_db'

最好的调用应该是在上下文之前声明数据库对象，然后导入它。或者你可以直接在my_app.py 里面创建你需要的地方？