1. 技术文章
  2. flask数据库迁移

flask数据库迁移

由 时光总嘲笑我的痴心妄想 提交于 2020-03-01 00:31:21

数据库迁移

  • 在开发过程中，需要修改数据库模型，而且还要在修改之后更新数据库。最直接的方式就是删除旧表，但这样会丢失数据。
  • 更好的解决办法是使用数据库迁移框架，它可以追踪数据库模式的变化，然后把变动应用到数据库中。
  • 在Flask中可以使用Flask-Migrate扩展，来实现数据迁移。并且集成到Flask-Script中，所有操作通过命令就能完成。
  • 为了导出数据库迁移命令，Flask-Migrate提供了一个MigrateCommand类，可以附加到flask-script的manager对象上。

首先要在虚拟环境中安装Flask-Migrate。

pip install flask-migrate
  • 代码文件内容：
  1. #coding=utf-8
  2. from flask import Flask
  3.  
  4. from flask_sqlalchemy import SQLAlchemy
  5. from flask_migrate import Migrate,MigrateCommand
  6. from flask_script import Shell,Manager
  7.  
  8. app = Flask(__name__)
  9. manager = Manager(app)
  10.  
  11. app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql://root:mysql@127.0.0.1:3306/Flask_test'
  12. app.config['SQLALCHEMY_COMMIT_ON_TEARDOWN'] = True
  13. app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = True
  14. db = SQLAlchemy(app)
  15.  
  16. #第一个参数是Flask的实例，第二个参数是Sqlalchemy数据库实例
  17. migrate = Migrate(app,db) 
  18.  
  19. #manager是Flask-Script的实例，这条语句在flask-Script中添加一个db命令
  20. manager.add_command('db',MigrateCommand)
  21.  
  22. #定义模型Role
  23. class Role(db.Model):
  24.     # 定义表名
  25.     __tablename__ = 'roles'
  26.     # 定义列对象
  27.     id = db.Column(db.Integer, primary_key=True)
  28.     name = db.Column(db.String(64), unique=True)
  29.     user = db.relationship('User', backref='role')
  30.  
  31.     #repr()方法显示一个可读字符串，
  32.     def __repr__(self):
  33.         return 'Role:'.format(self.name)
  34.  
  35. #定义用户
  36. class User(db.Model):
  37.     __talbe__ = 'users'
  38.     id = db.Column(db.Integer, primary_key=True)
  39.     username = db.Column(db.String(64), unique=True, index=True)
  40.     #设置外键
  41.     role_id = db.Column(db.Integer, db.ForeignKey('roles.id'))
  42.  
  43.     def __repr__(self):
  44.         return 'User:'.format(self.username)
  45.  
  46.  
  47. if __name__ == '__main__':
  48.     manager.run()

创建迁移仓库

  1. #这个命令会创建migrations文件夹，所有迁移文件都放在里面。
  2. python database.py db init


创建迁移脚本

  • 自动创建迁移脚本有两个函数
    • upgrade()：函数把迁移中的改动应用到数据库中。
    • downgrade()：函数则将改动删除。
  • 自动创建的迁移脚本会根据模型定义和数据库当前状态的差异，生成upgrade()和downgrade()函数的内容。
  • 对比不一定完全正确，有可能会遗漏一些细节，需要进行检查
python database.py db migrate -m 'initial migration'


更新数据库

python database.py db upgrade

返回以前的版本

可以根据history命令找到版本号,然后传给downgrade命令:

  1. python app.py db history
  2.  
  3. 输出格式：<base> ->  版本号 (head), initial migration
  • 回滚到指定版本
python app.py db downgrade 版本号

实际操作顺序:

  • 1.python 文件 db init
  • 2.python 文件 db migrate -m"版本名(注释)"
  • 3.python 文件 db upgrade 然后观察表结构
  • 4.根据需求修改模型
  • 5.python 文件 db migrate -m"新版本名(注释)"
  • 6.python 文件 db upgrade 然后观察表结构
  • 7.若返回版本,则利用 python 文件 db history查看版本号
  • 8.python 文件 db downgrade(upgrade) 版本号
                        <li class="tool-item tool-active is-like "><a href="javascript:;"><svg class="icon" aria-hidden="true">
                        <use xlink:href="#csdnc-thumbsup"></use>
                    </svg><span class="name">点赞</span>
                    <span class="count">1</span>
                    </a></li>
                    <li class="tool-item tool-active is-collection "><a href="javascript:;" data-report-click="{&quot;mod&quot;:&quot;popu_824&quot;}"><svg class="icon" aria-hidden="true">
                        <use xlink:href="#icon-csdnc-Collection-G"></use>
                    </svg><span class="name">收藏</span></a></li>
                    <li class="tool-item tool-active is-share"><a href="javascript:;" data-report-click="{&quot;mod&quot;:&quot;1582594662_002&quot;}"><svg class="icon" aria-hidden="true">
                        <use xlink:href="#icon-csdnc-fenxiang"></use>
                    </svg>分享</a></li>
                    <!--打赏开始-->
                                            <!--打赏结束-->
                                            <li class="tool-item tool-more">
                        <a>
                        <svg t="1575545411852" class="icon" viewBox="0 0 1024 1024" version="1.1" xmlns="http://www.w3.org/2000/svg" p-id="5717" xmlns:xlink="http://www.w3.org/1999/xlink" width="200" height="200"><defs><style type="text/css"></style></defs><path d="M179.176 499.222m-113.245 0a113.245 113.245 0 1 0 226.49 0 113.245 113.245 0 1 0-226.49 0Z" p-id="5718"></path><path d="M509.684 499.222m-113.245 0a113.245 113.245 0 1 0 226.49 0 113.245 113.245 0 1 0-226.49 0Z" p-id="5719"></path><path d="M846.175 499.222m-113.245 0a113.245 113.245 0 1 0 226.49 0 113.245 113.245 0 1 0-226.49 0Z" p-id="5720"></path></svg>
                        </a>
                        <ul class="more-box">
                            <li class="item"><a class="article-report">文章举报</a></li>
                        </ul>
                    </li>
                                        </ul>
            </div>
                        </div>
        <div class="person-messagebox">
            <div class="left-message"><a href="https://blog.csdn.net/feilzhang">
                <img src="https://profile.csdnimg.cn/E/4/7/3_feilzhang" class="avatar_pic" username="feilzhang">
                                        <img src="https://g.csdnimg.cn/static/user-reg-year/2x/2.png" class="user-years">
                                </a></div>
            <div class="middle-message">
                                    <div class="title"><span class="tit"><a href="https://blog.csdn.net/feilzhang" data-report-click="{&quot;mod&quot;:&quot;popu_379&quot;}" target="_blank">zfl273</a></span>
                                        </div>
                <div class="text"><span>发布了95 篇原创文章</span> · <span>获赞 164</span> · <span>访问量 8万+</span></div>
            </div>
                            <div class="right-message">
                                        <a href="https://im.csdn.net/im/main.html?userName=feilzhang" target="_blank" class="btn btn-sm btn-red-hollow bt-button personal-letter">私信
                    </a>
                                                        <a class="btn btn-sm  bt-button personal-watch" data-report-click="{&quot;mod&quot;:&quot;popu_379&quot;}">关注</a>
                                </div>
                        </div>
                </div>

    数据库迁移

    • 在开发过程中，需要修改数据库模型，而且还要在修改之后更新数据库。最直接的方式就是删除旧表，但这样会丢失数据。
    • 更好的解决办法是使用数据库迁移框架，它可以追踪数据库模式的变化，然后把变动应用到数据库中。
    • 在Flask中可以使用Flask-Migrate扩展，来实现数据迁移。并且集成到Flask-Script中，所有操作通过命令就能完成。
    • 为了导出数据库迁移命令，Flask-Migrate提供了一个MigrateCommand类，可以附加到flask-script的manager对象上。

    首先要在虚拟环境中安装Flask-Migrate。

    pip install flask-migrate
    • 代码文件内容：
    1. #coding=utf-8
    2. from flask import Flask
    3.  
    4. from flask_sqlalchemy import SQLAlchemy
    5. from flask_migrate import Migrate,MigrateCommand
    6. from flask_script import Shell,Manager
    7.  
    8. app = Flask(__name__)
    9. manager = Manager(app)
    10.  
    11. app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql://root:mysql@127.0.0.1:3306/Flask_test'
    12. app.config['SQLALCHEMY_COMMIT_ON_TEARDOWN'] = True
    13. app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = True
    14. db = SQLAlchemy(app)
    15.  
    16. #第一个参数是Flask的实例，第二个参数是Sqlalchemy数据库实例
    17. migrate = Migrate(app,db) 
    18.  
    19. #manager是Flask-Script的实例，这条语句在flask-Script中添加一个db命令
    20. manager.add_command('db',MigrateCommand)
    21.  
    22. #定义模型Role
    23. class Role(db.Model):
    24.     # 定义表名
    25.     __tablename__ = 'roles'
    26.     # 定义列对象
    27.     id = db.Column(db.Integer, primary_key=True)
    28.     name = db.Column(db.String(64), unique=True)
    29.     user = db.relationship('User', backref='role')
    30.  
    31.     #repr()方法显示一个可读字符串，
    32.     def __repr__(self):
    33.         return 'Role:'.format(self.name)
    34.  
    35. #定义用户
    36. class User(db.Model):
    37.     __talbe__ = 'users'
    38.     id = db.Column(db.Integer, primary_key=True)
    39.     username = db.Column(db.String(64), unique=True, index=True)
    40.     #设置外键
    41.     role_id = db.Column(db.Integer, db.ForeignKey('roles.id'))
    42.  
    43.     def __repr__(self):
    44.         return 'User:'.format(self.username)
    45.  
    46.  
    47. if __name__ == '__main__':
    48.     manager.run()

    创建迁移仓库

    1. #这个命令会创建migrations文件夹，所有迁移文件都放在里面。
    2. python database.py db init


    创建迁移脚本

    • 自动创建迁移脚本有两个函数
      • upgrade()：函数把迁移中的改动应用到数据库中。
      • downgrade()：函数则将改动删除。
    • 自动创建的迁移脚本会根据模型定义和数据库当前状态的差异，生成upgrade()和downgrade()函数的内容。
    • 对比不一定完全正确，有可能会遗漏一些细节，需要进行检查
    python database.py db migrate -m 'initial migration'


    更新数据库

    python database.py db upgrade

    返回以前的版本

    可以根据history命令找到版本号,然后传给downgrade命令:

    1. python app.py db history
    2.  
    3. 输出格式：<base> ->  版本号 (head), initial migration
    • 回滚到指定版本
    python app.py db downgrade 版本号

    实际操作顺序:

    • 1.python 文件 db init
    • 2.python 文件 db migrate -m"版本名(注释)"
    • 3.python 文件 db upgrade 然后观察表结构
    • 4.根据需求修改模型
    • 5.python 文件 db migrate -m"新版本名(注释)"
    • 6.python 文件 db upgrade 然后观察表结构
    • 7.若返回版本,则利用 python 文件 db history查看版本号
    • 8.python 文件 db downgrade(upgrade) 版本号
    标签
    Flask
    数据库文件
    python实例
    数据库
    命令模式
    版本号
    sqlalchemy
    python
    易学教程内所有资源均来自网络或用户发布的内容，如有违反法律规定的内容欢迎反馈！
    该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!