MyBooks.py

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://root:mysql2020>@127.0.0.1/flask_books'
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
# 创建数据库对象
db = SQLAlchemy(app)

app.secret_key = 'mrshi'
'''
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.删除书籍-->网页中删除-->点击需要发送书籍的ID给删除书籍的路由-->路由需要接受参数
      注：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是给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)


# 自定义表单类
class AuthorFrom(FlaskForm):
    author = StringField(u'作者', validators=[DataRequired()])
    book = StringField(u'书籍', validators=[DataRequired()])
    submit = SubmitField(u'提交')


# 删除作者
@app.route('/deleteAuthor/<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('删除作者错误')
            db.session.rollback()
    else:
        flash('找不到作者')
    return redirect(url_for('index'))


# 删除书籍  --> 网页中删除-->点击需要发送书籍的ID给删除书籍的路由-->路由需要接受
@app.route('/deleteBook/<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('书籍找不到')
    # 如何返回当前的网址-->重定向
    # return redirect('www.baidu.com')
    # return redirect('/')

    # redirect:重定向，需要传入网络/路由地址
    # url_for('index'):需要传入视图函数名，返回视图函数对应的路由地址
    return redirect(url_for('index'))


@app.route('/', methods=['GET', 'POST'])
def index():
    # 创建自定义表单类
    author_form = AuthorFrom()

    '''
    验证逻辑
    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(u'该书已存在')
            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(u'添加书籍失败')
                    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(u'参数不全！')

    # 查询所有的额作者信息，将信息交给模板
    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()