数据库：Flask-SQLAlchemy

一、安装以及使用

1、安装

　　安装 flask-sqlalchemy

　　　　pip install flask-sqlalchemy

　　如果连接的是 mysql 数据库，需要安装 mysqldb

　　　　pip install flask-mysqldb

2、连接配置

from flask import Flask
from flask_sqlalchemy import SQLAlchemy


app = Flask(__name__)

#设置连接数据库的URL
app.config['SQLALCHEMY_DATABASE_URI'] ='mysql://root:mysql@127.0.0.1:3306/test'

app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = True
#查询时会显示原始SQL语句
app.config['SQLALCHEMY_ECHO'] = True
db = SQLAlchemy(app)

if __name__ == '__main__':
    app.run(debug=True)

SQLalchemy

3、SQLalchemy模型类设置

　　常见SQLalchemy字段类型

　　常用SQLalchemy列选项

　　常用SQLalchemy关系选项

from flask import Flask
from flask_sqlalchemy import SQLAlchemy


app = Flask(__name__)

#设置连接数据库的URL
app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql://root:mysql@127.0.0.1:3306/test'

app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = True
#查询时会显示原始SQL语句
app.config['SQLALCHEMY_ECHO'] = True
db = SQLAlchemy(app)

class Role(db.Model):
    # 定义表名
    __tablename__ = 'roles'
    # 定义列对象
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(64), unique=True)
    us = db.relationship('User', backref='role')

    #repr()方法显示一个可读字符串
    def __repr__(self):
        return 'Role:%s'% self.name

class User(db.Model):
    __tablename__ = 'users'
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(64), unique=True, index=True)
    email = db.Column(db.String(64),unique=True)
    password = db.Column(db.String(64))
    role_id = db.Column(db.Integer, db.ForeignKey('roles.id'))

    def __repr__(self):
        return 'User:%s'%self.name
if __name__ == '__main__':
    app.run(debug=True)

model.py

class Role(db.Model):
    ...
    #关键代码
    us = db.relationship('User', backref='role', lazy='dynamic')
    ...

class User(db.Model):
    ...
    role_id = db.Column(db.Integer, db.ForeignKey('roles.id'))

#其中realtionship描述了Role和User的关系。在此文中，第一个参数为对应参照的类"User"
#第二个参数backref为类User申明新属性的方法
#第三个参数lazy决定了什么时候SQLALchemy从数据库中加载数据
#    如果设置为子查询方式(subquery)，则会在加载完Role对象后，就立即加载与其关联的对象，这样会让总查询数量减少，但如果返回的条目数量很多，就会比较慢
#        设置为 subquery 的话，role.users 返回所有数据列表
另外,也可以设置为动态方式(dynamic)，这样关联对象会在被使用的时候再进行加载，并且在返回前进行过滤，如果返回的对象数很多，或者未来会变得很多，那最好采用这种方式
#        设置为 dynamic 的话，role.users 返回查询对象，并没有做真正的查询，可以利用查询对象做其他逻辑，比如：先排序再返回结果

一对多

registrations = db.Table('registrations',  
    db.Column('student_id', db.Integer, db.ForeignKey('students.id')),  
    db.Column('course_id', db.Integer, db.ForeignKey('courses.id'))  
)  
class Course(db.Model):
    ...
class Student(db.Model):
    ...
    courses = db.relationship('Course',secondary=registrations,  
                                    backref='students',  
                                    lazy='dynamic')

多对多

class Comment(db.Model):
    """评论"""
    __tablename__ = "comments"

    id = db.Column(db.Integer, primary_key=True)
    # 评论内容
    content = db.Column(db.Text, nullable=False)
    # 父评论id
    parent_id = db.Column(db.Integer, db.ForeignKey("comments.id"))
    # 父评论(也是评论模型)
    parent = db.relationship("Comment", remote_side=[id],
                             backref=db.backref('childs', lazy='dynamic'))

# 测试代码
if __name__ == '__main__':
    db.drop_all()
    db.create_all()

    com1 = Comment(content='我是主评论1')
    com2 = Comment(content='我是主评论2')
    com11 = Comment(content='我是回复主评论1的子评论1')
    com11.parent = com1
    com12 = Comment(content='我是回复主评论1的子评论2')
    com12.parent = com1

    db.session.add_all([com1, com2, com11, com12])
    db.session.commit()
    app.run(debug=True)

自关联一对多

tb_user_follows = db.Table(
    "tb_user_follows",
    db.Column('follower_id', db.Integer, db.ForeignKey('info_user.id'), primary_key=True),  # 粉丝id
    db.Column('followed_id', db.Integer, db.ForeignKey('info_user.id'), primary_key=True)  # 被关注人的id
)

class User(db.Model):
    """用户表"""
    __tablename__ = "info_user"

    id = db.Column(db.Integer, primary_key=True)  
    name = db.Column(db.String(32), unique=True, nullable=False)

    # 用户所有的粉丝，添加了反向引用followed，代表用户都关注了哪些人
    followers = db.relationship('User',
                                secondary=tb_user_follows,
                                primaryjoin=id == tb_user_follows.c.followed_id,
                                secondaryjoin=id == tb_user_follows.c.follower_id,
                                backref=db.backref('followed', lazy='dynamic'),
                                lazy='dynamic')

自关联多对多

4、数据库操作

　　创建表

　　　　db.create_all()

　　删除表

　　　　db.drop_all()

　　增

　　　　增加一条数据

ro1 = Role(name='admin')
db.session.add(ro1)
db.session.commit()

　　　　增加多条数据

us1 = User(name='wang',email='wang@163.com',password='123456',role_id=ro1.id)
us2 = User(name='zhang',email='zhang@189.com',password='201512',role_id=ro2.id)
us3 = User(name='chen',email='chen@126.com',password='987654',role_id=ro2.id)
us4 = User(name='zhou',email='zhou@163.com',password='456789',role_id=ro1.id)
us5 = User(name='tang',email='tang@itheima.com',password='158104',role_id=ro2.id)
us6 = User(name='wu',email='wu@gmail.com',password='5623514',role_id=ro2.id)
us7 = User(name='qian',email='qian@gmail.com',password='1543567',role_id=ro1.id)
us8 = User(name='liu',email='liu@itheima.com',password='867322',role_id=ro1.id)
us9 = User(name='li',email='li@163.com',password='4526342',role_id=ro2.id)
us10 = User(name='sun',email='sun@163.com',password='235523',role_id=ro2.id)
db.session.add_all([us1,us2,us3,us4,us5,us6,us7,us8,us9,us10])
db.session.commit()

　　删

user = User.query.first()
db.session.delete(user)
db.session.commit()

　　改

user = User.query.first()
user.name = 'dong'
db.session.commit()

　　查

#filter_by()精确查询
User.query.filter_by(name='wang').all()

#first()返回查询到的第一个对象
User.query.first()

#all()返回查询到的所有对象(列表形式返回)
User.query.all()

#filter模糊查询，返回名字结尾字符为g的所有数据。
User.query.filter(User.name.endswith('g')).all()

#get():参数为主键，如果主键不存在没有返回内
User.query.get()

#逻辑非，返回名字不等于wang的所有数据
User.query.filter(User.name!='wang').all()

#not_ 相当于取反
from sqlalchemy import not_
User.query.filter(not_(User.name=='chen')).all()

#逻辑与，需要导入and，返回and()条件满足的所有数据
from sqlalchemy import and_
User.query.filter(and_(User.name!='wang',User.email.endswith('163.com'))).all()

#逻辑或，需要导入or_
from sqlalchemy import or_
User.query.filter(or_(User.name!='wang',User.email.endswith('163.com'))).all()

#分页
#从get方法中取得页码
page = request.args.get('page', 1, type = int)
#获取pagination对象
    pagination = Post.query.paginate(page, per_page=10, error_out = False)
#第一个参数表示当前页,第二个参数代表每页显示的数量,error_out=True的情况下如果指定页没有内容将出现404错误,否则返回空的列表
 
#pagination对象的items方法返回当前页的内容列表
    posts = pagination.items

#pagination对象常用方法:
#has_next :是否还有下一页
#has_prev :是否还有上一页
#items : 返回当前页的所有内容
#next(error_out=False) : 返回下一页的Pagination对象
#prev(error_out=False) : 返回上一页的Pagination对象
#page : 当前页的页码(从1开始)
#pages : 总页数
#per_page : 每页显示的数量
#prev_num : 上一页页码数
#next_num :下一页页码数
#query :返回 创建这个Pagination对象的查询对象
#total :查询返回的记录总数
#iter_pages(left_edge=2, left_current=2, right_current=5, right_edge=2)

一对多和多对多操作

"""
以下表关系：
一个用户对应多篇文章（一对多）
一篇文章对应多个标签，一个标签对应多个文章（多对多）
"""
"""
一对一关系中，需要设置relationship中的uselist=Flase，其他数据库操作一样。
一对多关系中，外键设置在多的一方中，关系（relationship）可设置在任意一方。
多对多关系中，需建立关系表，设置 secondary=关系表
"""

# 用户表
class User(db.Model):
    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
    username = db.Column(db.String(50))
    email = db.Column(db.String(50))

# 关系表（多对多）
article_tag_table = db.Table('article_tag',
                             db.Column('article_id', db.Integer, db.ForeignKey('article.id'), primary_key=True),
                             db.Column('tag_id', db.Integer, db.ForeignKey('tag.id'), primary_key=True))

# 文章表
class Article(db.Model):
    __tablename__ = 'article'
    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
    title = db.Column(db.String(100))
    content = db.Column(db.Text)
    # 关联字段，让author_id外键关联User表id. 这里ForeignKey一定要表名.id
    author_id = db.Column(db.Integer, db.ForeignKey('user.id'))
    # 这里不是字段，而是让Article与User表创建关联关系，backref是反向关联关键字
    author = db.relationship("User", backref="articles")
    tags = db.relationship("Tag", secondary=article_tag_table, backref='tags')

# 标签表
class Tag(db.Model):
    __tablename__ = 'tag'
    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
    name = db.Column(db.String(50))

一对多和多对多数据库建立

一对多操作

#一对多关系创建表通过relationship关联
#1数据增加
#方式1：先查出对象，再添加
# 添加一个文章标题是:python基础， 作者是xu
    create_obj = db.session.query(User).filter(User.username=="xu").first()
    article = Article(
        title="python基础",
        author_id=create_obj.id,
        content = "python入门，变量，函数，面向对象..."
    )
    db.session.add(article)
    db.session.commit()
#方式2：通过relationship 添加数据
# 通过定义relationship
    # 1.在Article 表中添加数据，
    # 2.在User表中添加一条数据，并将User表中刚刚添加数据的id，添加到Article中，使其关联起来
    create_obj = Article(
        title="python基础2",
        content="python中生成器，迭代器...",
        author=User(username="wenfeng",email="123@163.com")
                         )
    db.session.add(create_obj)
    db.session.commit()
# 方式3：relationship 反向添加数据
# 创建User对象
    create_obj = User(username = "sun",email="123@163.com")
    # 通过对象.backref定义名字，向Article添加2条数据。
    create_obj.articles = [
        Article(title="python进阶1",content="内存管理机制"),
        Article(title="python进阶2",content="垃圾回收机制")
    ]
    db.session.add(create_obj)
    db.session.commit()
# 最后别忘了，关闭连接
db.session.close()
#2数据查询
# 查询所有数据
# 查询所有文章，然后显示文章名，文章内容，--->连表作者名，作者email
    article_list = db.session.query(Article).all()
    for row in article_list:
        print(row.title,row.content,row.author.username,row.author.email)
# 反向查询
user_list = db.session.query(User).all()
# 先查询所有user.
    # 通过循环每一user对象， 再通过 backref 中的articles，反向查Article表中title
    for row in user_list:
        # print(row.__dict__)
        for row2 in row.articles:
            print(row.username,row2.title)
#3.更新数据
#外键数据更新
# 查询到作者对象
user_obj = db.session.query(User).filter(User.username == "xu").first()
# 查询用户对象id 与 Article外键id 关联所有文章对象，并更新title 
db.session.query(Article).filter(Article.author_id == user_obj.id).update({"title":"vue基础"})
db.session.commit()
#4删除数据
# 删除 作者名为xu的文章
user_obj = db.session.query(User).filter(User.username=="xu").first()
db.session.query(Article).filter(Article.author_id == user_obj.id).delete()
db.session.commit()

多对多操作

#多对多关系，通过第三表将两张表关联起来
#1基于relationship增加数据
# 通过 添加文章，再添加标签
# 首先新建Article对象，
article_obj = Article(title="Go基础",content="go语言指针。。。")
# 添加标签对象。tags为Article表中backref用于关联Tag表多对多关系
article_obj.tags = [Tag(name="编译型语言"),Tag(name="支持协程"),Tag(name="非常快")]
db.session.add(article_obj)
# 提交数据
db.session.commit()
# 关闭连接
db.session.close()
#2基于relationship查询数据
#正向查
# 查询所有书籍和其标签
article_obj = db.session.query(Article).all()
# 遍历循环文章
for row in article_obj:
    # 遍历循环每个文章tag:
    for row2 in row.tags:
        # 打印title,name
        print(row.title,row2.name)
#反向查
# 通过tag标签查询所有文章
# 查询所有tag标签
tag_obj = db.session.query(Tag).all()
# 遍历循环标签
for row in tag_obj:
    # 通过每个标签对象.tags (注意此时tags为Article中字段名)
    #（此字段通过relationship，关联Tag表）
    for row2 in row.tags:
        # 打印tag的名字和文章title
        print(row.name,row2.title)
#3删除数据
#单一删除
# 查询 title为Go基础的对象
article_obj = Article.query.filter(Article.title =="Go基础").first()
# 查询 tag为非常快的对象
tag = Tag.query.filter(Tag.name == "非常快").first()
# 通过文章对象.tags (tags通过relationship与Tag表建立关系)
article_obj.tags.remove(tag)
db.session.commit()
# 级联删除
# 级联删除
# 查询到 title为 Go基础 文章
article_obj = Article.query.filter(Article.title =="Go基础").first()
# 连表查询到所关联所有tags
tags = Tag.query.join(article_tag_table).join(Article).filter(Article.title =="Go基础").all()
for t in tags:
    # 遍历循环tag标签，逐个删除 title为Go基础   的对象
    article_obj.tags.remove(t)
    #最后删除 title为 Go基础 文章对象
    db.session.delete(article_obj)
    db.session.commit()
    # 关闭连接
    db.session.close()

5、数据库迁移：flask-migrate

　　sqlalchemy修改数据库表结构需要删除表，这样会造成数据的丢失。flask-migrate：为了方便数据库的修改，并且追踪每次修改的记录。

　　安装Flask-Migrate

　　　　pip install flask-migrate

#coding=utf-8
from flask import Flask

from flask_sqlalchemy import SQLAlchemy
from flask_migrate import Migrate,MigrateCommand
from flask_script import Shell,Manager

app = Flask(__name__)
manager = Manager(app)

app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql://root:mysql@127.0.0.1:3306/Flask_test'
app.config['SQLALCHEMY_COMMIT_ON_TEARDOWN'] = True
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = True
db = SQLAlchemy(app)

#第一个参数是Flask的实例，第二个参数是Sqlalchemy数据库实例
migrate = Migrate(app,db) 

#manager是Flask-Script的实例，这条语句在flask-Script中添加一个db命令
manager.add_command('db',MigrateCommand)

#定义模型Role
class Role(db.Model):
    # 定义表名
    __tablename__ = 'roles'
    # 定义列对象
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(64), unique=True)
    user = db.relationship('User', backref='role')

    #repr()方法显示一个可读字符串，
    def __repr__(self):
        return 'Role:'.format(self.name)

#定义用户
class User(db.Model):
    __talbe__ = 'users'
    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(64), unique=True, index=True)
    #设置外键
    role_id = db.Column(db.Integer, db.ForeignKey('roles.id'))

    def __repr__(self):
        return 'User:'.format(self.username)


if __name__ == '__main__':
    manager.run()

database.py

　　数据库初始化（创建migrations文件夹）

　　　　python database.py db init

　　数据库migrations

　　　　python database.py db migrate -m "注释"

　　数据库migrate

　　　　python database.py db upgrade

　　查询以前的版本

　　　　python database.py db history

　　回滚到指定版本

　　　　python database.py db downgrade 版本号

posted @ 2019-10-11 15:37 X小白的逆袭之旅阅读(577) 评论(0) 收藏举报

刷新页面返回顶部

X小白的逆袭之旅