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

#-*- 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)