Python-Flask框架之"图书管理系统"项目，附详解源代码及页面效果截图
该图书管理系统要实现的功能如下：
1. 可以通过添加窗口添加书籍或作者，如果要添加的作者和书籍已存在于书架上, 则给出相应的提示；
2. 如果要添加的作者存在，而要添加的书籍书架上没有，则将该书籍添加到该作者栏；
3. 如果要添加的作者和书籍都不存在于书架上，则将书籍和作者一起添加；
4. 每个书籍和作者旁边都有一个删除按钮，点击删除书籍的按钮可以将该书籍删除，若某作者栏的书籍全部删除完毕则显示"无"；
5. 若直接点击删除作者按钮，则可以将该作者和其书籍一起全部删掉。
该系统的实现工具: Python的Flask框架和MySQL数据库。
效果图及源码如下：
Python源代码如下:
# coding=utf-8 from flask import Flask,render_template,request,flash,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. 配置数据库 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 6. 实现相关的增删逻辑 a.添加作者/书籍 b.删除书籍: redirect(重定向)/url_for(指向路由)/for else 的使用. c.删除作者(要先删除该作者的书籍, 再删除该作者) """ # 配置数据库的地址URI , 格式 "数据库类型+数据库驱动名称://用户名:密码@机器地址:端口号/数据库名" , 端口号可以不写. # python3中用的mysql驱动是mysql-connector , 已经不支持python2的MySQLdb驱动. app.config["SQLALCHEMY_DATABASE_URI"] = "mysql+mysqlconnector://root:password@127.0.0.1/books_demo" # 跟踪数据库的修改 --> 不建议开启 , 一是消耗性能 , 二是未来的版本中会移除. app.config["SQLALCHEMY_TRACK_MODIFICATIONS"] = False app.secret_key = "hwhefsewljfejrlesjfl" # 没设置secret_key会有报错提醒 # 将app作为参数传入这个关联工具 , 创建一个两者相关联对象db db = SQLAlchemy(app) # 注意: web框架里面的模型类基本都是要继承自导入的模块中的某个父类 , 这样才会起到关联的作用. class Author(db.Model): """创建作者子类""" __tablename__ = "authors" # 定义表名 # 定义字段 # db.Column表示是一个字段 , db.Integer就代表id这个字段的数据类型是整数 , primary_key代表主键(主关键字) , 是作为表的行的唯一标识. # db.String代表是字符串类型 , 字符串长度定义个n个字节 , unique(唯一的) , unique=True代表这列不允许出现重复的值. id = db.Column(db.Integer,primary_key=True) name = db.Column(db.String(64),unique=True) # string的长度随便写个2的倍数就行了 # 在"一对多"的一中定义author_book属性 , 该属性不会出现在字段中 , 后面的backref="author"是给Book反向引用的 # 由于是"一对多" , 所以"多"的地方用Book参数 , "一"的地方用不加s的实例对象参数author. author_book = db.relationship("Book",backref="author") def __repr__(self): """返回定制消息, 与__str__作用类似""" return "Author: %d %s"%(self.id,self.name) class Book(db.Model): """创建书籍子类""" __tablename__ = "books" id = db.Column(db.Integer,primary_key=True) name = db.Column(db.String(64),unique=True) author_id = db.Column(db.Integer,db.ForeignKey("authors.id")) # 表名.id 来建立外键关联 def __repr__(self): return "Book: %d %s"%(self.id,self.name) class TrueForm(FlaskForm): """表单扩展常用的模型(类)有三种: StringField, PasswordField, SubmitField , 这里只用到两种 然后传入参数并创建出各自的实例对象 , 以供其它地方使用. """ author = StringField("作者",validators=[DataRequired()]) book = StringField("书籍",validators=[DataRequired()]) submit = SubmitField("添加") def make_author_book(): author1 = Author(name="金庸") author2 = Author(name="古龙") author3 = Author(name="鲁迅") author4 = Author(name="巴金") db.session.add_all([author1,author2,author3,author4]) db.session.commit() book1 = Book(name="<<射雕英雄传>>", author_id=author1.id) book2 = Book(name="<<天龙八部>>", author_id=author1.id) book3 = Book(name="<<鹿鼎记>>", author_id=author1.id) book4 = Book(name="<<笑傲江湖>>", author_id=author1.id) book5 = Book(name="<<武林外史>>", author_id=author2.id) book6 = Book(name="<<萧十一郎>>", author_id=author2.id) book7 = Book(name="<<小李飞刀>>", author_id=author2.id) book8 = Book(name="<<狂人日记>>", author_id=author3.id) book9 = Book(name="<<阿Q正传>>", author_id=author3.id) book10 = Book(name="<<家>>", author_id=author4.id) book11 = Book(name="<<春>>", author_id=author4.id) book12 = Book(name="<<秋>>", author_id=author4.id) db.session.add_all([book1,book2,book3,book4,book5,book6, book7,book8,book9,book10,book11,book12]) db.session.commit() @app.route("/",methods=["GET","POST"]) def add_author_book(): true_form = TrueForm() """ 1.调用WTF的函数实现验证 2.验证通过则获取数据 3.判断作者是否存在 4.如果作者存在, 则判断书籍是否存在, 没有重复的书籍就添加数据, 如果重复就提示错误. 5.如果作者不存在, 就添加作者和书籍 6.验证不通过就提示错误. """ # 调用WTF的函数实现验证 if true_form.validate_on_submit(): # 2.验证通过则获取此时填入的数据 author_name = true_form.author.data book_name = true_form.book.data # 3.判断作者是否存在, Author.query.filter_by(name=author_name)是查询, .first()才是拿到数据. author_query = Author.query.filter_by(name=author_name).first() # 4.如果作者存在 if author_query: book_query = Book.query.filter_by(name=book_name).first() # 查询并拿数据 if book_query: flash("您要添加的书籍已存在!") else: try: new_book = Book(name="<<%s>>"%book_name,author_id=author_query.id) db.session.add(new_book) db.session.commit() except Exception as 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="<<%s>>"%book_name, author_id=new_author.id) db.session.add(new_book) db.session.commit() except Exception as e: flash("添加作者和书籍错误!") db.session.rollback() else: # 验证不通过 if request.method == "POST": flash("参数错误!") # 查询所有的作者信息, 让信息传递给模板 all_authors = Author.query.all() return render_template("book_manage.html",all_authors=all_authors,form=true_form) # 网页中删除书籍-->将book_id参数传到路由, 路由再将book_id传入delete_book函数内部使用. # < >尖括号代表路由参数, 路由需要接受参数 @app.route("/delete_book/<book_id>",methods=["GET","POST"]) def delete_book(book_id): # 1.查询书籍并拿数据 book = Book.query.get(book_id) try: db.session.delete(book) db.session.commit() except Exception as e: flash("删除错误!") db.session.rollback() # redirect重定向回到根路径, redirect接收路由地址参数, 或者直接接收网址参数(http://xxxxx.com) # url_for("index"): 需要传入视图函数名, 返回该视图函数对应的路由地址(url) return redirect(url_for("add_author_book")) # 删除作者 @app.route("/delete_author/<author_id>",methods=["GET","POST"]) def delete_author(author_id): # 1.查询作者并拿数据 author = Author.query.get(author_id) try: # 查询书籍并删除, 直接在查询后面跟 .delete()就可以直接将查询到的结果删除掉 Book.query.filter_by(author_id=author.id).delete() db.session.delete(author) db.session.commit() except Exception as e: flash("删除错误!") db.session.rollback() # 回滚 return redirect(url_for("add_author_book")) # 重定向回到根路径 if __name__ == '__main__': # 先删除所有表, 在创建表之前要先删掉表 db.drop_all() # 再创建所有表 db.create_all() make_author_book() app.run(debug=True)
HTML源代码如下:
<!DOCTYPE html> <html lang="en"> <head> <meta charset="UTF-8"> <title>图书管理系统 | by-陈彬</title> </head> <body> 从此处添加书籍: <br> <br> <form method="post"> {{ form.csrf_token() }} {{form.author.label}} {{form.author}}<br> {{form.book.label}} {{form.book}}<br> {{form.submit}}<br> <br> {# 显示消息闪现的内容 #} {% for message in get_flashed_messages() %} {{message}} {%endfor%} </form> <br> <hr> <h1> 书籍目录</h1> <ul> <ul> <ul> {% for author in all_authors %} <li>{{author.name}} <a href="{{url_for("delete_author",author_id=author.id)}}">删除</a></li> <ul> {% for book in author.author_book %} <li>{{book.name}} <a href="{{ url_for("delete_book",book_id=book.id) }}">删除</a></li> {% else %} <li>无</li> {% endfor %} </ul> {% endfor %} </ul> </ul> </ul> </body> </html>
