自定义模型类
定义模型
模型表示程序使用的数据实体，在Flask-SQLAlchemy中，模型一般是Python类，继承自db.Model，db是SQLAlchemy类的实例，代表程序使用的数据库。
类中的属性对应数据库表中的列。id为主键，是由Flask-SQLAlchemy管理。db.Column类构造函数的第一个参数是数据库列和模型属性类型。
如下示例：定义了两个模型类，作者和书名。
#coding=utf-8
from flask import Flask,render_template,redirect,url_for
from flask_sqlalchemy import SQLAlchemy
app = Flask(name)
#设置连接数据
app.config[‘SQLALCHEMY_DATABASE_URI’] = ‘mysql://root:[email protected]:3306/test1’
#设置每次请求结束后会自动提交数据库中的改动
app.config[‘SQLALCHEMY_COMMIT_ON_TEARDOWN’] = True
#设置成 True，SQLAlchemy 将会追踪对象的修改并且发送信号。这需要额外的内存， 如果不必要的可以禁用它。
app.config[‘SQLALCHEMY_TRACK_MODIFICATIONS’] = True
#实例化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)
创建表 db.create_all()
生成表
查看author表结构 desc author
查看author
查看books表结构 desc books
查看books
#coding=utf-8
from flask import Flask,render_template,url_for,redirect,request
from flask_sqlalchemy import SQLAlchemy
from flask_wtf import FlaskForm
from wtforms.validators import DataRequired
from wtforms import StringField,SubmitField
app = Flask(name)
app.config[‘SQLALCHEMY_DATABASE_URI’] = ‘mysql://root:mysql@localhost/test1’
app.config[‘SQLALCHEMY_COMMIT_ON_TEARDOWN’] = True
app.config[‘SQLALCHEMY_TRACK_MODIFICATIONS’] = True
app.config[‘SECRET_KEY’]=‘s’
db = SQLAlchemy(app)
#创建表单类，用来添加信息
class Append(Form):
au_info = StringField(validators=[DataRequired()])
bk_info = StringField(validators=[DataRequired()])
submit = SubmitField(u’添加’)
@app.route(’/’,methods=[‘GET’,‘POST’])
def index():
#查询所有作者和书名信息
author = Author.query.all()
book = Book.query.all()
#创建表单对象
form = Append()
if form.validate_on_submit():
#获取表单输入数据
wtf_au = form.au_info.data
wtf_bk = form.bk_info.data
#把表单数据存入模型类
db_au = Author(name=wtf_au)
db_bk = Book(info=wtf_bk)
#提交会话
db.session.add_all([db_au,db_bk])
db.session.commit()
#添加数据后，再次查询所有作者和书名信息
author = Author.query.all()
book = Book.query.all()
return render_template(‘index.html’,author=author,book=book,form=form)
else:
if request.method==‘GET’:
render_template(‘index.html’, author=author, book=book,form=form)
return render_template(‘index.html’,author=author,book=book,form=form)
#删除作者
@app.route(’/delete_author’)
def delete_author(id):
#精确查询需要删除的作者id
au = Author.query.filter_by(id=id).first()
db.session.delete(au)
#直接重定向到index视图函数
return redirect(url_for(‘index’))
#删除书名
@app.route(’/delete_book’)
def delete_book(id):
#精确查询需要删除的书名id
bk = Book.query.filter_by(id=id).first()
db.session.delete(bk)
#直接重定向到index视图函数
return redirect(url_for(‘index’))
if name == ‘main’:
db.drop_all()
db.create_all()
#生成数据
au_xi = Author(name=‘我吃西红柿’,email=‘[email protected]’)
au_qian = Author(name=‘萧潜’,email=‘[email protected]’)
au_san = Author(name=‘唐家三少’,email=‘[email protected]’)
bk_xi = Book(info=‘吞噬星空’,lead=‘罗峰’)
bk_xi2 = Book(info=‘寸芒’,lead=‘李杨’)
bk_qian = Book(info=‘飘渺之旅’,lead=‘李强’)
bk_san = Book(info=‘冰火魔厨’,lead=‘融念冰’)
#把数据提交给用户会话
db.session.add_all([au_xi,au_qian,au_san,bk_xi,bk_xi2,bk_qian,bk_san])
#提交会话
db.session.commit()
app.run(debug=True)
生成数据后，查看数据：
生成数据
模板页面示例：
玄幻系列
{{ form.csrf_token }}作者：{{ form.au_info }}
书名：{{ form.bk_info }}
{{ form.submit }}
- {% for x in author %}
- {{ x }} 删除
- {% endfor %}
- {% for x in book %}
- {{ x }} 删除
- {% endfor %}