16 lines
570 B
Python
16 lines
570 B
Python
from flask_restx.reqparse import RequestParser
|
|
from werkzeug.exceptions import BadRequest
|
|
from app_common.const import MALFORMED_DATA, BAD_USER_ROLES
|
|
|
|
|
|
class QuarterRequestParser(RequestParser):
|
|
def parse_args(self, req=None, strict=False):
|
|
try:
|
|
return super().parse_args(req=req, strict=strict)
|
|
except BadRequest as err:
|
|
error_type = MALFORMED_DATA
|
|
if 'roles' in err.data.get('errors', {}):
|
|
error_type = BAD_USER_ROLES
|
|
err.data.update({'error_type': error_type})
|
|
raise err
|