Flask-SQLAlchemy 및 Alembic: Python 웹 애플리케이션에서 데이터베이스 마이그레이션 모범 사례(2부)-파이썬 튜토리얼-php.cn

Flask-SQLAlchemy 및 Alembic: Python 웹 애플리케이션에서 데이터베이스 마이그레이션 모범 사례(2부)

PHPz

풀어 주다： 2023-06-17 20:34:38

원래의

1341명이 탐색했습니다.

Flask-SQLAlchemy 및 Alembic: Python 웹 애플리케이션에서 데이터베이스 마이그레이션을 위한 모범 사례(2부)

이전 기사에서는 Flask-SQLAlchemy와 Alembic이 함께 작동하는 방식에 대해 논의했습니다. 이 기사에서는 일부 기본 데이터 모델에서 열을 추가 및 제거하는 방법과 일부 열의 유형 또는 제약 조건을 수정하는 방법을 주로 소개합니다. 이러한 변경은 실제 프로젝트 개발 중에 매우 일반적입니다.

열 추가 및 삭제

데이터베이스 마이그레이션을 위해 Flask-SQLAlchemy 및 Alembic을 사용할 때 테이블 열을 추가하고 제거하는 것은 매우 일반적입니다. 이 프로세스를 보여주기 위해 다음 예제 Person 모델에 몇 가지 새로운 열을 추가하겠습니다.

from flask_sqlalchemy import SQLAlchemy

db = SQLAlchemy()

class Person(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(255), nullable=False)
    age = db.Column(db.Integer)

    def __repr__(self):
        return '<Person %r>' % self.name

로그인 후 복사

두 개의 새 열을 추가한다고 가정해 보겠습니다. 하나는 생년월일(birthdate)이고 다른 하나는 결혼 여부(is_married)를 나타내는 부울 열입니다. 다음 명령을 사용하여 마이그레이션 스크립트를 생성할 수 있습니다.

$ alembic revision -m "add birthdate, is_married columns to person"

로그인 후 복사

다음으로 생성된 .py 마이그레이션 스크립트 파일을 수정하여 새 열을 추가해야 합니다. 업그레이드() 함수에서 add_column()을 사용할 수 있습니다.

from alembic import op
import sqlalchemy as sa

def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.add_column('person', sa.Column('birthdate', sa.Date(), nullable=True))
    op.add_column('person', sa.Column('is_married', sa.Boolean(), nullable=True))
    # ### end Alembic commands ###

로그인 후 복사

열을 삭제하려면 해당 소멸자 Degrad()에서 drop_column() 함수를 사용하여 데이터베이스 모델에서 열을 삭제할 수 있습니다.

def downgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.drop_column('person', 'birthdate')
    op.drop_column('person', 'is_married')
    # ### end Alembic commands ###

로그인 후 복사

이 마이그레이션 스크립트의 전체 샘플 코드는 아래에서 찾을 수 있습니다.

"""add birthdate, is_married columns to person"""
from alembic import op
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision = 'eab2c4f1c9fb'
down_revision = '7cfae59c2402'
branch_labels = None
depends_on = None


def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.add_column('person', sa.Column('birthdate', sa.Date(), nullable=True))
    op.add_column('person', sa.Column('is_married', sa.Boolean(), nullable=True))
    # ### end Alembic commands ###


def downgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.drop_column('person', 'birthdate')
    op.drop_column('person', 'is_married')
    # ### end Alembic commands ###

로그인 후 복사

열 유형 변경 및 제약 조건 수정

많은 경우 열 유형 및 제약 조건을 수정해야 합니다. Person 모델의 age 열 유형을 INTEGER에서 SMALLINT로 변경한다고 가정합니다. 이를 달성하기 위해 생성된 .py 마이그레이션 스크립트 파일에서 alter_column() 함수를 사용할 수 있습니다.

def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.alter_column('person', 'age', existing_type=sa.Integer(),
               type_=sa.SmallInteger(), nullable=True)
    # ### end Alembic commands ###

로그인 후 복사

열에 대한 제약 조건을 수정할 수도 있습니다. Person 모델을 조사해 보면 모델에 고유 값 제약 조건이 없다는 것을 알 수 있습니다. 다음 코드를 사용하여 이름 및 생년월일 열에 고유 값 제약 조건을 추가할 수 있습니다.

from alembic import op
import sqlalchemy as sa

def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.create_unique_constraint(op.f('uq_person_name'), 'person', ['name'])
    op.create_unique_constraint(op.f('uq_person_birthdate'), 'person', ['birthdate'])
    # ### end Alembic commands ###

로그인 후 복사

나중에 고유 값 제약 조건을 취소해야 하는 경우 drop_constraint() 함수를 사용할 수 있습니다. 예:

def downgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.drop_constraint(op.f('uq_person_name'), 'person', type_='unique')
    op.drop_constraint(op.f('uq_person_birthdate'), 'person', type_='unique')
    # ### end Alembic commands ###

로그인 후 복사

이 문서에서는 열 추가 및 제거, 열 제약 조건 변경, 열 데이터 유형 변경 등 몇 가지 일반적인 유형의 데이터베이스 스키마 변경을 다룹니다. Flask-SQLAlchemy 및 Alembic을 사용한 데이터베이스 마이그레이션 모범 사례를 보여줍니다. 이를 통해 데이터베이스 스키마를 보다 효율적으로 관리하고 팀 환경에서 마이그레이션 파일을 보다 쉽게 공유할 수 있습니다.

참조 링크: