数据库的管理
当开发到一半的时候,数据库需要修改是一件很麻烦的事情,好在flask提供了很好用的拓展,flask-migrate和flask-script,主要还是前者.flask-migrate是对Alembic的封装。
有些文章的介绍有些语焉不详,所以我直接找了一个开源项目进行了对比改造。
首先我的文件架构如下
这里的models
存放了我的各个关系表,主角是下面的manager.py
文件,
代码如下
import os
from app import create_app,db
from app.models import User,Task,Comment
from flask import g
from flask_script import Manager, Shell
from flask_migrate import Migrate, MigrateCommand
app = create_app(os.getenv('FLASK_CONFIG') or 'default')
manager = Manager(app)
migrate = Migrate(app, db)
with app.app_context():
g.contextPath = ''
def make_shell_context():
return dict(app=app, db=db, User=User,Task=Task,Comment=Comment)
manager.add_command("shell", Shell(make_context=make_shell_context))
manager.add_command('db', MigrateCommand)
@manager.command
def myprint():
print('hello world')
if __name__ == '__main__':
manager.run()
其中manager.add_command()
添加了不同的指令,会在之后用到
- 首先我们要初始化
>python manager.py db init
初始化成功后就会生成一个migrations
文件夹,里面存放了版本控制的信息
- 迁移数据库
>python manager.py db migrate
会在version
文件夹下生成一个包含版本信息的文件,里面就定义了更新的方法
- 更新
>python manager.py db upgrade
这一步我在本地的windows环境下总是会报错某个表已存在,然后清空了数据库里面的表和migration
文件夹再次重复操作,还是报错。但是放在linux服务器上时完全没有问题。查看了服务器上的数据库确实修改了。具体原因未知。听说是Alembic不能在windows下使用,但是其他功能挺正常的。
希望大牛指点。