使用Flask创建简易的图书管理系统
一、效果演示
二、项目目录结构
三、前端代码
DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<title>Titletitle>
head>
<body>
<form action="" method="post">
{# 可以用标签包裹以下内容，以便于优化前端页面#}
{{ form.csrf_token }}
{{ form.author.label }}{{ form.author }}<br>
{{ form.book.label }}{{ form.book }}<br>
{{ form.submit }}<br>
{# 显示消息闪现的内容 #}
<div style="color: red">
{% for message in get_flashed_messages() %}
{{ message }}
{% endfor %}
div>
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>
四、后端代码
from imp import reload
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.config['SQLALCHEMY_DATABASE_URI'] = 'mysql+pymysql://root:[email protected]:3306/flask_books'
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
# 创建密钥
app.secret_key = 'itheima'
# 创建数据库对象
db = SQLAlchemy(app)
"""
1、配置数据库
a.导入SQLAlchemy扩展
b.创建db对象，并配置参数
c.终端创建数据库
2、添加书和作者模型
a.模型继承db.Model
b.__tablename__:表名
c.db.Column:字段
d.db.relationship:添加引用
3、添加数据
4、使用模板显示数据库查询的数据
a.查询所有作者的信息，让信息传递给模板
b.模板按照格式，依次循环作者和书籍（作者获取书籍，用的是关系引用）
5、使用WTF显示表单
a.自定义表单类
b.模板中显示
c.secret_key/编码/csrf_token
6、实现相关增删逻辑
a.增加数据
b.删除书籍 url_for的使用 / for else的使用 / redirect的使用
c.删除作者
"""
# 定义书和作者模型
class Author(db.Model):
# 表名
__tablename__ = 'authors'
# 字段
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(16), unique=True)
# 关系引用
# books是给自己用（Author模型）用的,author是给Book模型用的
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)
@app.route('/', methods=['GET', 'POST'])
def book():
# 创建自定义表单类
author_form = AuthorForm()
"""
验证逻辑：
1、调用WTF的函数实现验证
2、验证通过获取数据
3、判断作者是否存在
4、如果作者存在，判断书籍是否存在，没有重复书籍就添加数据，如果重复就提示错误
5、如果作者不存在，添加作者和书籍
6、验证不通过就提示错误
"""
# 1、调用WTF的函数实现验证 author_form.validate_on_submit就是post请求成功提交数据，不需要额外验证 if request.method == 'POST
if author_form.validate_on_submit():
# 2、验证通过获取数据
# author_form.author.data可以获取author这个表单的内容
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()的下面还有代码例如添加书籍，如果不回滚就会影响下面代码的执行如果回滚那么下面的代码就可以继续执行
db.session.rollback()
# 5、如果作者不存在，添加作者和书籍
else:
# 注意有作者不存在和存在的区别，作者不存在的话要先添加作者，不然添加书籍的时候需要作者id就会出问题
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('book.html', authors=authors, form=author_form)
# 删除书籍-->网页中删除-->点击需要发送书籍的ID给删除书籍的路由-->路由需要接收参数
@app.route('/delete_book/' )
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.itheima.com')
# return redirect('/')
# redirect:需要传入网络/路由地址
# url_for('index'):需要传入视图函数名（注意是函数名！！！！），返回视图函数对应的路由地址
# print(url_for('book')) #'/'就是根路由
return redirect(url_for('book'))
# 删除作者
@app.route('/delete_author/' )
def delete_author(author_id):
# 1.查询数据库，是否有该ID的作者，如果有就删除(要先删书，再删作者)，没有就提示错误
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('book'))
# 自定义表单类
class AuthorForm(FlaskForm):
# wtforms.validators:存放一些验证函数，用于表单的验证
"""
如：DataRequired', 'data_required', 'Email', 'email', 'EqualTo', 'equal_to',
'IPAddress', 'ip_address', 'InputRequired', 'input_required', 'Length',
'length', 'NumberRange', 'number_range', 'Optional', 'optional',
'Required', 'required', 'Regexp', 'regexp', 'URL', 'url', 'AnyOf',
'any_of', 'NoneOf', 'none_of', 'MacAddress', 'mac_address', 'UUID'
"""
# DataRequired限制字段不能为空
author = StringField('作者', validators=[DataRequired()])
book = StringField('书籍', validators=[DataRequired()])
submit = SubmitField('提交')
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()