专栏首页Devops专栏Flask 自定义模型类

Flask 自定义模型类

自定义模型类

本篇章介绍Flask自定义模型类的概念,以及写一个快速入门的示例,基本内容如下:

  • 定义两个模型类,并创建数据库表
  • 创建数据,写入数据库
  • 编写模板以及视图函数,在页面展示数据

定义模型

模型表示程序使用的数据实体,在Flask-SQLAlchemy中,模型一般是Python类,继承自db.Model,db是SQLAlchemy类的实例,代表程序使用的数据库。

类中的属性对应数据库表中的列。id为主键,是由Flask-SQLAlchemy管理。db.Column类构造函数的第一个参数是数据库列和模型属性类型。

如下示例:定义了两个模型类,作者和书名。

from flask import Flask
from flask_sqlalchemy import SQLAlchemy
import pymysql
pymysql.install_as_MySQLdb()

app = Flask(__name__)

class Config(object):
    """配置参数"""
    # 设置连接数据库的URL
    user = 'root'
    password = '*********'
    database = 'flask_ex'
    app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql://%s:%s@127.0.0.1:3306/%s' % (user,password,database)

    # 设置sqlalchemy自动更跟踪数据库
    SQLALCHEMY_TRACK_MODIFICATIONS = True

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

    # 禁止自动提交数据处理
    app.config['SQLALCHEMY_COMMIT_ON_TEARDOWN'] = False

# 读取配置
app.config.from_object(Config)

# 创建数据库sqlalchemy工具对象
db = SQLAlchemy(app)

#定义模型类-作者
class Author(db.Model):
    __tablename__ = 'author'
    id = db.Column(db.Integer,primary_key=True)
    name = db.Column(db.String(32),unique=True)
    email = db.Column(db.String(64))
    au_book = db.relationship('Book',backref='author')
    def __str__(self):
        return 'Author:%s' %self.name

#定义模型类-书名
class Book(db.Model):
    __tablename__ = 'books'
    id = db.Column(db.Integer,primary_key=True)
    info = db.Column(db.String(32),unique=True)
    leader = db.Column(db.String(32))
    au_book = db.Column(db.Integer,db.ForeignKey('author.id'))
    def __str__(self):
        return 'Book:%s,%s'%(self.info,self.lead)


if __name__ == '__main__':

    # 删除所有表
    db.drop_all()

    # 创建所有表
    db.create_all()

执行脚本之后,到mysql中查看表结构如下。

创建表 db.create_all()

mysql> show tables;
+--------------------+
| Tables_in_flask_ex |
+--------------------+
| author             |
| books              |
+--------------------+
4 rows in set (0.00 sec)

mysql> 

查看author表结构 desc author

mysql> desc author;
+-------+-------------+------+-----+---------+----------------+
| Field | Type        | Null | Key | Default | Extra          |
+-------+-------------+------+-----+---------+----------------+
| id    | int(11)     | NO   | PRI | NULL    | auto_increment |
| name  | varchar(32) | YES  | UNI | NULL    |                |
| email | varchar(64) | YES  |     | NULL    |                |
+-------+-------------+------+-----+---------+----------------+
3 rows in set (0.00 sec)

查看books表结构 desc books

mysql> desc books;
+---------+-------------+------+-----+---------+----------------+
| Field   | Type        | Null | Key | Default | Extra          |
+---------+-------------+------+-----+---------+----------------+
| id      | int(11)     | NO   | PRI | NULL    | auto_increment |
| info    | varchar(32) | YES  | UNI | NULL    |                |
| leader  | varchar(32) | YES  |     | NULL    |                |
| au_book | int(11)     | YES  | MUL | NULL    |                |
+---------+-------------+------+-----+---------+----------------+
4 rows in set (0.00 sec)

创建数据

...
if __name__ == '__main__':

    # 生成数据
    author_1 = Author(name='我吃西红柿', email='xihongshi@163.com')
    author_2 = Author(name='萧潜', email='xiaoqian@126.com')
    author_3 = Author(name='唐家三少', email='sanshao@163.com')
    book_1 = Book(info='吞噬星空', leader='罗峰')
    book_2 = Book(info='寸芒', leader='李杨')
    book_3 = Book(info='飘渺之旅', leader='李强')
    book_4 = Book(info='冰火魔厨', leader='融念冰')
    # 把数据提交给用户会话
    db.session.add_all([author_1, author_2, author_3, book_1, book_2, book_3, book_4])
    # 提交会话
    db.session.commit()

在main方法下创建多个作者以及书本的数据,执行脚本后,查看mysql数据,如下:

mysql> select * from author;
+----+-----------------+-------------------+
| id | name            | email             |
+----+-----------------+-------------------+
|  1 | 我吃西红柿      | xihongshi@163.com |
|  2 | 萧潜            | xiaoqian@126.com  |
|  3 | 唐家三少        | sanshao@163.com   |
+----+-----------------+-------------------+
3 rows in set (0.00 sec)

mysql> select * from books;
+----+--------------+-----------+---------+
| id | info         | leader    | au_book |
+----+--------------+-----------+---------+
|  1 | 吞噬星空     | 罗峰      |    NULL |
|  2 | 寸芒         | 李杨      |    NULL |
|  3 | 飘渺之旅     | 李强      |    NULL |
|  4 | 冰火魔厨     | 融念冰    |    NULL |
+----+--------------+-----------+---------+
4 rows in set (0.00 sec)

mysql> 

使用 flask_wtf 编写视图函数的表单

1.编写完整flask应用,包含视图函数

from flask import Flask,render_template,url_for,redirect,request
from flask_sqlalchemy import SQLAlchemy
import pymysql
pymysql.install_as_MySQLdb()
# 导入Flask-WTF表单
from flask_wtf import FlaskForm
# 导入表单所需要的字段类型
from wtforms import StringField, PasswordField, SubmitField
# 导入表单的验证器
from wtforms.validators import DataRequired, EqualTo
# 启动命令的管理类
from flask_script import Manager   

app = Flask(__name__)

class Config(object):
    """配置参数"""
    # 设置连接数据库的URL
    user = 'root'
    password = '*************'
    database = 'flask_ex'
    app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql://%s:%s@127.0.0.1:3306/%s' % (user,password,database)

    # 设置sqlalchemy自动更跟踪数据库
    SQLALCHEMY_TRACK_MODIFICATIONS = True

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

    # 禁止自动提交数据处理
    app.config['SQLALCHEMY_COMMIT_ON_TEARDOWN'] = False

    # 设置密钥,用于csrf_token的加解密
    app.config["SECRET_KEY"] = "xhosd6f982yfhowefy29f"

# 读取配置
app.config.from_object(Config)

# 创建数据库sqlalchemy工具对象
db = SQLAlchemy(app)

#定义模型类-作者
class Author(db.Model):
    __tablename__ = 'author'
    id = db.Column(db.Integer,primary_key=True)
    name = db.Column(db.String(32),unique=True)
    email = db.Column(db.String(64))
    au_book = db.relationship('Book',backref='author')
    def __str__(self):
        return 'Author:%s' %self.name


#定义模型类-书名
class Book(db.Model):
    __tablename__ = 'books'
    id = db.Column(db.Integer,primary_key=True)
    info = db.Column(db.String(32),unique=True)
    leader = db.Column(db.String(32))
    au_book = db.Column(db.Integer,db.ForeignKey('author.id'))
    def __str__(self):
        return 'Book:%s,%s'%(self.info,self.leader)


# 定义表单的模型类,用来添加书本以及作者数据
class AddAuthorBook(FlaskForm):
    """自定义的注册表单模型类"""
    # DataRequired 保证数据必须填写,并且不能为空
    author_name = StringField(label="作者名称", validators=[DataRequired("作者名称不能为空")]) #  参数:名字,验证器列表
    book_info = StringField(label="书本信息", validators=[DataRequired("书本信息不能为空")])
    submit = SubmitField(label="添加")

@app.route('/',methods=['GET','POST'])
def index():
    #查询所有作者和书名信息
    authors = Author.query.all()
    books = Book.query.all()
    #创建表单对象
    form = AddAuthorBook()
    if form.validate_on_submit():
        #获取表单输入数据
        author_name = form.author_name.data
        book_info = form.book_info.data
        #把表单数据存入模型类
        db_author = Author(name=author_name)
        db_book = Book(info=book_info)
        db_book.leader = author_name
        db_book.au_book = db_author.id
        #提交会话
        db.session.add_all([db_author,db_book])
        db.session.commit()

        #添加数据后,再次查询所有作者和书名信息
        authors = Author.query.all()
        books = Book.query.all()
        return render_template('index.html',author=authors,book=books,form=form)
    else:
        if request.method=='GET':
            render_template('index.html', author=authors, book=books,form=form)
    return render_template('index.html',author=authors,book=books,form=form)

#删除作者
@app.route('/delete_author<id>')
def delete_author(id):
    print('delete author id = %s' % id)
    #精确查询需要删除的作者id
    author = Author.query.get(id)
    db.session.delete(author)
    db.session.commit()
    #直接重定向到index视图函数
    return redirect(url_for('index'))

#删除书名
@app.route('/delete_book<id>')
def delete_book(id):
    print('delete book id = %s' % id)
    #精确查询需要删除的书名id
    book = Book.query.get(id)
    db.session.delete(book)
    db.session.commit()
    #直接重定向到index视图函数
    return redirect(url_for('index'))

if __name__ == '__main__':

    # 创建Manager管理类的对象
    manager = Manager(app)

    # 通过管理对象来启动flask
    manager.run()

2.设置index.html模板

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>Title</title>
</head>
<body>
    <h1>玄幻系列</h1>
    <form method="post">
        {{ form.csrf_token }}
        <p>作者:{{ form.author_name }}</p>
        <p>书名:{{ form.book_info }}</p>
        <p>{{ form.submit }}</p>
    </form>

    <hr>

    <ul>
        {% for item in author %}
        <li>{{ item }}</li><a href='/delete_author{{ item.id }}'>删除</a>
        {% endfor %}
    </ul>

    <hr>

    <ul>
        {% for item in book %}
        <li>{{ item }}</li><a href='/delete_book{{ item.id }}'>删除</a>
        {% endfor %}
    </ul>
</body>
</html>

3.执行python3 db_demo2.py runserver启动服务后,进行功能测试

从上面的几个示例,基本清楚讲解了模型类如何定义,表单如何设置,模板中如何展示数据,表单如何提交数据,数据如何设置删除等功能。

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

我来说两句

0 条评论
登录 后参与评论

相关文章

  • 25. Flask 数据库迁移 flask-migrate

    在开发过程中,需要修改数据库模型,而且还要在修改之后更新数据库。最直接的方式就是删除旧表,但这样会丢失数据。

    Devops海洋的渔夫
  • 解决GitBook不支持[TOC]生成的本页目录

    在编写GitBook的过程中,不可能避免有些篇章会篇幅比较长,此时就会想到markdown有个[TOC]可以生成目录。 但是在使用GitBook运行服务git...

    Devops海洋的渔夫
  • Mysql 5.7.x 导出数据、表结构、导入数据

    如果是导入所有数据库的数据之后,需要flush一下数据库。因为mysql库是包含用户的,如果不flush权限,则会导致这些导入的用户无法登陆使用。

    Devops海洋的渔夫
  • 硬件设备——磁盘相关 原

    磁盘的组成:主要由盘片、机械手臂、磁头、与主轴马达所组成。而数据的写入其实是在盘片上面。盘片上面又可细分出扇区(Sector)与柱面(Cylinder)两种单位...

    阿dai学长
  • Spark源码 —— 从 SparkSubmit 到 Driver启动

    本文主要是以笔记的整理方式写的, 仅以分享的方式供你阅读, 如有不对的地方欢迎指点错误。 读完本文可以学到: 当你用 shell 命令执行 spark-...

    solve
  • python图像处理-图片裁剪

    本系列课程是针对无基础的,争取用简单明了的语言来讲解,学习前需要具备基本的电脑操作能力,准备一个已安装python环境的电脑。如果觉得好可以分享转发,有问题的地...

    用户7054460
  • MOOC教学,什么最重要?

    MOOC教学的关键,不在于录视频,也不在于做幻灯。最重要的是赋能。赋能的具体方法,是在认知资源和学习者之间,建立丰富而有效的链接。

    王树义
  • Spring Cloud 系列之 Eureka 实现服务注册与发现

    如果你对 Spring Cloud 体系还不是很了解,可以先读一下 Spring Cloud 都有哪些模块

    古时的风筝
  • 如何查找并下载rpm依赖包并使用yum离线安装rpm包

    最近在工作中需要postgresql + postgis的离线安装。安装有两种方式:

    create17
  • jdk源码追踪-LinkedList

    关于List接口的类结构和AbstractList的讲解,可以看看上一篇文章jdk源码追踪-ArrayList。

    逝兮诚

扫码关注云+社区

领取腾讯云代金券