Flask学习笔记-（三）数据库ORM操作

最新推荐文章于 2024-05-29 16:06:49 发布

Super.Wong

最新推荐文章于 2024-05-29 16:06:49 发布

阅读量762

点赞数 1

分类专栏： flask 文章标签： flask 数据库 Flask-SQLAlchemy Flask-Migrate

本文链接：https://blog.csdn.net/sdwang198912/article/details/89339215

版权

flask 专栏收录该内容

15 篇文章 10 订阅

订阅专栏

数据库ORM操作

文章目录

数据库ORM操作

1. 数据库ORM操作实现

为了实现ORM方式操作数据库，而不是使用SQL语句，就需要通过安装Flask-SQLAlchemy扩展库来实现。同时通过安装Flask-Migrate扩展库实现数据库更新的跟踪，并实现数据迁移。

pip install Flask-SQLAlchemy
pip install Flask-Migrate

2. 配置sqlite数据库

对于小型应用来说，使用sqlite数据库更为便利，每一个数据库都单独存放在一个文件中。使用sqlite数据库需要在app/config.py配置文件进行相关配置。

# 配置SQLITE数据库信息
BASE_DIR = os.path.abspath(os.path.dirname(__file__))
# 数据库文件存放路径
SQLALCHEMY_DATABASE_URI = 'sqlite:///' + os.path.join(BASE_DIR, '..', 'instance', 'flask.sqlite')
# Flask-SQLAlchemy 是否需要追踪对象的修改并且发送信号。
# 这需要额外的内存， 如果不必要的可以禁用它。
SQLALCHEMY_TRACK_MODIFICATIONS = False

3. 初始化数据库

修改app/__init__.py初始化文件，初始化数据库

......
import os
from flask import Flask
from flask_sqlalchemy import SQLAlchemy
from flask_migrate import Migrate


# 实例化flask_sqlalchemy
db = SQLAlchemy()
# 实例化flask_migrate
migrate = Migrate()


def create_app():
    """应用工厂函数"""
    application = Flask(__name__)
    # 加载config配置文件
    application.config.from_pyfile('config.py', silent=True)

    # 初始化数据库flask_sqlalchemy
    db.init_app(application)
    # 初始化数据库flask_migrate
    import app.models
    migrate.init_app(application, db)
......

4. 创建数据库模型

新建app/models.py文件，创建用户信息的数据库模型。

from app import db


class User(db.Model):
    __tablename__ = 'users'
    id = db.Column(db.Integer, primary_key=True)
    # index用于添加索引
    # unique用于设置唯一索引
    username = db.Column(db.String(64), index=True, unique=True)
    email = db.Column(db.String(120), index=True, unique=True)
    password_hash = db.Column(db.String(128))

    def __repr__(self):
        """打印类对象时的展示方式"""
        return '<User %r>' % self.username

5. 数据库模型初始化

5.1. 迁移目录初始化

通过flask db init命令，初始化数据库模型迁移目录migrations。

(venv) D:\Projects\learn\flask-mega-tutorial>flask db init
Creating directory D:\Projects\learn\flask-mega-tutorial\migrations ... done
Creating directory D:\Projects\learn\flask-mega-tutorial\migrations\versions ... done
Generating D:\Projects\learn\flask-mega-tutorial\migrations\alembic.ini ... done
Generating D:\Projects\learn\flask-mega-tutorial\migrations\env.py ... done
Generating D:\Projects\learn\flask-mega-tutorial\migrations\README ... done
Generating D:\Projects\learn\flask-mega-tutorial\migrations\script.py.mako ... done
Please edit configuration/connection/logging settings in 'D:\\Projects\\learn\\flask-mega-tutorial\\migrations\\alembic.ini' before proceeding.

在这里插入图片描述

5.2. 迁移文件创建

通过flask db migrate命令，将数据库迁移文件生成到迁移目录versions中，同时也初始化创建了Flask-Migrate扩展库依赖的数据库表。

(venv) D:\Projects\learn\flask-mega-tutorial>flask db migrate
INFO  [alembic.runtime.migration] Context impl SQLiteImpl.
INFO  [alembic.runtime.migration] Will assume non-transactional DDL.
INFO  [alembic.autogenerate.compare] Detected added table 'users'
INFO  [alembic.autogenerate.compare] Detected added index 'ix_users_email' on '['email']'
INFO  [alembic.autogenerate.compare] Detected added index 'ix_users_username' on '['username']'
Generating D:\Projects\learn\flask-mega-tutorial\migrations\versions\7c2bb80cfe7b_.py ... done

在这里插入图片描述

5.3. 数据库同步

通过flask db upgrade命令，将数据库迁移文件的变化同步到数据库中。

(venv) D:\Projects\learn\flask-mega-tutorial>flask db upgrade
INFO  [alembic.runtime.migration] Context impl SQLiteImpl.
INFO  [alembic.runtime.migration] Will assume non-transactional DDL.
INFO  [alembic.runtime.migration] Running upgrade  -> 7c2bb80cfe7b, empty message

在这里插入图片描述

6. 新增数据库模型、数据库同步

6.1. 新增数据库模型定义

编辑app/models.py文件，新增posts表的数据库模型定义

from app import db


class User(db.Model):
    __tablename__ = 'users'
    id = db.Column(db.Integer, primary_key=True)
    # index用于添加索引
    # unique用于设置唯一索引
    username = db.Column(db.String(64), index=True, unique=True)
    email = db.Column(db.String(120), index=True, unique=True)
    password_hash = db.Column(db.String(128))
    # 初始化db.relationship
    # 并非实际的数据库字段，只是创建一个虚拟的列，该列会与 Post.user_id (db.ForeignKey) 建立联系
    # 第一个参数表示关联的模型类名；backref用于指定表之间的双向关系；lazy用于定义加载关联对象的方式
    posts = db.relationship('Post', backref='author', lazy='dynamic')

    def __repr__(self):
        """打印类对象时的展示方式"""
        return '<User %r>' % self.username


class Post(db.Model):
    __tablename__ = 'posts'
    id = db.Column(db.Integer, primary_key=True)
    body = db.Column(db.String(140))
    timestamp = db.Column(db.DateTime)
    user_id = db.Column(db.Integer, db.ForeignKey('users.id'))

    def __repr__(self):
        return '<Post %r>'.format(self.body)

6.2. 迁移文件创建

(venv) D:\Projects\learn\flask-mega-tutorial>flask db migrate
INFO  [alembic.runtime.migration] Context impl SQLiteImpl.
INFO  [alembic.runtime.migration] Will assume non-transactional DDL.
INFO  [alembic.autogenerate.compare] Detected added table 'posts'
Generating D:\Projects\learn\flask-mega-tutorial\migrations\versions\7bddf580d26a_.py ... done

6.3. 数据库同步

(venv) D:\Projects\learn\flask-mega-tutorial>flask db upgrade
INFO  [alembic.runtime.migration] Context impl SQLiteImpl.
INFO  [alembic.runtime.migration] Will assume non-transactional DDL.
INFO  [alembic.runtime.migration] Running upgrade 7c2bb80cfe7b -> 7bddf580d26a, empty message

在这里插入图片描述

7. 数据库操作回退

通过flask db downgrade命令，对已同步的数据库进行回退处理，此操作仅限于数据库同步的回退，但是迁移文件还在。

(venv) D:\Projects\learn\flask-mega-tutorial>flask db downgrade
INFO  [alembic.runtime.migration] Context impl SQLiteImpl.
INFO  [alembic.runtime.migration] Will assume non-transactional DDL.
INFO  [alembic.runtime.migration] Running downgrade 7bddf580d26a -> 7c2bb80cfe7b, empty message

在这里插入图片描述

8. 数据库操作测试

8.1. 数据查询

>>> from app import create_app, db
>>> app = create_app()
>>> app.app_context().push()
>>> with app.app_context():
...     from app.models import User
...     user = User.query.all()
...     print(user)
...     
[]

8.2. 数据插入

>>> from app import create_app, db
>>> app = create_app()
>>> app.app_context().push()
>>> with app.app_context():
...     from app.models import User
...     user = User(username='admin', email='1@1.com')
...     db.session.add(user)
...     db.session.commit()
...     db.session.close()
...     users = User.query.all()
...     print(users)
...     
[<User 'admin'>]

在这里插入图片描述

8.3. 数据删除

>>> with app.app_context():
...     from app.models import User
...     db.session.delete(User.query.filter(User.username == 'admin').one())
...     db.session.commit()
...     db.session.close()
...     user = User.query.all()
...     print(user)
...     
[]

在这里插入图片描述

Super.Wong

关注

1
点赞
踩
2

收藏

觉得还不错? 一键收藏
3
评论
Flask学习笔记-（三）数据库ORM操作

数据库操作及用登录实现文章目录数据库操作及用登录实现1. 数据库ORM操作实现2. 配置sqlite数据库3. 编写数据库定义函数5. 初始化数据库6. 创建数据库模型7. 数据库模型初始化7.1. 迁移目录初始化7.2. 迁移文件创建7.3. 数据库同步8. 新增数据库模型、数据库同步8.1. 新增数据库模型定义8.2. 迁移文件创建9. 数据库操作回退10. 数据库操作测试10.1. 数据查...
复制链接

扫一扫