from flask import Flask,render_template, flash, request,url_for,redirect
# 导入数据库工具
from flask_sqlalchemy import SQLAlchemy
from flask_wtf import FlaskForm
# 提交
from wtforms import StringField, SubmitField
from wtforms.validators import DataRequired,EqualTo
#编码
# import sys
# sys.setdefaultencoding("utf-8")
'''
1、配置数据库
a.导入SQLAlchemy扩展
b.创建db对象， 并设置参数
c.终端创建数据库
2、添加书和作者模型
a.模型继承 db.Model
b.__tablename__ :表明
c. db.Column :字段
3、添加数据
4、使用模板显示数据库查询的数据
a.查询所有的作者的信息，让信息传递给模板
b.模板中按照格式，依次for循环作者的书籍即可，（作者获取书籍， 用的是关系引用）
5、使用WTF显示表单
a.自定义表单类
b.模板中的显示
c.secret_key /编码 /csrf_token
6、实现相关的增删逻辑
a.增加数据
b.删除书籍 url_for的使用 、 for else的使用 、redirect（重定向）的使用
'''
app = Flask(__name__)
# 配置数据库的地址 数据库
app.config["SQLALCHEMY_DATABASE_URI"] = "mysql+pymysql://root:123456@localhost:3306/flask_books"
# 跟踪数据库的修改 -->不建议开启、 开启以后会性能会有影响、未来的版本会移除
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
#在flask项目中，Session, Cookies以及一些第三方扩展都会用到SECRET_KEY值，这是一个比较重要的配置值。
app.secret_key = "itheima"
#创建数据库对象
db = SQLAlchemy(app)
'''
这个网站解释了 flask框架封装数据库字段的解释 https://blog.csdn.net/appleyuchi/article/details/79372240
'''
class Book(db.Model):
# 表名字
__tablename__ = "books"
# 字段
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(16), unique=True)
# 外键 就是Book类中添加了一个author_id变量，数据类型db.Integer,第二个参数指定外键是哪个表中哪个(authors)的id。
author_id = db.Column(db.Integer,db.ForeignKey("authors.id"))
def __repr__(self):
return "Book: %s %s" % (self.name,self.author_id)
#定义书和作者模型
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")
'''
这句话比较复杂，仔细读下面的话：
0.添加到Author模型中的books属性代表这个关系的面向对象视角。对于一个Author类的实例，其books属性将返回与角色相关联的用户组成的列表。
1.db.Relationship()第一个参数表明这个关系的另一端是哪个模型（类）。如果模型类尚未定义，可使用字符串形式指定。
2.db.Relationship()第二个参数backref，将向Book类中添加一个author属性，从而定义反向关系。这一属性可替代author_id访问Author模型，此时获取的是模型对象，而不是外键的值。
'''
def __repr__(self):
return "Author: %s" % self.name
#自定义表单类
class AuthorForm(FlaskForm):
#validators=[DataRequired()] 验证函数
author = StringField("作者",validators=[DataRequired()])
book = StringField("书籍",validators=[DataRequired()])
submit = SubmitField("提交")
@app.route("/delete_author/<author_id>")
def delete_author(author_id):
#查询数据库,查看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"))
@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("书籍找不到")
print(url_for("index"))
#删除以后重定向到index所对应的视图
return redirect(url_for("index"))
#验证就会出现GET/POST请求
@app.route('/',methods=["GET","POST"])
def index():
# 创建自定义的表单类
author_form = AuthorForm()
'''
6、 WTF可以一句话实现验证数据:
1、调用WTF的函数实现验证
2、验证通过获取数据
3、判断作者是否存在
4、判断书籍是否存在, 如果没有重复书籍就添加数据，如果重复就提示错误
5、如果作者不存在, 添加作者和书籍
6、验证不通过就提示错误
'''
#1、调用WTF的函数实现验证, 提交的时候验证 \ 验证是否有数据
if author_form.validate_on_submit():
#2、验证通过获取数据
#拿到作者input信息
author_name = author_form.author.data
#书籍input信心
book_name = author_form.book.data
#3、判断作者是否存在 (涉及到数据库查询操作、query.filter_by)
author = Author.query.filter_by(name = author_name).first()
#如果作者存在
if author:
#4、判断书籍是否存在, 如果没有重复书籍就添加数据，如果重复就提示错误
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:
#flash("这个无论如何都会提示、因为没有做启动方式验证、启动方式是get或者post")
#6、如果是post请求， 才会提示有问题 验证不通过就提示错误 加上from flask import Flask,render_template, flash, request
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)