Warning: file_get_contents(/data/phpspider/zhask/data//catemap/4/postgresql/10.json): failed to open stream: No such file or directory in /data/phpspider/zhask/libs/function.php on line 167

Warning: Invalid argument supplied for foreach() in /data/phpspider/zhask/libs/tag.function.php on line 1116

Notice: Undefined index: in /data/phpspider/zhask/libs/function.php on line 180

Warning: array_chunk() expects parameter 1 to be array, null given in /data/phpspider/zhask/libs/function.php on line 181
Postgresql 使用具有相同Alchemy模型的多个POSTGRES数据库和模式_Postgresql_Flask_Sqlalchemy_Flask Sqlalchemy_Sharding - Fatal编程技术网

Postgresql 使用具有相同Alchemy模型的多个POSTGRES数据库和模式

Postgresql 使用具有相同Alchemy模型的多个POSTGRES数据库和模式,postgresql,flask,sqlalchemy,flask-sqlalchemy,sharding,Postgresql,Flask,Sqlalchemy,Flask Sqlalchemy,Sharding,这里我将非常具体,因为有人问过类似的问题,但没有一个解决方案能解决这个问题 我正在做一个有四个postgres数据库的项目,但是为了简单起见,假设有两个。即A&B酒店 A、 B表示两个地理位置,但数据库中的表和模式是相同的 示例模型: from flask_sqlalchemy import SQLAlchemy from sqlalchemy import * from sqlalchemy.ext.declarative import declarative_base db = SQLAl

这里我将非常具体,因为有人问过类似的问题,但没有一个解决方案能解决这个问题

我正在做一个有四个postgres数据库的项目,但是为了简单起见,假设有两个。即A&B酒店

A、 B表示两个地理位置,但数据库中的表和模式是相同的

示例模型:

from flask_sqlalchemy import SQLAlchemy
from sqlalchemy import *
from sqlalchemy.ext.declarative import declarative_base

db = SQLAlchemy()
Base = declarative_base()

class FRARecord(Base):
    __tablename__ = 'tb_fra_credentials'

    recnr = Column(db.Integer, primary_key = True)
    fra_code = Column(db.Integer)
    fra_first_name = Column(db.String)
此模型在两个数据库中都进行了复制,但模式不同,因此要使其在中工作,我需要执行以下操作:

__table_args__ = {'schema' : 'A_schema'}
我想使用一个单一的内容提供者,该提供者被赋予访问数据库的权限,但具有相同的方法:

class ContentProvider():
    def __init__(self, database):
        self.database = database

    def get_fra_list():
        logging.debug("Fetching fra list")
        fra_list = db.session.query(FRARecord.fra_code)
两个问题是,我如何决定指向哪个db,以及我如何不为不同的模式复制模型代码(这是postgres特有的问题)

以下是我迄今为止所尝试的:

1) 我为每个模型制作了单独的文件并继承了它们,因此:

class FRARecordA(FRARecord):
    __table_args__ = {'schema' : 'A_schema'}
这似乎不起作用,因为我得到了错误:

"Can't place __table_args__ on an inherited class with no table."
这意味着在db.Model(在其父级中)已经声明之后,我无法设置该参数

2) 所以我试着用多重继承做同样的事情

class FRARecord():
    recnr = Column(db.Integer, primary_key = True)
    fra_code = Column(db.Integer)
    fra_first_name = Column(db.String)

class FRARecordA(Base, FRARecord):
    __tablename__ = 'tb_fra_credentials'
    __table_args__ = {'schema' : 'A_schema'}
但得到了可预测的错误:

"CompileError: Cannot compile Column object until its 'name' is assigned."
显然,我无法将列对象移动到FRARecordA模型,而不必为B重复它们(实际上有4个数据库和更多的模型)

3) 最后,我正在考虑做一些切分(这似乎是正确的方法),但我找不到一个例子来说明我该如何做。我的感觉是,我只会使用这样一个对象:

class FRARecord(Base):
    __tablename__ = 'tb_fra_credentials'

    @declared_attr
    def __table_args__(cls):
        #something where I go through the values in bind keys like
        for key, value in self.db.app.config['SQLALCHEMY_BINDS'].iteritems():
            # Return based on current session maybe? And then have different sessions in the content provider?

    recnr = Column(db.Integer, primary_key = True)
    fra_code = Column(db.Integer)
    fra_first_name = Column(db.String)
为了明确起见,我访问不同数据库的意图如下:

app.config['SQLALCHEMY_DATABASE_URI']='postgresql://%(user)s:\
%(pw)s@%(host)s:%(port)s/%(db)s' % POSTGRES_A

app.config['SQLALCHEMY_BINDS']={'B':'postgresql://%(user)s:%(pw)s@%(host)s:%(port)s/%(db)s' % POSTGRES_B,
                                  'C':'postgresql://%(user)s:%(pw)s@%(host)s:%(port)s/%(db)s' % POSTGRES_C,
                                  'D':'postgresql://%(user)s:%(pw)s@%(host)s:%(port)s/%(db)s' % POSTGRES_D
                                 }
其中POSTGRES字典包含连接到数据的所有键

我假设对于继承的对象,我只需要像这样连接到正确的对象(这样sqlalchemy查询就会自动知道):


终于找到了解决办法

本质上,我没有为每个数据库创建新的类,只是为每个数据库使用了不同的数据库连接

这种方法本身非常常见,棘手的部分(我找不到例子)是处理模式差异。我最终做了这样的事:

from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker

Session = sessionmaker()

class ContentProvider():

    db = None
    connection = None
    session = None

    def __init__(self, center):
        if center == A:
            self.db = create_engine('postgresql://%(user)s:%(pw)s@%(host)s:%(port)s/%(db)s' % POSTGRES_A, echo=echo, pool_threadlocal=True)
            self.connection = self.db.connect()
            # It's not very clean, but this was the extra step. You could also set specific connection params if you have multiple schemas
            self.connection.execute('set search_path=A_schema')
        elif center == B:
            self.db = create_engine('postgresql://%(user)s:%(pw)s@%(host)s:%(port)s/%(db)s' % POSTGRES_B, echo=echo, pool_threadlocal=True)
            self.connection = self.db.connect()
            self.connection.execute('set search_path=B_schema')

    def get_fra_list(self):
        logging.debug("Fetching fra list")
        fra_list = self.session.query(FRARecord.fra_code)
        return fra_list
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker

Session = sessionmaker()

class ContentProvider():

    db = None
    connection = None
    session = None

    def __init__(self, center):
        if center == A:
            self.db = create_engine('postgresql://%(user)s:%(pw)s@%(host)s:%(port)s/%(db)s' % POSTGRES_A, echo=echo, pool_threadlocal=True)
            self.connection = self.db.connect()
            # It's not very clean, but this was the extra step. You could also set specific connection params if you have multiple schemas
            self.connection.execute('set search_path=A_schema')
        elif center == B:
            self.db = create_engine('postgresql://%(user)s:%(pw)s@%(host)s:%(port)s/%(db)s' % POSTGRES_B, echo=echo, pool_threadlocal=True)
            self.connection = self.db.connect()
            self.connection.execute('set search_path=B_schema')

    def get_fra_list(self):
        logging.debug("Fetching fra list")
        fra_list = self.session.query(FRARecord.fra_code)
        return fra_list