alembic create_table с использованием производных объектов declarative_base

У меня есть объект ORM Alchemy:

from sqlalchemy.ext.declarative import declarative_base
Base = declarative_base()


class MyORM(Base):
    id = Column(Integer, primary_key=True)
    name = Column(String(128), unique=True, nullable=False)

При использовании перегонного куба для создания таблицы я делаю следующее:

def upgrade():
    op.create_table(
        'myorm',
        sa.Column('id', sa.Integer, primary_key=True),
        sa.Column('name', sa.String(128), nullable=False),
    )

Вопрос. Можно ли использовать класс MyORM для создания таблицы? Что-то вроде этого:

def upgrade():
    op.create_table(
        'myorm',
        sa.BaseObject(MyORM)
    )

python sqlalchemy alembic
person ezdazuzena    schedule 25.06.2014    source источник

Ответы (1)


arrow_upward
7
arrow_downward

Это именно то, чего стараются избежать миграции Alembic. Если вы привяжете свою миграцию к текущему состоянию вашей модели, это не будет последовательным путем обновления.

Вы можете использовать декларативный подход в своих миграциях для создания таблиц и переноса данных, но не для их изменения. Вам придется заново создать определения отдельно от определений приложений. Это может быть полезно, если вы хотите выполнить миграцию данных и лучше знакомы с запросами ORM, а не с основными запросами.

Вот пример миграции, которая создает модели Foo и Bar с отношением «многие ко многим» с использованием декларативного подхода, создает таблицы и вставляет некоторые данные.

"""declarative

Revision ID: 169ad57156f0
Revises: 29b4c2bfce6d
Create Date: 2014-06-25 09:00:06.784170
"""

revision = '169ad57156f0'
down_revision = '29b4c2bfce6d'

from alembic import op
import sqlalchemy as sa
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker, relationship

Session = sessionmaker()
Base = declarative_base()


class Foo(Base):
    __tablename__ = 'foo'

    id = sa.Column(sa.Integer, primary_key=True)
    name = sa.Column(sa.String, nullable=False, unique=True)


class Bar(Base):
    __tablename__ = 'bar'

    id = sa.Column(sa.Integer, primary_key=True)
    name = sa.Column(sa.String, nullable=False, unique=True)
    foos = relationship(Foo, lambda: foo_bar, backref='bars')


foo_bar = sa.Table(
    'foo_bar', Base.metadata,
    sa.Column('foo_id', sa.Integer, sa.ForeignKey('foo.id'), primary_key=True),
    sa.Column('bar_id', sa.Integer, sa.ForeignKey('bar.id'), primary_key=True)
)

def upgrade():
    bind = op.get_bind()

    Base.metadata.create_all(bind=bind)

    session = Session(bind=bind)
    session._model_changes = False  # if you are using Flask-SQLAlchemy, this works around a bug

    f1 = Foo(name='f1')
    f2 = Foo(name='f2')
    b1 = Bar(name='b1')
    b2 = Bar(name='b2')

    f1.bars = [b1, b2]
    b2.foos.append(f2)

    session.add_all([f1, f2, b1, b2])
    session.commit()


def downgrade():
    bind = op.get_bind()

    # in this case all we need to do is drop the tables
    # Base.metadata.drop_all(bind=bind)

    # but we could also delete data
    session = Session(bind=bind)
    session._model_changes = False  # if you are using Flask-SQLAlchemy, this works around a bug

    b1 = session.query(Bar).filter_by(name='b1').one()

    session.delete(b1)
    session.commit()
person davidism    schedule 25.06.2014