黑马的免费视频课关于入门flask的我看完了,最后第五章实现了一个小案例,基本涵盖了之前学到的所有知识。跟着敲了一遍,觉得还不错,因此把代码在这里分享一下。重要地方也都做了记录。
前端截图如下:
(竟然支持qq截图直接复制,csdn真的是越来越优秀了,点赞!!!)
from flask import Flask
from flask import flash, request, redirect, url_for
from flask import render_template
from flask_sqlalchemy import SQLAlchemy
from flask_wtf import FlaskForm
from wtforms import StringField, SubmitField
from wtforms.validators import DataRequired
app = Flask(__name__)
'''
1.配置数据库
2.添加书和作者的模型
3.添加数据
4,使用模板显示数据库查询的数据
5.使用WTF显示表单
6.实现相关的增删逻辑
'''
#配置数据库的地址
app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql://root:root@localhost:3306/books'
#跟踪数据库的修改,不建议开启
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
app.secret_key='xck'
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: {}".format(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 'Books: {} {}'.format(self.name, self.author_id)
class AuthorForm(FlaskForm):
author = StringField(u'作者', validators=[DataRequired()])
book = StringField(u'书籍', validators=[DataRequired()])
submit = SubmitField(u'提交')
db.drop_all()
db.create_all()
#生成数据
au1 = Author(name=u"老王")
au2 = Author(name=u"老惠")
au3 = Author(name=u"老刘")
db.session.add_all([au1, au2, au3])
db.session.commit()
bk1 = Book(name=u"老王回忆录", author_id=au1.id)
bk2 = Book(name=u"我读书少,你别骗我", author_id=au1.id)
bk3 = Book(name=u"如何才能让自己更骚", author_id=au2.id)
bk4 = Book(name=u"怎么样才能征服美丽少女", author_id=au3.id)
bk5 = Book(name=u"如何征服英俊少男", author_id=au3.id)
db.session.add_all([bk1, bk2, bk3, bk4, bk5])
db.session.commit()
@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(u"删除作者失败!")
db.session.rollback()
else:
flash(u"作者找不到!")
return redirect(url_for('index'))
#删除书籍-》网页中删除-》点击需要发送书籍的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(u"删除书籍失败!")
db.session.rollback()
else:
flash(u"书籍找不到!")
#如何返回当前网址,---》重定向
#redirect:重定向,需要传入网址或者路由地址
#url_for('index'):需要传入视图函数名,返回该视图函数对应的路由地址
return redirect(url_for('index'))
@app.route('/', methods=['POST', 'GET'])
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__':
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 }}
{# 显示消息显示的内容 #}
{% for message in get_flashed_messages() %}
{{ message }}
{% endfor %}
</form>
<hr>
{% for author in authors %}
<li>{{ author }}<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 %}
</body>
</html>
PS:这个视频课程只能算是入门,可以距我自己能写项目的程度还要远远不够,因此我准备继续学习下一门有关flask的项目,学一个案例实战,争取拿下这个框架!