sqlalchemy + alembic数据迁移

需要安装的包工具

---

• pip install pymysql
• pip install sqlalchemy
• pip install alembic

 

创建表

---

• 新建models.py

 from sqlalchemy import create_engine
 from sqlalchemy.ext.declarative import declarative_base
 from sqlalchemy import Column, Integer, String

 DATABASE = {
     'HOST': 'localhost',
     'USERNAME': 'root',
     'PASSWORD': 'qwe123',
     'PORT': '',
     'DATABASE': 'test',
     'PARAMS': 'utf8'
 }

 MYSQL_URL = 'mysql+pymysql://{USERNAME}:{PASSWORD}@{HOST}:{PORT}/{DATABASE}?{PARAMS}'

 engine = create_engine(MYSQL_URL.format(**DATABASE))

 Base = declarative_base(engine)

 class Test(Base):
     __tablename__ = 'test'
     id = Column(Integer, autoincrement=True, primary_key=True)
     name = Column(String(20))
     # password = Column(String(30))

 if __name__ == '__main__':
     Base.metadata.create_all()

alembic 初始化和配置

---

• 在项目的根目录下：

  • alembic init alembic
• 执行完命令之后会在项目的目录下生成如下目录
• 
• 修改alembic.ini 文件
• 

  该配置等于model.py里面的mysql_url格式化后的结果(后面的参数去掉)

• 修改env.py
• 

  在 env.py 中设置，将target_metadata赋值成数据库的元数据(metadata)
  如果执行 revision 有 import 报错，注意是否正确将当前项目目录添加到 sys.path 路径

　　

迁移表

---

• 将models.py中的password的注释消除
• 执行alembic revision --autogenerate -m "add password"
• 执行alembic upgrade head
• 再去数据库查看，新的表的字段已经被加上了
• alembic在windows下会有些异常
• alembic迁移数据表时，不是项目的数据表(没有继承Base)的表会被删除