2010-09-03 30 views
6

我的應用程序使用類似於this的技術連接到多個數據庫。只要我不嘗試在同一個請求中訪問不同的數據庫,它就會工作。說完回頭看了看上面的腳本我看他們寫爲此評論:如何在Cherrypy和SQLAlchemy的相同請求中使用多個數據庫?

SQLAlchemy integration for CherryPy, 
such that you can access multiple databases, 
but only one of these databases per request or thread. 

我的應用程序現在要求我從數據庫A和B.數據庫獲取數據是否有可能做到這一點在單個請求?

請參閱下面的來源和例子:

工作實施例1:

from model import meta 

my_object_instance = meta.main_session().query(MyObject).filter(
    MyObject.id == 1 
).one() 

工作實施例2:

from model import meta 

my_user = meta.user_session().query(User).filter(
    User.id == 1 
).one() 

錯誤示例:

from model import meta 

my_object_instance = meta.main_session().query(MyObject).filter(
    MyObject.id == 1 
).one() 

my_user = meta.user_session().query(User).filter(
    User.id == 1 
).one() 

這個錯誤有:

(sqlalchemy.exc.ProgrammingError) (1146, "Table 'main_db.user' doesn't exist") 

來源:

# meta.py 
import cherrypy 
import sqlalchemy 
from sqlalchemy import MetaData 
from sqlalchemy.orm import scoped_session, sessionmaker 
from sqlalchemy.ext.declarative import declarative_base 

# Return an Engine 
def create_engine(defaultschema = True, schema = "", **kwargs): 

    # A blank DB is the same as no DB so to specify a non-schema-specific connection just override with defaultschema = False 
    connectionString = 'mysql://%s:%[email protected]%s/%s?charset=utf8' % (
     store['application'].config['main']['database-server-config-username'], 
     store['application'].config['main']['database-server-config-password'], 
     store['application'].config['main']['database-server-config-host'], 
     store['application'].config['main']['database-server-config-defaultschema'] if defaultschema else schema 
    ) 
    # Create engine object. we pass **kwargs through so this call can be extended 
    return sqlalchemy.create_engine(connectionString, echo=True, pool_recycle=10, echo_pool=True, encoding='utf-8', **kwargs) 

# Engines 
main_engine = create_engine() 
user_engine = None 

# Sessions 
_main_session = None 
_user_session = None 

# Metadata 
main_metadata = MetaData() 
main_metadata.bind = main_engine 
user_metadata = MetaData() 

# No idea what bases are/do but nothing works without them 
main_base = declarative_base(metadata = main_metadata) 
user_base = declarative_base(metadata = user_metadata) 

# An easy collection of user database connections 
engines = {} 

# Each thread gets a session based on this object 
GlobalSession = scoped_session(sessionmaker(autoflush=True, autocommit=False, expire_on_commit=False)) 

def main_session(): 
    _main_session = cherrypy.request.main_dbsession 
    _main_session.configure(bind=main_engine) 

    return _main_session 

def user_session(): 
    _user_session = cherrypy.request.user_dbsession 
    _user_session.configure(bind = get_user_engine()) 

    return _user_session 

def get_user_engine(): 

    # Get dburi from the users instance 
    dburi = cherrypy.session['auth']['user'].instance.database 

    # Store this engine for future use 
    if dburi in engines: 
     engine = engines.get(dburi) 
    else: 
     engine = engines[dburi] = create_engine(defaultschema = False, schema = dburi) 

    # Return Engine 
    return engine 


def get_user_metadata(): 
    user_metadata.bind = get_user_engine() 
    return user_metadata 

# open a new session for the life of the request 
def open_dbsession(): 
    cherrypy.request.user_dbsession = cherrypy.thread_data.scoped_session_class 
    cherrypy.request.main_dbsession = cherrypy.thread_data.scoped_session_class 
    return 

# close the session for this request 
def close_dbsession(): 
    if hasattr(cherrypy.request, "user_dbsession"): 
     try: 
      cherrypy.request.user_dbsession.flush() 
      cherrypy.request.user_dbsession.remove() 
      del cherrypy.request.user_dbsession 
     except: 
      pass 
    if hasattr(cherrypy.request, "main_dbsession"): 
     try: 
      cherrypy.request.main_dbsession.flush() 
      cherrypy.request.main_dbsession.remove() 
      del cherrypy.request.main_dbsession 
     except: 
      pass 

    return 

# initialize the session factory class for the selected thread 
def connect(thread_index): 
    cherrypy.thread_data.scoped_session_class = scoped_session(sessionmaker(autoflush=True, autocommit=False)) 
    return 

# add the hooks to cherrypy 
cherrypy.tools.dbsession_open = cherrypy.Tool('on_start_resource', open_dbsession) 
cherrypy.tools.dbsession_close = cherrypy.Tool('on_end_resource', close_dbsession) 
cherrypy.engine.subscribe('start_thread', connect) 
+0

我不想'回答'我自己的問題,但解決方案似乎是爲'connect'中的其他數據庫添加一個額外的scoped_session,因此它看起來像'def connect(thread_index):cherrypy.thread_data.user_scoped_session_class = scoped_session(sessionmaker(autoflush = True,autocommit = False))cherrypy.thread_data.main_scoped_session_class = scoped_session(sessionmaker(autoflush = True,autocommit = False))return'然後在'open_dbsession'中單獨引用它們。 – OrganicPanda 2010-09-03 13:20:00

回答

1

你也可以選擇從地上爬起來的多個數據庫,如Dejavu設計了一個ORM。

0

看看這個:

http://pythonhosted.org/Flask-SQLAlchemy/binds.html

基本上,它建議您使用綁定參數:param - 爲每個連接。這就是說,這似乎有點破解。

這個問題有答案了很多更詳細:

With sqlalchemy how to dynamically bind to database engine on a per-request basis

這就是說,無論這個問題和一個引用是不是最新的,SQLAlchemy的可能會從那以後移動了。

+0

這看起來不錯。我最終解決了我正面臨的這個問題的直接問題(請參閱我對該問題的評論),但最終出現了幾個進一步的問題;它現在都在運行,但代碼和解釋對於StackOverflow來說太大了。多個線程+每個用戶多個動態數據庫連接+每個請求設置/拆卸=痛苦 – OrganicPanda 2013-04-24 16:25:54

相關問題