2015-04-29 9 views
7

Czy można połączyć instancję _BoundDeclarativeMeta z drugą?Scal wiele deklaratywnych baz w SQLAlchemy

Na przykład:

from sqlalchemy.ext.declarative import declarative_base 

Base1 = declarative_base() 
Base2 = declarative_base() 

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

W rzeczywistości będę instacji modeli ORM z każdego z nich przed włączeniem. Problem polega na tym, że mam niezależny pakiet, który wymaga pewnych tabel do działania. Jednak będzie używany przez inny niezależny pakiet.

Odpowiedz

1

Dla mojego przypadku użycia tego pracował:

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) 

lub, jeszcze lepiej, automatycznie scalić wszystkie deklaratywne zasady:

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) 

To metadane mogą być następnie wykorzystane na przykład do tworzenia diff do bieżąca baza danych:

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)