HandlerBase.py 2.8 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182
  1. import simplejson as json
  2. import traceback
  3. from datetime import date
  4. from concurrent.futures import ThreadPoolExecutor
  5. from tornado.web import RequestHandler
  6. from model.log import logger
  7. from model.common import errors
  8. log = logger()
  9. class BaseHandler(RequestHandler):
  10. def __init__(self, application, request, **kwargs):
  11. RequestHandler.__init__(self, application, request, **kwargs)
  12. self._status_code = 200
  13. self.executor = ThreadPoolExecutor(200)
  14. self.set_default_headers()
  15. def options(self):
  16. # 返回方法1
  17. self.set_status(200)
  18. self.finish()
  19. def set_default_headers(self):
  20. super().set_default_headers()
  21. # 设置允许的请求头
  22. self.set_header("Access-Control-Allow-Methods", "GET,POST,PUT,DELETE,OPTIONS")
  23. self.set_header("X-XSS-Protecion", "1")
  24. self.set_header("Content-Security-Policy", "default-src 'self'")
  25. self.set_header("Access-Control-Allow-Credentials", "true")
  26. # 设置一些自己定义的请求头
  27. self.set_header("Access-Control-Allow-Headers",
  28. "Content-Type, Depth, User-Agent, Token, Origin, X-Requested-With, Accept, Authorization")
  29. self.set_header("Content-Type", "application/json; charset=UTF-8")
  30. self.set_header("Access-Control-Allow-Origin","*")
  31. def write_json(self, data, status_code=200, msg='success'):
  32. self.write(json.dumps({'status': {'msg': msg, "RetCode": status_code}, 'data': data}))
  33. def get_args(self):
  34. di=json.loads(self.request.body.decode())
  35. if isinstance(di,str):
  36. di=json.loads(di)
  37. return di
  38. def write_error(self, status_code, msg=None, **kwargs):
  39. if self.settings.get("serve_traceback") and "exc_info" in kwargs:
  40. # in debug mode, try to send a traceback
  41. lines = []
  42. for line in traceback.format_exception(*kwargs["exc_info"]):
  43. lines.append(line)
  44. self.write_json(dict(traceback=''.join(lines)), status_code, self._reason)
  45. elif msg:
  46. self.write_json(None, status_code, msg)
  47. else:
  48. self.write_json(None, status_code, self._reason)
  49. def _authentication(self):
  50. """
  51. :return: True, 认证通过, False 认证不通过
  52. """
  53. return True
  54. log.info("author %s" % self.request.headers)
  55. # log.info(self.request.remote_ip)
  56. if self.request.headers.get("Gip_real") == '183.129.168.74':
  57. return True
  58. if not self.request.headers.get("Authorization"):
  59. return False
  60. else:
  61. # redis 中判断值是否存在
  62. # ur = UserRedisComm()
  63. # key = "admin_account_check%s" % (self.request.headers.get("Authorization"))
  64. # return True if ur.r.get(key) else False
  65. return True