Flask SQLAlchemy设置动态URI

时间:2013-01-24 00:04:35

标签: python sqlalchemy flask uri

我在WSGI下提供了一个Flask应用程序,数据库URI随时间变化。 URI每两个小时切换到另一个数据库。我正在利用这段时间来填充一个数据库,而另一个正在为应用程序提供数据。

我很难弄清楚如何最好地配置会话,以便在切换发生时,客户端将在下次请求时获得正确(不同)的数据库。从我的测试来看,如果我在顶层初始化数据库,当切换发生时,客户端仍然指向旧数据库。

我考虑过在页面内设置会话(索引等),但是多么痛苦,然后我担心打开和关闭太多的数据库连接并让它们四处闲逛。我想我可以通过在启动时初始化两个会话,然后只选择在每个页面内切换时使用哪个会话来使其工作。这似乎效率低下,我确信有更好的方法。

帮助?!

~~~~~~~~~

这是我目前正在做的一般概念,但没有办法改变请求之间的URI。顶级代码只运行一次,或者每隔一段时间运行一次。

if now.hour % 2:
    db_name = 'db1'
else:
    db_name = 'db2'

app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = "mysql://foo:poo@localhost:3306/%s" % db_name

def init_db(uri, **kwargs):
    engine = create_engine(uri, **kwargs)

    Base.metadata.create_all(bind=engine)

    global db_session
    db_session = scoped_session(sessionmaker(autocommit=False, autoflush=False, bind=engine))
    Base.query = db_session.query_property()


init_db(app.config['SQLALCHEMY_DATABASE_URI'], pool_recycle=3600)

@app.teardown_request
def shutdown_session(exception=None):
    db_session.remove()

@app.route('/')
def index():
    ...etc...

工作示例 - 美丽。

app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = "mysql://foo:poo@localhost:3306/%s"

class SessionManager(object):
    def __init__(self, base_uri=None, **kwargs):
        self.session = None
        self.base_uri = base_uri
        self.kwargs = kwargs

    def get_session(self):
        if now.hour % 2:
            db_name = 'db1'
        else:
            db_name = 'db2'

        if self.session:
            if self.session.name == db_name:
                return self.session
            else:
                self.session.remove()
                self.session = None

        if not self.session:
            engine = create_engine(self.base_uri % db_name, **self.kwargs)
            db_session = scoped_session(sessionmaker(bind=engine))
            db_session.name = db_name
            self.session = db_session
        return db_session

session_manager = SessionManager(base_uri=app.config['SQLALCHEMY_DATABASE_URI'], pool_recycle=3600)

db_session = LocalProxy(session_manager.get_session)

1 个答案:

答案 0 :(得分:4)

您可以为会话创建自定义构建器,以便在规则指示时重新创建引擎和作用域会话。像

这样的东西
class SessionManager(object):

    def __init__(self):
        self.session = None

    def get_session(self):
        # return existing session or make a new engine and scoped_session

要使此类透明,请使用Werkzeug's LocalProxy。使用会话的代码根本不需要改变。

session_manager = SessionManager()
db_session = LocalProxy(session_manager.get_session)