Flask之自定义模型类

Posted 2021-01-12 alexzhang92

4.3自定义模型类

定义模型

模型表示程序使用的数据实体，在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

查看books表结构 desc 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:[email protected]/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<id>‘)
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<id>‘)
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)

生成数据后，查看数据：

模板页面示例：

   <h1>玄幻系列</h1>
    <form method="post">
        {{ form.csrf_token }}
        <p>作者：{{ form.au_info }}</p>
        <p>书名：{{ form.bk_info }}</p>
        <p>{{ form.submit }}</p>
    </form>
    <ul>
        <li>{% for x in author %}</li>
        <li>{{ x }}</li><a href=‘/delete_author{{ x.id }}‘>删除</a>
        <li>{% endfor %}</li>
    </ul>
    <hr>
    <ul>
        <li>{% for x in book %}</li>
        <li>{{ x }}</li><a href=‘/delete_book{{ x.id }}‘>删除</a>
        <li>{% endfor %}</li>
    </ul>

 

添加数据后，查看数据：