24. Flask 自定义模型类

2024-06-29 11:27| 来源: 网络整理| 查看: 265

自定义模型类

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

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

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

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

如下示例：定义了两个模型类，作者和书名。代码语言：javascript复制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:%[email protected]: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中查看表结构如下。

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

代码语言：javascript复制mysql> select * from author; +----+-----------------+-------------------+ | id | name | email | +----+-----------------+-------------------+ | 1 | 我吃西红柿 | [email protected] | | 2 | 萧潜 | [email protected] | | 3 | 唐家三少 | [email protected] | +----+-----------------+-------------------+ 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应用，包含视图函数

代码语言：javascript复制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:%[email protected]: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 requesthod=='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') 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') 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模板

代码语言：javascript复制 Title 玄幻系列 {{ form.csrf_token }}

作者：{{ form.author_name }}

书名：{{ form.book_info }}

{% for item in author %} {{ item }}删除 {% endfor %} {% for item in book %} {{ item }}删除 {% endfor %}

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

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

【本文地址】

公司简介

联系我们

今日新闻

推荐新闻

专题文章