api_exception.py 789 B

123456789101112131415161718192021222324252627282930
  1. from flask import request, json
  2. from werkzeug.exceptions import HTTPException
  3. class APIException(HTTPException):
  4. code = 500
  5. msg = 'sorry, we make a mistake'
  6. error = 500
  7. def __init__(self, msg=None, error=None, code=None, header=None):
  8. if code:
  9. self.code = code
  10. if msg:
  11. self.msg = msg
  12. if error:
  13. self.code = error
  14. if error:
  15. self.error = error
  16. super(APIException, self).__init__(msg, None)
  17. def get_body(self, environ=None, scope=None):
  18. body = dict(
  19. msg=self.msg,
  20. code=self.code
  21. )
  22. text = json.dumps(body)
  23. return text
  24. def get_headers(self, environ=None, scope=None):
  25. return [('Content-Type', 'application/json')]