这周学的内容是flask，项目需要在原来的基础上增加一个新功能，实际上就是在原来的一系列文件上改改就能实现，功能也很简单，连接页面与数据库，并且在页面上实现增删改查等功能。
项目是用flask实现的，所以先学了flask，好在flask是一个相对来说功能比较简单的框架，因此学的时候基础内容比较少，但是还是在理解上遇到一些困难。在这里记录一下flask的原理和案例
# -*- coding:utf-8 -*-
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
# python2中的固定写法，当程序中出现中文时，用来修改默认编码方式为utf-8
import sys
reload(sys)
sys.setdefaultencoding("utf-8")
# 固定写法，实例化一个flask类
app = Flask(__name__)
# 数据库配置: 数据库地址
app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql://root:mysql@127.0.0.1/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. 模板中按照格式, 依次for循环作者和书籍即可 (作者获取书籍, 用的是关系引用)
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模型用的，即可以使用author.books和book.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>的request都使用 delete_author这个函数来执行
# <author_id>捕获url里传过来的参数，放到后续的delete_author里
app.route('/delete_author/<author_id>')
def delete_author(author_id):
# 查询数据库, 是否有该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
# html中有捕获flash闪现的语句
flash('删除作者出错')
db.session.rollback()
else:
# 3. 没有提示错误
flash('作者找不到')
# redirect中是一个重定向的网页，url_for是提取这个通过route绑定了“index”这个函数的网页
return redirect(url_for('index'))
# 删除书籍 --> 网页中删除-->点击需要发送书籍的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('书籍找不到')
# redirect: 重定向, 需要传入网络/路由地址
# url_for('index'): 需要传入视图函数名, 返回改视图函数对应的路由地址
print url_for('index')
return redirect(url_for('index'))
# 如何返回当前网址 --> 重定向
# return redirect('wwww.itheima.com')
# return redirect('/')
@app.route('/', methods=['GET', 'POST'])
def index():
# 创建自定义的表单类
author_form = AuthorForm()
'''
表单填充和显示的流程：
1、打开网页，会向URL发送一个GET请求，因为是GET请求，所以author_form为空，网页上的表单中只显示框架，框架中无值
2、在表单中键入值，发送POST请求，满足条件后，author_form中保存了表单中的值，return 之后又把值传回给页面，页面再渲染上值，保证了键入的值一直存在于表单中，而框架里也可以随时调用。
'''
验证逻辑:
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)
