我正在使用没有 flask extension 的 py2neo (因为它在我的 flask 项目中表现得相当 strangely )。
我想要做的是在应用程序上下文中保留我的数据库连接(因为它在整个请求处理过程中被保留,对吧?)。
根据flask的文档,我有以下代码:
def get_neo4j():
with app.app_context():
neo4j = getattr(g, '_neo4j', None)
if neo4j is None:
connection_string = http blah blah blah
neo4j = g._neo4j = Graph(connection_string)
return neo4j
这是正确的吗?我如何编辑上述代码以从 werkzeug 的本地代理中受益?
最佳答案
好的, 在阅读 Flask 关于 extensions 的文档之后, 开发我自己的类来保持连接很简单:
from py2neo import Graph
from flask import current_app
try:
from flask import _app_ctx_stack as stack
except ImportError:
from flask import _request_ctx_stack as stack
class Neo4jConnection(object):
def __init__(self, app=None):
self.app = app
if app is not None:
self.init_app(app)
def init_app(self, app):
if hasattr(app, 'teardown_appcontext'):
app.teardown_appcontext(self.teardown)
else:
app.teardown_request(self.teardown)
def connect(self):
# Read configuations from app config
return Graph(bolt=True,
host=current_app.config['NEO4J_SERVER_ADDRESS'],
bolt_port=current_app.config['NEO4J_PORT'],
user=current_app.config['NEO4J_USER'],
password=current_app.config['NEO4J_PASSWORD'],)
def teardown(self, exception):
ctx = stack.top
if hasattr(ctx, 'neo4j_db'):
ctx.neo4j_db = None
@property
def connection(self):
ctx = stack.top
if ctx is not None:
if not hasattr(ctx, 'neo4j_db'):
ctx.neo4j_db = self.connect()
return ctx.neo4j_db
现在使用工厂模式让它工作:
初始化:
neo4j = Neo4jConnection()
neo4j.init_app(app)
用法:
myneo = Neo4jConnection().connection
myneo.push(.....)
我已经测试过了。希望它是正确的。
关于python - 在 Flask 的应用上下文中保持 py2neo 连接,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/44222256/