from tornado.web import Application as tornadoApp

class Application(tornadoApp):
? ? """ 定制 Tornado Application 集成日志、sqlalchemy 等功能 """
? ? def __init__(self):
? ? ? ? self.ops_mongo = None ?# ops mongo
? ? ? ? self.redis_cluster = None ?# redis基礎(chǔ)session
? ? ? ? self.redis_manager = None ?# redis_manager session
? ? ? ? self.redis = None
? ? ? ? tornado_settings = {
? ? ? ? ? ? 'autoreload': settings.service.server.autoreload,
? ? ? ? ? ? 'debug': settings.service.server.debug,
? ? ? ? ? ? "default_handler_class": DefaultHandler,
? ? ? ? }
? ? ? ? route = getRoutes(settings)
? ? ? ? self.prepare()
? ? ? ? super(Application, self).__init__(handlers=route, **tornado_settings)
? ? def prepare(self):
? ? ? ? self.redis_cluster = asyncio.get_event_loop().run_until_complete(
? ? ? ? ? ? init_cluster()
? ? ? ? )
? ? ? ? self.ops_mongo = asyncio.get_event_loop().run_until_complete(
? ? ? ? ? ? init_ops_mongodb()
? ? ? ? )
? ? ? ? loop = asyncio.get_event_loop()
? ? ? ? self.redis = RedisPool(loop=loop).get_conn()
? ? ? ? self.redis_manager = RedisManager(self.redis)
? ? ? ? asyncio.get_event_loop().run_until_complete(
? ? ? ? ? ? create_all_indexes()
? ? ? ? )
? ? async def on_close(self, server_conn: object) -> None:
? ? ? ? await self.redis_manager.close()
? ? ? ? logger.info("close redis")
? ? ? ? self.ops_mongo.close()
? ? ? ? logger.info("close mongo")

class DefaultHandler(tornado.web.RequestHandler):
? ? def get(self):
? ? ? ? raise tornado.web.HTTPError(
? ? ? ? ? ? status_code=404,
? ? ? ? ? ? reason="Not Found"
? ? ? ? )
? ? def write_error(self, status_code, exc_info=None, **kwargs):
? ? ? ? self.finish({"error": self._reason})
def getRoutes(setting):
? ? Routes = [
? ? ? ? (r"/redis", home.TestRedisHandler),
? ? ]
? ? return Routes

class BaseHandler(RequestHandler, ABC):
????traceid = None
????start_time = None
????# 普通返回
????response_dict = {'code': 0, 'message': "", 'tid': "", 'data': None}
????# 分頁(yè)返回
????response_page_dict = {'code': 0, 'message': "", 'tid': "", 'data': None, "pagination":
????????{"pagesize": 0, "pages": 0, "items": 0, "pageno": 0}
??????????????????????????}
????@classmethod
????def return_response(cls):
????????resp = AttrDict(cls.response_dict)
????????return resp
????@classmethod
????def return_page_response(cls):
????????resp = AttrDict(cls.response_page_dict)
????????return resp
????@classmethod
????def create_page(cls, response: AttrDict, total=None, pagesize=0, pageno=None):
????????response.pagination["items"] = total
????????response.pagination["pagesize"] = pagesize
????????response.pagination["pages"] = total // pagesize if total % pagesize == 0 else total // pagesize + 1
????????response.pagination["pageno"] = pageno
????????return response
????def get_request(self):
????????request_body = None
????????if self.request.body:
????????????request_body = tornado.escape.json_decode(self.request.body)
????????return request_body
????def return_json(self, response):
????????*"""*
**返回統(tǒng)一的*json*格式結(jié)果
*****:param*** *response:*
*"""*
** ?
**self.set_header("Content-Type", "application/json; charset=UTF-8")
????????self.response_dict = json.dumps(response, default=json_util.default)
????????self.write(self.response_dict)
????def prepare(self):
????????trace_id = self.request.headers.get("tid", None)
????????self.__class__.start_time = datetime.datetime.now()
????????if not trace_id:
????????????trace_id = str(uuid.uuid4())
????????????self.request.headers["tid"] = trace_id
????????self.__class__.traceid = trace_id
????def write_error(self, status_code, exc_info=None, api_code=None, **kwargs):
????????resp = self.return_response()
????????resp.code = status_code
????????resp.tid = self.traceid
????????ex = exc_info[1]
????????if isinstance(ex, Exception):
????????????err_msg = str(ex)
????????????resp.message = err_msg
????????if isinstance(ex, tornado.web.HTTPError):
????????????err_msg = str(ex)
????????????resp.message = err_msg
????????if isinstance(ex, tornado.web.HTTPError) and ex.log_message:
????????????err_msg = str(ex.log_message)
????????????resp.message = err_msg
????????self.response_dict = json.dumps(resp, default=json_util.default)
????????self.finish(resp)
????def on_finish(self):
????????# 處理完請(qǐng)求后輸出日志，將請(qǐng)求頭信息和響應(yīng)結(jié)果json化
????????end_time = datetime.datetime.now()
????????request_data = {
????????????"tid": self.request.headers.get("tid", None),
????????????"url": self.request.uri,
????????????"method": self.request.method,
????????????"remote_ip": self.request.remote_ip,
????????????"user_agent": self.request.headers["User-Agent"],
????????????"request_body": json.loads(self.request.body) if self.request.body else ""
????????}
????????response_data = {
????????????"status": self.get_status(),
????????????"response_body": json.loads(self.response_dict) if not isinstance(self.response_dict,
??????????????????????????????????????????????????????????????????????????????dict) else self.response_dict,
????????????"response_time": (end_time - self.start_time).microseconds / 1000
????????}
????????logger = get_logger("access")
????????logger.bind(**request_data).bind(**response_data).info("request success")

import tornado.web
class CustomErrorHandler(tornado.web.RequestHandler):
????def write_error(self, status_code, **kwargs):
????????if self.settings.get("serve_traceback"):
????????????self.write('Some Exception occurs: {}'.format(kwargs['exc_info'][1]))
????????else:
????????????self.write('Some Exception occurs. We are on it!')
????????self.set_status(status_code)
class ErrorExample(tornado.web.RequestHandler):
????def get(self, error_printer):
????????if error_printer == "raise":
????????????raise Exception("An unknown error occurred.")
????????elif error_printer == "syntax":
????????????raise SyntaxError("There is a syntax error in your code.")
????????else:
????????????self.write("This is normal operation!")

def on_finish(self) -> None:
????????"""
????????Finish this response, ending the HTTP request.
????????"""
????????pass