编写业务逻辑

我们的业务逻辑代码主要在 controllers 目录中，新建一个 todo.py 文件, 核心代码如下 (完整代码参考这里)，代码说明可以参考注释：

  1. # -*- coding: utf-8 -*-
  3. import flask
  4. from flask import request, redirect, flash, render_template, url_for
  5. from application.extensions import db
  6. from application.models import Todo
  8. todo_bp = flask.Blueprint(
  9.     'todo',
  10.     __name__,
  11.     template_folder='../templates'
  12. )
  14. # 主页
  15. @todo_bp.route('/', methods=['GET', 'POST'])
  16. def index():
  17.     todo = Todo.query.order_by('-id')
  18.     _form = request.form
  20.     if request.method == 'POST':
  21.         # 添加任务
  22.         title = _form["title"]
  23.         td = Todo(title=title)
  24.         try:
  25.             td.store_to_db()  # 将数据保存到数据库
  26.             flash("add task successfully!")
  27.             return redirect(url_for('todo.index'))
  28.         except Exception as e:
  29.             print e
  30.             flash("fail to add task!")
  32.     return render_template('index.html', todo=todo)
  34. # 删除任务
  35. @todo_bp.route('/<int:todo_id>/del')
  36. def del_todo(todo_id):
  37.     todo = Todo.query.filter_by(id=todo_id).first()
  38.     if todo:
  39.         todo.delete_todo()
  40.     flash("delete task successfully")
  41.     return redirect(url_for('todo.index'))
  43. # 编辑 (更新) 任务
  44. @todo_bp.route('/<int:todo_id>/edit', methods=['GET', 'POST'])
  45. def edit(todo_id):
  46.     todo = Todo.query.filter_by(id=todo_id).first()
  48.     if request.method == 'POST':
  49.         Todo.query.filter_by(
  50.             id=todo_id
  51.         ).update({
  52.             Todo.title: request.form['title']
  53.         })
  54.         db.session.commit()
  55.         flash("update successfully!")
  56.         return redirect(url_for('todo.index'))
  58.     return render_template('edit.html', todo=todo)
  60. # 标记任务完成
  61. @todo_bp.route('/<int:todo_id>/done')
  62. def done(todo_id):
  63.     todo = Todo.query.filter_by(id=todo_id).first()
  64.     if todo:
  65.         Todo.query.filter_by(id=todo_id).update({Todo.status: True})
  66.         db.session.commit()
  67.         flash("task is completed!")
  69.     return redirect(url_for('todo.index'))
  71. # 重置任务
  72. @todo_bp.route('/<int:todo_id>/redo')
  73. def redo(todo_id):
  74.     todo = Todo.query.filter_by(id=todo_id).first()
  75.     if todo:
  76.         Todo.query.filter_by(id=todo_id).update({Todo.status: False})
  77.         flash("redo successfully!")
  78.         db.session.commit()
  80.     return redirect(url_for('todo.index'))
  82. # 404 处理
  83. @todo_bp.errorhandler(404)
  84. def page_not_found():
  85.     return render_template('404.html'), 404

上面只是核心的业务逻辑，关于模板渲染，数据模型的定义等可以参考完整代码。

由于该项目的业务逻辑比较简单，因此也就不做过多介绍，相信读者可以轻松看懂代码，如果有问题，也可以在 github 上面提 issue，欢迎指正。