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': '3306',
    '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)的表会被删除