2015-04-29 15 views
7

È possibile unire un'istanza _BoundDeclarativeMeta con un'altra?Unione di più basi dichiarative in SQLAlchemy

Ad esempio:

from sqlalchemy.ext.declarative import declarative_base 

Base1 = declarative_base() 
Base2 = declarative_base() 

# Something like this? 
CombinedBase = Base1.merge(Base2) 

In realtà sarei sottoclasse modelli ORM da ciascuno di essi prima di fondersi. Il problema è che ho un pacchetto indipendente che ha bisogno di certe tabelle per funzionare. Eppure sarà utilizzato da un altro pacchetto indipendente.

risposta

1

Per il mio caso d'uso questo ha funzionato:

from sqlalchemy import MetaData 

combined_meta_data = MetaData() 

for declarative_base in [Base1, Base2]: 
    for (table_name, table) in declarative_base.metadata.tables.items(): 
     combined_meta_data._add_table(table_name, table.schema, table) 

O, meglio ancora, unire automaticamente tutte le basi dichiarativi:

import gc 
from sqlalchemy import MetaData 

combined_meta_data = MetaData() 

for declarative_base in ([obj for obj in gc.get_objects() if isinstance(obj, DeclarativeMeta)]): 
    for (table_name, table) in declarative_base.metadata.tables.items(): 
     combined_meta_data._add_table(table_name, table.schema, table) 

Quella meta dati possono poi essere utilizzati ad esempio per creare un diff per il database corrente:

from sqlalchemy import create_engine 
from alembic.migration import MigrationContext 
from alembic.autogenerate import compare_metadata 
import pprint 

engine = create_engine(...) 
migration_context = MigrationContext.configure(engine.connect()) 

diff = compare_metadata(migration_context, combined_meta_data) 
pprint.pprint(diff)