#
后端快速上手

接下来,我们将开始一个简单的图书项目,来帮你熟悉整个项目的开发流程。

# 视图控制

打开 app/api/v1 文件夹,在该文件夹里新建 book.py 文件。我们从 Lin 从导入红图来创建 API 视图:

from lin.redprint import Redprint

而后,初始化一个名为 book_api 的红图,并创建一个视图函数get_book

WARNING

请记住此处视图函数所代表的意思,在后续的章节中它会非常重要!!!

from lin.redprint import Redprint
from flask import jsonify

book_api = Redprint('book')


@book_api.route('/<id>', methods=['GET'])
def get_book(id):
    return jsonify({
        'msg': 'hello, I am a book'
    })

如果你熟悉 Flask,应该可以发现这与 Flask 的 API 开发几乎一致,不过我们提供了一个红图的机制,让 API 的开发更加细粒度化。

到此一个简单的图书 API 开发就实现了,但是我们此时运行程序并不能访问到该 API。我们还需要将该红图挂载到项目的默认 API 蓝图上。

打开 app/api/v1/__init__.py 文件,向其中添加如下内容:

from flask import Blueprint
from app.api.v1 import book


def create_blueprint_v1():
    bp_v1 = Blueprint('v1', __name__)
    book.book_api.register(bp_v1)
    return bp_v1

现在,我们已经可以通过 HTTP 请求到该 API 了,运行程序。然后通过 curl(你可以使用任何你喜爱的测试工具,postman 甚至浏览器都行) 访问 http://127.0.0.1:5000/v1/book/1

如果一切顺利,你就可以在 terminal 中看到下面返回数据:

{
  "msg": "hello, I am a book"
}

# 模型使用

打开 app/models 文件夹,在该文件夹下新建 book.py 文件,并加入以下内容:

from sqlalchemy import Column, String, Integer

from lin.interface import InfoCrud as Base


class Book(Base):
    id = Column(Integer, primary_key=True, autoincrement=True)
    title = Column(String(50), nullable=False)
    author = Column(String(30), default='未名')
    summary = Column(String(1000))
    image = Column(String(50))

我们新建了一个 Book 的 Model,该模型继承自 InfoCrud,有关于 InfoCrud 的详细,将在后续详细介绍,本章我们注重于 CMS 的开发使用。

有了该模型后,我们打开根目录下的 fake.py 文件,并运行 fake.py 文件,如果你的操作正确,你的数据库将会在多出两条书籍的数据。

下面,我们继续完善 app/api/v1/book.py 下的 get_book 函数,在上一步中,我们只是简单的返回了一条信息,但是在真正的开发中,数据库几乎是任何应用都摆脱不了的一环。

我们重写 get_book 函数:

from lin.redprint import Redprint
from flask import jsonify
from lin.exception import NotFound
from app.models.book import Book

book_api = Redprint('book')


@book_api.route('/<id>', methods=['GET'])
def get_book(id):
    book = Book.query.filter_by(id=id).first() # 通过Book模型在数据库中查询id=`id`的书籍
    if book is None:
        raise NotFound(msg='没有找到相关书籍') # 如果书籍不存在,返回一个异常给前端
    return jsonify(book) # 如果存在,返回该数据的信息

在上面重写的 book.py 文件中,我们使用到了 flask-sqlalchemy 提供的便捷查询 API,详细使用请参考官网

至此,我们再次运行根目录下的 starter.py 文件,并通过 curl 再次访问http://127.0.0.1:5000/v1/book/1,我们会得到如下数据:

{
  "author": "Randal E.Bryant",
  "create_time": 1539702050000,
  "id": 1,
  "image": "https://img3.doubanio.com/lpic/s1470003.jpg",
  "summary": "********"
}

# 参数校验

在应用的开发中,尤其是在 Web 领域中,对于任何从用户传入的数据都是无规则可循的,我们无法去预测用户传入的数据,因此参数(数据)校验是开发中不可或缺的一环。

我们打开 app/validators/forms.py 文件,向其中添加如下内容:

# ********
# ********

class BookSearchForm(Form):
    q = StringField(validators=[DataRequired(message='必须传入搜索关键字')]) # 前端的请求参数中必须携带`q`

我们新建了一个 BookSearchForm 的校验类,当前端传入数据时必须携带 q 这个参数,它被用来查询书籍。

然后,我们在 app/api/v1/book.py 文件中新增一个视图函数 search:

from app.validators.forms import BookSearchForm
# *******
@book_api.route('/search', methods=['GET'])
def search():
    form = BookSearchForm().validate_for_api() # 对前端的参数进行校验
    q = '%' + form.q.data + '%' # 取出参数中的`q`参数,mysql的特性,加`%`进行模糊查询
    books = Book.query.filter(Book.title.like(q)).all() # 搜索书籍标题
    if books is None or len(books) < 1:
        raise NotFound(msg='没有找到相关书籍')
    return jsonify(books) # 返回书籍

最后,我们再次运行根目录下的 starter.py 文件,并通过 curl 再次访问 http://127.0.0.1:5000/v1/book/search,我们会得到如下数据:

{
    "error_code": 10030,
    "msg": {"q": ["必须传入搜索关键字"]},
    "request": "GET  /v1/book/search"
}

很明显我们未输入参数 q,因此校验未通过。我们修改 url 为 http://127.0.0.1:5000/v1/book/search?q=C,会得到如下数据:

[
  {
    "author": "(美)Brian W. Kernighan",
    "create_time": 1539702050000,
    "id": 2,
    "image": "https://img3.doubanio.com/lpic/s1106934.jpg",
    "summary": "在计算机发展的历史上,没有哪一种程序设计语言像C语言这样应用广泛。本书原著即为C语言的设计者之一Dennis M.Ritchie和著名计算机科学家Brian W.Kernighan合著的一本介绍C语言的权威经典著作。我们现在见到的大量论述C语言程序设计的教材和专著均以此书为蓝本。原著第1版中介绍的C语言成为后来广泛使用的C语言版本——标准C的基础。人们熟知的“hello,World\"程序就是由本书首次引入的,现在,这一程序已经成为众多程序设计语言入门的第一课。\n原著第2版根据1987年制定的ANSIC标准做了适当的修订.引入了最新的语言形式,并增加了新的示例,通过简洁的描述、典型的示例,作者全面、系统、准确地讲述了C语言的各个特性以及程序设计的基本方法。对于计算机从业人员来说,《C程序设计语言》是一本必读的程序设计语 言方面的参考书。",
    "title": "C程序设计语言"
  }
]

# 自定义异常

在刚才的图书搜索 API 中,当程序没有找到相关的书籍时,会抛出一个 NotFound 的异常。NotFound 是 Lin 基础库提供的一种异常,现在有如下需求:我们需要自定义一个 BookNotFound 的异常来专门表示图书未找到。打开 app/libs/error_code.py,在其中添加如下内容:

class BookNotFound(APIException):
    code = 404 # http状态码
    msg = '没有找到相关图书' # 异常信息
    error_code = 80010 # 约定的异常码

WARNING

BookNotFound异常继承自APIException,所有继承自APIException的子类异常均会被框架的异常处理机制所捕捉到,并以固定的信息结构返回给前端,方便前端快速捕捉错误。 在codemsgerror_code中,msg 和 error_code 异常重要,msg 能直接的告诉前端错误信息,error_code 可以让前端快速判断是何种异常。

然后更改图书搜索未找到图书的异常:

from app.libs.error_code import BookNotFound

# *******

@book_api.route('/search', methods=['GET'])
def search():
    form = BookSearchForm().validate_for_api()
    q = '%' + form.q.data + '%'
    books = Book.query.filter(Book.title.like(q)).all()
    if books is None or len(books) < 1:
        raise BookNotFound()
    return jsonify(books)

再次运行程序,如果未找到图书,则前端会得到如下提示信息:

{
    "error_code": 80010,
    "msg": '没有找到相关图书',
    "request": "GET  /v1/book/search"
}

# 完善 API

上面,我们陆续地完成了查询图书和搜索图书两个 API,此处我们再补上创建、更新、删除和获取图书这四个 API。

  • 创建图书
# book.py
from lin import db
from lin.redprint import Redprint
from flask import jsonify
from lin.exception import NotFound, ParameterException, Success

from app.libs.error_code import BookNotFound
from app.models.book import Book
from app.validators.forms import BookSearchForm, CreateOrUpdateBookForm

# ************

@book_api.route('/', methods=['POST'])
def create_book():
    form = CreateOrUpdateBookForm().validate_for_api()  # 校验参数
    book = Book.query.filter_by(title=form.title.data).first()  # 避免同名图书
    if book is not None:
        raise ParameterException(msg='图书已存在')
    # 新增图书
    with db.auto_commit():
        new_book = Book()
        new_book.title = form.title.data
        new_book.author = form.author.data
        new_book.summary = form.summary.data
        new_book.image = form.image.data
        db.session.add(new_book)
    return Success(msg='新建图书成功')
  • 更新图书
@book_api.route('/<id>', methods=['PUT'])
def update_book(id):
    form = CreateOrUpdateBookForm().validate_for_api()  # 校验参数
    book = Book.query.filter_by(id=id).first()  # 通过Book模型在数据库中查询id=`id`的书籍
    if book is None:
        raise NotFound(msg='没有找到相关书籍')  # 如果书籍不存在,返回一个异常给前端
    # 更新图书
    with db.auto_commit():
        book.title = form.title.data
        book.author = form.author.data
        book.summary = form.summary.data
        book.image = form.image.data
    return Success(msg='更新图书成功')
  • 删除图书
@book_api.route('/<id>', methods=['DELETE'])
def delete_book(id):
    book = Book.query.filter_by(id=id).first()  # 通过Book模型在数据库中查询id=`id`的书籍
    if book is None:
        raise NotFound(msg='没有找到相关书籍')  # 如果书籍不存在,返回一个异常给前端
    # 删除图书,软删除
    book.delete(commit=True)
    return Success(msg='删除图书成功')
  • 获取所有图书
@book_api.route('/', methods=['GET'])
def get_books():
    books = Book.query.filter_by().all()
    if books is None or len(books) < 1:
        raise NotFound(msg='没有找到相关书籍')
    return jsonify(books)

我们完成了一个关于图书 CRUD 的简单 demo,事实上很多应用都是重复如此操作,希望你能熟练掌握这套流程。

详细代码参考示例项目book

# 小结

到此,我们运行了示例的工程项目,并通过 starter 完成了一个简单的图书 API 的开发,我们使用了如下几点:

  • 使用红图细粒度的创建 API
url description method
/v1/book/:id 查询指定 id 的书籍 GET
/v1/book/:id 更新指定 id 的书籍 PUT
/v1/book/:id 删除指定 id 的书籍 DELETE
/v1/book/search 根据关键字q搜索书籍 GET
/v1/book/ 创建图书 POST
/v1/book/ 获取所有图书 GET
  • 使用 Model 层创建和查询数据
  • 使用 WTForms 来校验参数
  • 继承 APIException 来自定义异常

当然在本章中,我们只是一个快速上手的指导,如果你熟悉 Flask 及其生态,那么你完全可以按照此种模式开发自己的应用,后续我们会讨论项目开发的基础约定和流程。

后端开发完成!移步前端部分?