版权声明:转载请注明出处 https://blog.csdn.net/qq_42844049/article/details/86909039
思路梳理
- 配置数据库
a.导入SQLAlchemy扩展
b.创建db对象并配置参数
c.创建数据库 - 添加书和作者模型
a.设计表结构
b.模型继承自db.Model
c._ _ tablename _ _ = ‘表名’
d.db.Column 标明字段
e.db.relationship 关系引用 - 添加数据
- 使用模板显示数据库查询数据
a.查询所有作者信息,并把信息传递给模板
b.模板for循环出作者和书籍(作者获取书籍是关系引用books属性) - 使用WTF显示表单
a.自定义表单类
b.在模板中显示
c.secret_key |csrf_token - 实现相关的增删逻辑
增:
a.调用WTF函数实现验证
b.验证通过获取数据
c.判断作者是否存在
d.如果作者存在,判断书籍是否存在,没有重复书籍就添加数据,有就提示重复
e.如果作者不存在添加作者书籍
f.验证不通过就提示错误
删:
a.删除书籍—>网页中删除—>点击需要发送书籍的ID给删除书籍的路由—>路由需要接收参数(先查询是否存在该书籍,如果存在就删除如果不存在提示错误,使用url_for()传参)
b.删除作者(先删除书籍再删除作者)
数据表结构
Author(作者表) | 属性(便于查找) | |
---|---|---|
id (主键) | name | books |
1 | chengxuyuan | python入门flask入门 |
Book(书籍表) | 属性 | ||
---|---|---|---|
id | name | author_id(外键 多) | author(引用) |
1 | Python入门 | 1 | author.name = chengxuyuan |
2 | Flask入门 | 1 | author.name = chengxuyuan |
完整代码
from flask import Flask,render_template,flash,request,redirect,url_for
from flask_sqlalchemy import SQLAlchemy
from flask_wtf import FlaskForm
from wtforms import StringField,SubmitField
from wtforms.validators import DataRequired
app = Flask(__name__)
#数据库配置
app.secret_key = 'csdnblog'
app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql://root:[email protected]:3306/flask_books'
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
db = SQLAlchemy(app)
#作者模型
class Author(db.Model):
__tablename__ = 'authors'
id = db.Column(db.Integer,primary_key=True)
name = db.Column(db.String(16),unique=True)
#关系引用
books = db.relationship('Book',backref='author')
def __repr__(self):
return 'Author :%s' % self.name
#书籍模型
class Book(db.Model):
__tablename__ = 'books'
id = db.Column(db.Integer,primary_key=True)
name = db.Column(db.String(16),unique=True)
author_id = db.Column(db.Integer,db.ForeignKey('authors.id'))
def __repr__(self):
return 'Book : %s %s' % (self.name,self.author_id)
#自定义表单类
class AuthorForm(FlaskForm):
author = StringField('作者',validators=[DataRequired()])
book = StringField('书籍',validators=[DataRequired()])
submit = SubmitField('提交')
#删除作者逻辑同删除书籍
@app.route('/delete_author/<author_id>')
def delete_author(author_id):
author = Author.query.get(author_id)
if author:
try:
#查询完后直接删除
Book.query.filter_by(author_id=author.id).delete()
#删除作者
db.session.delete(author)
db.session.commit()
except Exception as e:
print(e)
flash('删除作者失败')
db.session.rollback()
else:
# 书籍不存在提示错误
flash('作者没有找到')
return redirect(url_for('index'))
#删除书籍 - -->网页中删除 - -->点击需要发送书籍的ID给删除书籍的路由 - -->路由需要接收参数
#传入书籍ID
@app.route('/delete_book/<book_id>')
def delete_book(book_id):
#查询是否存在该ID的书
book= Book.query.get(book_id)
#如果有就删除
if book:
try:
db.session.delete(book)
db.session.commit()
except Exception as e:
print(e)
flash('删除书籍失败')
db.session.rollback()
else:
#书籍不存在提示错误
flash('书籍没有找到')
#url_for传入视图函数名返回该视图函数对应的路由地址
return redirect(url_for('index'))
@app.route('/',methods=['POST','GET'])
def index():
author_form = AuthorForm()
#调用WTF函数实现验证
if author_form.validate_on_submit():
#验证通过获取数据
author_name = author_form.author.data
book_name = author_form.book.data
#判断作者是否存在
author = Author.query.filter_by(name = author_name).first()
#如果作者存在
if author:
#判断书籍是否存在
book = Book.query.filter_by(name = book_name).first()
#有就提示重复
if book:
flash('已存在同名书籍')
#没有重复书籍就添加数据
else:
try:
new_book = Book(name=book_name,author_id=author.id)
db.session.add(new_book)
db.session.commit()
except Exception as e:
print(e)
flash('添加书籍失败')
db.session.rollback()
#如果作者不存在添加作者书籍
else:
try:
new_author = Author(name=author_name)
db.session.add(new_author)
db.session.commit()
new_book = Book(name=book_name,author_id=new_author.id)
db.session.add(new_book)
db.session.commit()
except Exception as e:
print(e)
flash('添加作者和书籍失败')
db.session.rollback()
#验证不通过就提示错误
else:
if request.method == 'POST':
flash('参数不全')
authors = Author.query.all()
return render_template('books.html',authors=authors,form = author_form)
if __name__ == '__main__':
db.drop_all()
db.create_all()
#生成数据
au1 = Author(name='老王')
au2 = Author(name='老宋')
au3 = Author(name='老刘')
db.session.add_all([au1,au2,au3])
db.session.commit()
bk1 = Book(name='老王回忆录',author_id=au1.id)
bk2 = Book(name='我读书少你别骗我',author_id=au1.id)
bk3 = Book(name='如何征服美丽少女',author_id=au2.id)
bk4 = Book(name='如何征服美丽少男',author_id=au2.id)
bk5 = Book(name='如何使自己变得更骚',author_id=au3.id)
bk6 = Book(name='我变秃了,也变强了',author_id=au3.id)
db.session.add_all([bk1, bk2, bk3,bk4,bk5,bk6])
db.session.commit()
app.run(debug=True)
模板代码
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<title>Title</title>
</head>
<body>
<form method="post">
{{ form.csrf_token() }}
{{ form.author.label }}{{ form.author }}<br>
{{ form.book.label }}{{ form.book }}<br>
{{ form.submit }}<br>
{% for message in get_flashed_messages() %}
{{ message }}
{% endfor %}
</form>
<hr>
<ul>
{% for author in authors %}
<li>{{ author.name }}<a href="{{ url_for("delete_author",author_id = author.id) }}">删除</a></li>
<ul>
{% for book in author.books %}
<li>{{ book.name }}<a href="{{ url_for("delete_book",book_id = book.id) }}">删除</a></li>
{% else %}
<li>无</li>
{% endfor %}
</ul>
{% endfor %}
</ul>
</body>
</html>