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 SubmitField, StringField
from wtforms.validators import DataRequired


app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql+pymysql://root:[email protected]/books'
# 自动跟踪修改
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
app.secret_key = 'like'
# 创建的数据库对象
db = SQLAlchemy(app)
"""
1, 配置数据库
    a, 导入SQLAlchemy b, 创建db对象, 并配置参数, c, 终端创建数据库
2,添加书和作者的数据
  a,模型继承db.Model 
  b, __tablename__ 
  c, db.Column
  d,# 关系引用. Book是给自己(Author模型 )用的, author 是给Book模型用的
    books = db.relationship('Book', backref='author')
3,添加数据
4,使用模板显示数据查询的数据
    a, 查询所有的作者信息,让信息传递给模板
    b, 模板中按照格式,依次for 循环作者的书记即可(作者获取书籍,用的是关系引用)
5, 使用WTF显示表单
    a, 自定义表单类
    b, 模板中显示
    c, seret_key /编码  / srcf_token
6,使用象关的增删该查
    a. 增加书籍
    b, 删除书籍 --->网页中删除-->点击需要发送书籍的id 给删除书籍的路由-->路由需要接受参数
    url_for 的使用  for else 的使用  redirect 的使用
    c, 删除作者 
"""
# 定义书和作者模型
class Author(db.Model):
    __tablename__ = 'authors'
    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
    name = db.Column(db.String(32), unique=True)
    # 关系引用. Book是给自己(Author模型 )用的, 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, autoincrement=True)
    name = db.Column(db.String(32), unique=True)
    author_id = db.Column(db.Integer, db.ForeignKey(Author.id))
    def __repr__(self):
        return 'Book: %s %s' % (self.name, self.author_id)


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


# 删除作者
@app.route('/delete_author/')
def delete_author(author_id):
    # 1,查询数据库, 是否有该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:
            flash('删除作者出错')
            db.session.rollback()
    else:
        # 3, 没有就提示错误
         flash('作者找不到')


    return redirect(url_for('index'))




@app.route('/delete_book/')
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('书籍找不到')




    # 如何返回当前网址-->重定向 , 目的时返回跟路由
    # return redirect('/)
    # redirect('重定向, 需要传入路由')
    # url_for(index)  需要传入视图函数名, 返回该视图函数对那个的路由地址
    return redirect(url_for('index'))


@app.route('/', methods=['POST', 'GET'])
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('已存在同名书籍')
             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='老zhang')
    au4 = Author(name='老da')
    db.session.add_all([au1, au2, au3, au4])
    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=au2.id)
    bk5 = Book(name='红楼', author_id=au3.id)
    bk6 = Book(name='浪d', author_id=au3.id)
    bk7 = Book(name='浪c', author_id=au1.id)
    bk8 = Book(name='浪f', author_id=au1.id)
    db.session.add_all([bk1, bk2, bk3, bk4, bk5, bk6, bk7, bk8])
    db.session.commit()

    app.run(debug=True)

htmL

   
   

    {{ form.csrf_token() }}

    {{ form.author.label }}{{ form.author }}

    {{ form.book.label}}{{ form.book }}

    {{ form.submit }}

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

---

{#先便利作者,然后在作者里便利书籍#}

    {% for author in authors %}
   
• {{ author.name }}删除作者

       

        {% for book in author.books %}
           
• {{ book.name }}删除

            {% else %}
               
• 无

        {% endfor %}


       

    {% endfor %}