入参校验

parent 48220f86
...@@ -19,6 +19,7 @@ class Errors: ...@@ -19,6 +19,7 @@ class Errors:
REPEAT_BUY = '9007' REPEAT_BUY = '9007'
PROD_NOPRICE = '9008' PROD_NOPRICE = '9008'
NO_ORDERS = '9009' NO_ORDERS = '9009'
PARAMS_ERROR = '9010'
MSG = { MSG = {
TOKEN_INVALID: 'TOKEN失效', TOKEN_INVALID: 'TOKEN失效',
...@@ -29,5 +30,6 @@ class Errors: ...@@ -29,5 +30,6 @@ class Errors:
INPUT_PARAMS_ERROR: '入参错误', INPUT_PARAMS_ERROR: '入参错误',
REPEAT_BUY: '请勿重复购买商品', REPEAT_BUY: '请勿重复购买商品',
PROD_NOPRICE: '商品未定价', PROD_NOPRICE: '商品未定价',
NO_ORDERS: '未找到订单' NO_ORDERS: '未找到订单',
PARAMS_ERROR: '入参错误'
} }
...@@ -27,9 +27,13 @@ class TopUpOrder(Resource): ...@@ -27,9 +27,13 @@ class TopUpOrder(Resource):
def get(self): def get(self):
""".""" """."""
self.parser.add_argument('pageNumber', type=int, required=True) self.parser.add_argument('pageNumber', type=int, required=False)
self.parser.add_argument('pageSize', type=int, required=True) self.parser.add_argument('pageSize', type=int, required=False)
args = self.parser.parse_args() args = self.parser.parse_args()
if not args.get('pageNumber', ''):
raise CustomFlaskErr(Errors.PARAMS_ERROR)
if not args.get('pageSize', ''):
raise CustomFlaskErr(Errors.PARAMS_ERROR)
args['user_id'] = request.user_id args['user_id'] = request.user_id
data = order_service.TopUpOrderService().get_order(args) data = order_service.TopUpOrderService().get_order(args)
resp = request.return_success resp = request.return_success
...@@ -72,6 +76,10 @@ class ConsumeOrder(Resource): ...@@ -72,6 +76,10 @@ class ConsumeOrder(Resource):
self.parser.add_argument('pageNumber', type=int, required=True) self.parser.add_argument('pageNumber', type=int, required=True)
self.parser.add_argument('pageSize', type=int, required=True) self.parser.add_argument('pageSize', type=int, required=True)
args = self.parser.parse_args() args = self.parser.parse_args()
if not args.get('pageNumber', ''):
raise CustomFlaskErr(Errors.PARAMS_ERROR)
if not args.get('pageSize', ''):
raise CustomFlaskErr(Errors.PARAMS_ERROR)
user_id = request.user_id user_id = request.user_id
args['user_id'] = request.user_id args['user_id'] = request.user_id
data = order_service.CurriculumOrderService().get_order(args) data = order_service.CurriculumOrderService().get_order(args)
...@@ -206,6 +214,10 @@ class AllOrders(Resource): ...@@ -206,6 +214,10 @@ class AllOrders(Resource):
self.parser.add_argument('pageNumber', type=int, required=True) self.parser.add_argument('pageNumber', type=int, required=True)
self.parser.add_argument('pageSize', type=int, required=True) self.parser.add_argument('pageSize', type=int, required=True)
args = self.parser.parse_args() args = self.parser.parse_args()
if not args.get('pageNumber', ''):
raise CustomFlaskErr(Errors.PARAMS_ERROR)
if not args.get('pageSize', ''):
raise CustomFlaskErr(Errors.PARAMS_ERROR)
args['user_id'] = request.user_id args['user_id'] = request.user_id
data = order_service.getAllOrders(args) data = order_service.getAllOrders(args)
resp = request.return_success resp = request.return_success
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment