专栏首页开源优测接口测试 | 21 基于flask弄个restful API服务出来

接口测试 | 21 基于flask弄个restful API服务出来

概述

上篇我们很简单的分享了如何基于flask搞一个支持http GET\POST\HEAD\DELETE方法的服务,大家可以根据这个简单的实例进行扩展。

下面我们基于flask来构建一个简单的restful风格的API服务出来,以便大家进一步了解和掌握,说不定哪天你就需要自己去实现一个简单的mock server以便让你的测试更加顺畅。

注意

  1. 实现一个简单的restful api
  2. 简单到就像没有任何封装
  3. 不要问我什么是restful风格

安装

使用以下命令安装flask-restful

pip install flask-restful

示例

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

__author__ = "苦叶子"

from flask import Flask
from flask_restful import reqparse, abort, Api, Resource
import random
import time

# 生成随机字符串

def random_str(lenght):
    # 待选随机数据
    data = "1234567890abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"

    # 用时间来做随机播种
    random.seed(time.time())
    time.sleep(0.3)    

    # 随机选取数据
    sa = []    
    for i in range(lenght):
        sa.append(random.choice(data))

    salt = ''.join(sa)    

    return salt

# 初始化

app = Flask(__name__)
api = Api(app)
# 初始化源数据# 随机生成

USERS = {        "user1": { 
        "username": random_str(10),        
        "password": random_str(16),        
        "token": random_str(32)
    },    
    "user2": { 
        "username": random_str(10),        
        "password": random_str(16),              "token": random_str(32)
    },        "user3": { 
        "username": random_str(10),               "password": random_str(16),        
        "token": random_str(32)
    }
}

# 判断用户id是否存在
def abort_if_user_not_exist(user_id):
    if user_id not in USERS:
        abort(404, message="user {%s} is not exist" % user_id)

parser = reqparse.RequestParser()
parser.add_argument("username", type=str)

# 用户管理
class User(Resource):
    # 获取指定用户信息
    def get(self, user_id):

        abort_if_user_not_exist(user_id)      
  
        return USERS[user_id]    

    # 删除指定用户
    def delete(self, user_id):
        abort_if_user_not_exist(user_id)        

        del USERS[user_id]        

        return "", 204

    # 新增/修改用户
    def put(self, user_id):
        args = parser.parse_args()
        print(args)

        user = {"username": args["username"],            
                "password": random_str(16),                            "token": random_str(32)}

        USERS[user_id] = user        

        return user, 201

    # 新增/修改用户
    def post(self, user_id):
        args = parser.parse_args()
        print(args)
        user = {"username": args["username"],                        "password": random_str(16),                        "token": random_str(32)}

        USERS[user_id] = user        
        
        return user, 201


# 查询所有用户信息
class UserList(Resource):
    def get(self):
        return USERS

# 新增资源
api.add_resource(UserList, "/user")
api.add_resource(User, "/user/<user_id>")

# 主入口程序if __name__ == "__main__":
    app.run(debug=True)

小结

如果把上述USERS数据源替换为数据库,你发现是不是就变成了一个基本的管理系统了?自己去尝试吧,come on 小白

对应的flask代码及jmeter测试代码请参见:

链接: https://pan.baidu.com/s/1pLv7Mo7

密码: iha4

本文分享自微信公众号 - 开源优测(DeepTest),作者:苦叶子

原文出处及转载信息见文内详细说明,如有侵权,请联系 yunjia_community@tencent.com 删除。

原始发表时间:2017-09-07

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

我来说两句

0 条评论
登录 后参与评论

相关文章

  • 如果有一千张表,如何测试?

    在我们日常的测试过程中,大部分的测试人员几乎不会去关注数据库表的结构、字段的属性、主外键关系等测试

    苦叶子
  • Python数据库测试实战教程

    我们在细化成具体的就是,我们需要验证表结构、字段、长度、主键、索引、unique等等属性。

    苦叶子
  • AutoLine源码分析之前端js代码分析

    html模板负责UI显示 - js负责前端逻辑 -python api负责后端逻辑

    苦叶子
  • 数据库MySQL-设计规范

    a、采用26个英文字母(区分大小写)和0-9的自然数(经常不需要)加上下划线’_'组成; b、命名简洁明确(长度不能超过30个字符); c、例如:user,...

    cwl_java
  • Clojure 学习入门(3)—— 数字类型

    一、算术运算 加法+:加法函数(+)接受任意数值类型的参数,返回它们的和;没有参数时返回0。 

    阳光岛主
  • Web应用多账号系统设计及微信扫码登录实现

    Web应用多账号系统设计及微信扫码登录实现 1   前言概述 公司对功能测试,性能测试,安全测试等等都做了比较好的自动化后,急需要一个MIS系统来统一管理这些结...

    用户1170933
  • 如何在PostgreSQL中更新大表

    在Postgres中更新大型表并不像看起来那样简单。如果您的表包含数亿行,您将发现很难及时进行简单的操作,例如添加列或更改列类型。

    蒋老湿
  • Flink SQL 核心概念剖析与编程案例实战

    本次,我们从 0 开始逐步剖析 Flink SQL 的来龙去脉以及核心概念,并附带完整的示例程序,希望对大家有帮助!

    kk大数据
  • springboot (11) mybatis

    IT故事会
  • SpringBoot整合Mybatis

    介绍一下SpringBoot整合mybatis,数据库选用的是mysql。 首先创建数据库 CREATE DATABASE test; 建表以及插入初始数据(s...

    dalaoyang

扫码关注云+社区

领取腾讯云代金券