Flask小项目实战笔记

1.创建数据库时设置charset

CREATE DATABASE dbname DEFAULT CHARSET utf8 COLLATE utf8_general_ci;

2.Flask MYSQL配置时出错

#报错信息
UserWarning: Neither SQLALCHEMY_DATABASE_URI nor SQLALCHEMY_BINDS is set. Defaulting SQLALCHEMY_DATABASE_URI to "sqlite:///:memory:".
  'Neither SQLALCHEMY_DATABASE_URI nor SQLALCHEMY_BINDS is set. '

FSADeprecationWarning: SQLALCHEMY_TRACK_MODIFICATIONS adds significant overhead and will be disabled by default in the future.  Set it to True or False to suppress this warning.
  'SQLALCHEMY_TRACK_MODIFICATIONS adds significant overhead and '

解决方法：
https://blog.csdn.net/zhongqiushen/article/details/79139346
https://segmentfault.com/q/1010000008767533/a-1020000008769064
https://blog.csdn.net/weixin_41085315/article/details/79813858

主要就是：
1. db = SQLAlchemy(app) 要在 app.config[ ] 后
2. 没有密码时候直接写成 root:@127.0.0.1:3306

3.逻辑代码

# B站黑马 Flask 实战小项目

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__)

'''
整体实现思路：
1.配置数据库
2.添加模型
3.添加数据
4.使用模板显示数据库查询的数据
    # a.查询所有的作者信息，让信息传递给模板
    # b.模板中按照格式，依次for循环作者和书籍即可
5.使用WTF显示表单
    # a.自定义表单类
    # b.模板中显示
    # c.secret_key / 编码 / csrf_token
6.使用相关的增删逻辑
    # a.增加数据
    # b.删除书籍 （网页中删除-点击需要发送书籍的ID给删除书籍的路由-路由需要接收参数）
    # c.删除作者 
'''

# 第一步 配置数据库
app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql://root:@localhost:3306/books'
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
app.secret_key = 'itheima'

db = SQLAlchemy(app)

# 第二步 定义书和作者模型

# 1.作者模型

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模型用的)，参数Book表示是在Author模型中与Books模型相关联
    # 因为使用是利用 Author.books / Book.author 这样就是实现了引用
    # 参数： 表名（books）  模型名（Book）  模型名小写（author）
    books = db.relationship('Book',backref='author')

    def __repr__(self):
        return 'Author: %s' % self.name

# 2.书籍模型

class Book(db.Model):

    __tablename__ = 'books'

    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(16),unique=True)
    # 表明外键时，是 表名.id
    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>')
def delete_author(author_id):

    # 1.查询数据库，是否有该ID的作者，如果有就先删除书，再删除作者，没有就提示错误
    author = Book.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.rollbakc()
    else:
        # 3.没有提示错误
        flash('作者找不到')

    # 如何返回当前网址 — 重定向
    # redirect:重定向，需要传入网络/路由地址
    # ulr_for('index') 需要传入视图函数名，返回该试图函数对应的路由地址

    return redirect(url_for('index'))

@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.rollbakc()
    else:
        # 3.没有提示错误
        flash('书籍找不到')

    # 如何返回当前网址 — 重定向
    # redirect:重定向，需要传入网络/路由地址
    # ulr_for('index') 需要传入视图函数名，返回该试图函数对应的路由地址

    return redirect(url_for('index'))

@app.route('/', methods=['GET', 'POST'])
def index():

    # 第五步 使用WTF显示表单
    # 创建自定义的表单类
    author_form = AuthorForm()

    '''
    验证逻辑：
    1. 调用WTF的函数实现验证
    2. 验证通过获取数据
    3. 判断作者是否存在
    4. 如果作者存在，判断书是否存在，没有重复书籍就添加数据，如果重复就提示错误
    5. 如果作者不存在，添加作者和书籍
    6. 验证不通过就提示错误
    '''
    # 1. 调用WTF的函数实现验证
    if author_form.validate_on_submit():

        # 2. 验证通过获取数据(.data拿到对象对应的数据)
        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.rollbakc()


    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()

4.网页代码

<!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 }}
    {{ form.book.label }}{{ form.book }}
    {{ form.submit }}

    {# 显示消息闪现的内容 #}
    {% for message in get_flashed_messages() %}
          {{ message }}
    {% endfor %}


</form>

<hr>

{# 先遍历作者，再在作者里遍历书籍 #}
<ul>

    {% for author in authors %}

        <li>{{ author.name }}</li> <a href="{{ url_for('delete_author', author_id=author.id) }}">删除作者</a>

        <ul>
            {% for book in author.books %}
                {# url_for拼接, 先是根据delete_book找到路由, 然后再拼接book_id参数 #}
                <li>{{ book.name }} <a href="{{ url_for('delete_book', book_id=book.id) }}">删除</a> </li>

            {% else %}
                <li>无</li>
            {% endfor %}
        </ul>

    {% endfor %}

</ul>

</body>
</html>