一 项目,目录结构
[root@kolla ~]# tree Flask_books_project/
Flask_books_project/
├── Flask_books_project.py
├── static
│ └── __init__.py
└── templates
└── books.html
2 directories, 3 files
[root@kolla ~]#
二 books.html文件内容
<!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>
三 Flask_books_project.py文件内容
#1.先创建数据库
# create database flask_books charset=utf8;
# -*- coding:utf-8 -*-
from flask import Flask, render_template, flash, request
from flask_sqlalchemy import SQLAlchemy
from flask_wtf import FlaskForm
from wtforms import StringField, SubmitField
from wtforms.validators import DataRequired
#重定向会引入这些
from flask import redirect,url_for
# 解决编码问题
import sys
import importlib
importlib.reload(sys)
app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql://root:[email protected]/flask_books'
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
app.secret_key = 'itheima'
#创建数据库对象
#数据库创建命令
# create database flask_books charset=utf8;
#
db = SQLAlchemy(app)
'''
1. 配置数据库
a. 导入SQLAchemy扩展
b. 创建db对象,并配置参数
c. 终端创建数据库
2. 添加书和作者模型
a.模型继承db.Model
b.__tablename__表名
c. db.Column:字段
d. db.relationship:关系引用
3. 添加数据
4. 使用模板显示数据库查询的数据
a. 查询所有的作者信息,让信息传递给模板
b. 模板中按照格式,一次for循环作者和书籍即可(作者获取书籍,用的是关系引用)
5. 使用WTF显示表单
a. 自定义表单类
b. 模板中显示
c. secret_key /编码 / csrf_token
6. 实现增删逻辑
a. 增加数据
b. 删除书籍 url_for的使用 / for else的使用/redirect的使用
'''
#定义书和作者模型
#作者模型
class Author(db.Model):
#表明
__tablename__= 'authors'
#字段
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(16), unique = True)
#关系引用
#books是给自己(Author)用的,
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):
# 查询数据库,是否有该作者,如果有就删除(先删除书籍再删除作者),没有就提示错误
# 1.查询数据库
author = Author.query.get(author_id)
# 2. 如果有就删除(先删书,再删作者)
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:
#3. 没有提示错误
flash('作者找不到')
return redirect(url_for('index'))
#b. 删除书籍--> 网页中删除-->点击需要发送书籍的ID给删除书籍的路由-->路由需要接受参数
@app.route('/delete_book/<book_id>')
def delete_book(book_id):
#1 查询数据库,是否有该ID,如果有就删除,没有就提示错误
book = Book.query.get(book_id)
#2 如果有就删除
if book:
try:
db.session.delete(book)
db.session.commit()
except Exception as e:
print(e)
flash('删除书籍出错')
db.session.rollback()
else:
# 3.没有提示错误
flash('书籍找不到')
#如何返回当前网址-->重定向
#return redirect('www.baidu.com')
# return redirect('/')
#redirect:重定向,需要传入一个网址/路由地址
#url_for('index'):需要传入视图函数名,返回改视图函数对应的路由地址
print(url_for('index'))
return redirect(url_for('index')) #index为函数
@app.route('/',methods=['GET','POST'])
def index():
author_form = AuthorForm()
'''
验证逻辑:
1. 调用WTF的函数实现验证
2. 验证通过获取我们的数据
3. 判断作者是否存在
4. 如果作者存在,查看书是否存在,没有重复书籍就添加数据,如果重复就提示错误
5. 如果作者不存在,添加作者和书籍
6. 验证不通过则提示错误
'''
#1. 调用WTF的函数实现验证
if author_form.validate_on_submit():
#2. 验证通过获取数据
author_name = author_form.author.data
book_name = author_form.book.data
# 3. 判断作者是否存在
author = Author.query.filter_by(name=author_name).first()
# 4. 如果作者存在
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:
# 5. 如果作者不存在,添加作者和书籍
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:
# 6. 验证不通过则提示错误
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=au3.id)
bk5 = Book(name='如何征服英俊少男', author_id=au3.id)
#把数据提交给用户会话
db.session.add_all([bk1, bk2, bk3, bk4, bk5])
db.session.commit()
app.run(debug=True)