PostgreSQL
 sql >> Base de données >  >> RDS >> PostgreSQL

Utilisation de plusieurs bases de données et schémas POSTGRES avec le même modèle Flask-SQLAlchemy

J'ai enfin trouvé une solution à ce problème.

Essentiellement, je n'ai pas créé de nouvelles classes pour chaque base de données, j'ai juste utilisé différentes connexions de base de données pour chacune.

Cette méthode en elle-même est assez courante, la partie délicate (dont je n'ai pas pu trouver d'exemples) était la gestion des différences de schéma. J'ai fini par faire ceci :

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)[email protected]%(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)[email protected]%(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