Flask ,Mysql,的简单图书管理列表页面

主要知识点:Wtforms的使用,ORM模式的数据库连接使用,
{% for i in list %}

{%endfor%}遍历

面向新手的一个简单的demo:
直接拿的话记得修改数据库的连接配置,以及import相关包即可使用
app.py

from flask import Flask, render_template, flash, request, redirect, url_for
from flask_sqlalchemy import SQLAlchemy
from sqlalchemy.orm import relationship
from flask_wtf import FlaskForm
from wtforms import StringField, SubmitField
from wtforms.validators import DataRequired

app = Flask(__name__)

app.config['SQLALCHEMY_DATABASE_URI'] = "mysql+pymysql://root:123456@localhost/first_flask"
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
app.config['SQLALCHEMY_COMMIT_TEARDOWN'] = True
app.config['SQLALCHEMY_COMMIT_ON_TEARDOWN'] = True
app.secret_key = 'iii'

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给自己用 ,author是给Book模型用
    books = 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)

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


@app.route('/delete_author/')
def delete_author(author_id):
    # 1.查询数据库
    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/')
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('书籍找不到')
    # 如何返回当前网址 -->重定向
    # redirect:重定向,需要传入网络/路由地址
    return redirect(url_for('index'))


@app.route('/', methods=['GET', 'POST'])
def index():
    # 查询所有的作者信息,让信息传递给模板
    # 创建自定义表单类
    """
       验证逻辑:
       1.调用TF的函数实现验证
       2.验证通过获取数据
       3.判断作者是否存在
       4.如果作者存在,判断书籍是否存在,没有重复书籍就添加数据,如果重复就提示错误
       5.如果作者不存在,添加作者和书籍
       6.验证不通过就提示错误
   """
    author_form = AuthorForm()
    if author_form.validate_on_submit():
        author_name = author_form.author.data
        book_name = author_form.book.data
        author = Author.query.filter_by(name=author_name).first()
        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:
            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:
        if request.method == 'POST':
            flash('参数不全')

    author_form = AuthorForm()
    authors = Author.query.all()
    return render_template('books.html', authors=authors, form=author_form)



au1 = Author(name='小王')
au2 = Author(name='jack')
au3 = Author(name='mike')
db.session.add_all([au1, au2, au3])
db.session.commit()
bk1 = Book(name='python', author_id=au1.id)
bk2 = Book(name='java', author_id=au1.id)
bk3 = Book(name='c++', author_id=au2.id)
bk4 = Book(name='c#', author_id=au3.id)
bk5 = Book(name='ruby', author_id=au3.id)
db.session.add_all([bk1, bk2, bk3, bk4, bk5])
db.session.commit()
if __name__ == '__main__':
    app.run(debug=True)

books.html

<!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 }}<br>
{#    显示消息闪现#}
    {% for message in get_flashed_messages() %}
        {{ message }}
    {% endfor %}
</form>
<hr>
<ul>
    {% for author in authors %}
        <li>{{ author.name }}<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 %}
</ul>
</body>
</html>

Flask ,Mysql,的简单图书管理列表页面_第1张图片

你可能感兴趣的:(Flask ,Mysql,的简单图书管理列表页面)