实践操作alembic数据库迁移增删改查以及协同开发

前言：开发过程中需要用到，这边实践操作alembic数据库迁移增删改查

安装alembic

pip3 install alembic

使用alembic

根据模型model类生成数据表

alembic revision --autogenerate -m "initial migration"

执行完上面的命令之后，alembic目录中会多出对应commit的文件，如下图所示

应用当前数据表alembic commit更新操作

alembic upgrade head

此时的数据库结构已经同步到最新情况，如下图所示

当执行完revision操作之后，对应的数据库中会有一个alembic_version表来记录版本号信息，如下图所示

注意点，如果在执行alembic upgrade head的时候提示type "vector" does not exist情况，如下图所示

记得在pg数据库中安装相关扩展即可

CREATE EXTENSION IF NOT EXISTS vector;

现有数据库中添加新模型model数据表

此时有一个新的业务需求，需要编写一个新数据表对应的model，如下所示

class ModelTest(Base):
    __tablename__ = "modeltest"
    __table_args__ = {'comment': '指纹测试表'}

    id = Column(Integer, primary_key=True, index=True, comment="指纹ID")
    product_id = Column(Integer, ForeignKey("products.id"), comment="产品ID")
    content = Column(Text, comment="指纹内容")
    updated_at = Column(DateTime(timezone=True), onupdate=func.now(),server_default=func.now(),comment="更新时间")
    created_at = Column(DateTime(timezone=True), server_default=func.now(), comment="记录创建时间")
    deleted_at = Column(DateTime(timezone=True), comment="记录删除时间")

    product = relationship("Product", back_populates="fingerprint")

重新进行revision操作，执行如下命令

alembic revision --autogenerate -m "add modeltest demo"

应用当前数据表alembic commit更新操作

alembic upgrade head

重新查看数据库，可以看到对应的modeltest已经出现了，如下图所示

修改现在数据表字段

原modeltest类的定义如下所示

class ModelTest(Base):
    __tablename__ = "modeltest"
    __table_args__ = {'comment': '指纹测试表'}

    id = Column(Integer, primary_key=True, index=True, comment="指纹ID")
    product_id = Column(Integer, ForeignKey("products.id"), comment="产品ID")
    content = Column(Text, comment="指纹内容")
    updated_at = Column(DateTime(timezone=True), onupdate=func.now(),server_default=func.now(),comment="更新时间")
    created_at = Column(DateTime(timezone=True), server_default=func.now(), comment="记录创建时间")
    deleted_at = Column(DateTime(timezone=True), comment="记录删除时间")

    product = relationship("Product", back_populates="fingerprint")

添加一个新的content2字段来进行定义，如下所示

class ModelTest(Base):
    __tablename__ = "modeltest"
    __table_args__ = {'comment': '指纹测试表'}

    id = Column(Integer, primary_key=True, index=True, comment="指纹ID")
    product_id = Column(Integer, ForeignKey("products.id"), comment="产品ID")
    content = Column(Text, comment="指纹内容")
    content2 = Column(Text, comment="指纹内容2")
    updated_at = Column(DateTime(timezone=True), onupdate=func.now(),server_default=func.now(),comment="更新时间")
    created_at = Column(DateTime(timezone=True), server_default=func.now(), comment="记录创建时间")
    deleted_at = Column(DateTime(timezone=True), comment="记录删除时间")

    product = relationship("Product", back_populates="fingerprint")