AutoLine源码分析之如何构建restful API

什么是AutoLine开源平台

AutoLine开源平台是一个开源自动化测试解决方案，基于RobotFramework进行二次开发，支持RobotFramework几乎所有的库。

from .api import api_bp as api_blueprint
app.register_blueprint(api_blueprint, url_prefix='/api/v1')

# 导入blueprint及restful
from flask import Blueprint
from flask_restful import Api

# 构建blueprint及restful Api对象
api_bp = Blueprint('api', __name__)
api = Api(api_bp)

# 添加认证restful api资源
from .auth import Auth
api.add_resource(Auth, "/auth/")

# 添加产品管理restful api资源
from .product import Product
api.add_resource(Product, "/product/")

# -*- coding: utf-8 -*-

__author__ = "苦叶子"

"""

公众号: 开源优测

Email: lymking@foxmail.com

"""

from datetime import datetime

from flask_restful import Resource, reqparse
from flask_login import current_user

from ..models import AutoProduct, User
from .. import db

# 继承Resource
class Product(Resource):
    def __init__(self):
        # 定义从前端传入服务端要转换参数
        self.parser = reqparse.RequestParser()
        self.parser.add_argument('name', type=str)
        self.parser.add_argument('desc', type=str)
        self.parser.add_argument('tags', type=str)
        self.parser.add_argument('enable', type=bool, default=True)
        self.parser.add_argument('id', type=int, default=-1)
        self.parser.add_argument('method', type=str)
        self.parser.add_argument('page', type=int, default=1)
        self.parser.add_argument('rows', type=int, default=15)
    
    # get方法的实现
    def get(self):
        args = self.parser.parse_args()

        pass

        return data
    
    # post方法的实现
    # 通过参数method的值来实现增删改查动作
    # 目的是避免再去实现put、delete等方法，统一由post方法实现
    def post(self):
        args = self.parser.parse_args()
        method = args["method"].lower()
        if method == "create":
            return self.__create(args), 201
        elif method == "edit":
            return self.__edit(args), 201
        elif method == "delete":
            return self.__delete(args), 201
        elif method == "query":
            return self.__query(args), 201

        return {"status": "fail", "msg": "方法: %s 不支持" % method}, 201
    
    # 实现创建动作，即新增
    def __create(self, args):
        pass
        
    # 实现查询
    def __query(self, args):
        pass

    # 实现编辑
    def __edit(self, args):
        pass
    
    # 实现删除
    def __delete(self, args):
        pass