当前位置：网站首页>flask_restful中reqparse解析器继承

flask_restful中reqparse解析器继承

2022-07-27 01:36:00 【IT之一小佬】

示例代码1：

from flask import Flask
from flask_restful import reqparse, Api, Resource

app = Flask(__name__)
api = Api(app)

USERS = {
    'row1': {'name': 'aaa', 'rate': [70, 65]},
    'row2': {'name': 'bbb', 'rate': [80, 90, 68]},
    'row3': {'name': 'ccc', 'rate': [90, 80]},
}

parser = reqparse.RequestParser()
parser.add_argument('name', type=str, required=True)  # required=True 必选
parser.add_argument('rate', type=int, help='rate is a number', action='append')  # action='append' 可重复选的值
parser.add_argument('User-Agent', type=str, location='headers')

# # 解析器继承
parser_copy = parser.copy()
parser_copy.add_argument('bar', type=int)
parser_copy.replace_argument('bar', type=str, required=True)
parser_copy.remove_argument('User-Agent')


# 真正处理请求的地方
class UserInfo(Resource):
    def get(self):
        return USERS, 200

    def post(self):
        args = parser.parse_args()
        # args = parser_copy.parse_args()
        user_id = int(max(USERS.keys()).lstrip('row')) + 1
        user_id = 'row%i' % user_id
        USERS[user_id] = {'name': args['name'], 'rate': args['rate']}
        USERS[user_id]['ua'] = args.get('User-Agent')
        USERS[user_id]['bar'] = args.get('bar')
        return USERS[user_id], 201


api.add_resource(UserInfo, '/')

if __name__ == '__main__':
    app.run(debug=True)

运行结果：

示例代码2：【注意看和示例代码1中的区别】

from flask import Flask
from flask_restful import reqparse, Api, Resource

app = Flask(__name__)
api = Api(app)

USERS = {
    'row1': {'name': 'aaa', 'rate': [70, 65]},
    'row2': {'name': 'bbb', 'rate': [80, 90, 68]},
    'row3': {'name': 'ccc', 'rate': [90, 80]},
}

parser = reqparse.RequestParser()
parser.add_argument('name', type=str, required=True)  # required=True 必选
parser.add_argument('rate', type=int, help='rate is a number', action='append')  # action='append' 可重复选的值
parser.add_argument('User-Agent', type=str, location='headers')

# # 解析器继承
parser_copy = parser.copy()
parser_copy.add_argument('bar', type=int)
parser_copy.replace_argument('bar', type=str, required=True)
parser_copy.remove_argument('User-Agent')


# 真正处理请求的地方
class UserInfo(Resource):
    def get(self):
        return USERS, 200

    def post(self):
        # args = parser.parse_args()
        args = parser_copy.parse_args()
        user_id = int(max(USERS.keys()).lstrip('row')) + 1
        user_id = 'row%i' % user_id
        USERS[user_id] = {'name': args['name'], 'rate': args['rate']}
        USERS[user_id]['ua'] = args.get('User-Agent')
        USERS[user_id]['bar'] = args.get('bar')
        return USERS[user_id], 201


api.add_resource(UserInfo, '/')

if __name__ == '__main__':
    app.run(debug=True)

运行结果：