init

web/api/__init__.py

@@ -0,0 +1,99 @@
+
+""" 接口加载器
+"""
+from flask import g, Blueprint
+from nc_http.core import ResponseMeta
+from werkzeug.http import HTTP_STATUS_CODES
+
+import os
+import logging
+from logging.handlers import TimedRotatingFileHandler
+from importlib import import_module
+import glob
+
+from config import APP_MODE, API_ROOT
+
+LOGGER = logging.getLogger()
+
+
+def http_error_handler(err):
+    # 忽略404日志记录
+    if err.code not in [400, 403, 404]:
+        LOGGER.error(err)
+    return ResponseMeta(code=err.code, http_code=err.code, description=err.name)
+
+
+def response_meta_handler(response_meta):
+    return response_meta.get_response()
+
+
+def generic_error_handler(err):
+    LOGGER.exception(err)
+
+    if APP_MODE == 'Production':
+        return ResponseMeta(http_code=500, description='出错了，请及时联系我们，我们的工程师将全力为您解决。')
+    else:
+        return ResponseMeta(http_code=500, description=str(err))
+
+
+def before_request():
+    """
+    g.auth 授权之后
+    g.pagination 返回分页
+    """
+    # 自定义的请求头部
+    g.headers = {}
+
+
+def app_teardown(exc):
+    pass
+
+
+def init_logger(app):
+    log_dir = app.config.get('LOG_DIR')
+
+    if log_dir:
+        file_handler = TimedRotatingFileHandler(os.path.join(log_dir, 'app.log'), 'midnight', 1, 15)
+        file_handler.suffix = '%Y%m%d'
+        file_handler.setFormatter(logging.Formatter(app.config.get('LOG_FORMAT')))
+        file_handler.setLevel(logging.DEBUG)
+        logger = logging.getLogger()
+        logger.addHandler(file_handler)
+        logger.setLevel(logging.DEBUG)
+
+
+# 初始化接口
+def init_app(app):
+    app.teardown_appcontext(app_teardown)
+    app.before_request(before_request)
+
+    for code in HTTP_STATUS_CODES:
+        if code in [400, 401, 403, 404, 405, 406, 409, 500, 502, 503, 504]:
+            app.register_error_handler(code, http_error_handler)
+
+    app.register_error_handler(ResponseMeta, response_meta_handler)
+    app.register_error_handler(Exception, generic_error_handler)
+
+    # from api.helpers.db import main
+    # main()
+
+    # 注册接口
+    path = os.path.dirname(__file__)
+
+    file_names = glob.glob(os.path.join(path, 'components', '*.py'))
+    file_names = file_names + glob.glob(os.path.join(path, 'components', '**', '*.py'))
+    for filename in file_names:
+        filename = filename.replace(os.path.join(path, "components"), '').replace(os.sep, '.')
+        module_name = os.path.basename(filename)[:-3]
+        import_module('.components' + module_name, os.path.basename(path))
+
+    from . import blueprint
+
+    for blueprint_name in dir(blueprint):
+        if not blueprint_name.startswith('__'):
+            bp = getattr(blueprint, blueprint_name)
+            if isinstance(bp, Blueprint):
+                app.register_blueprint(
+                    bp,
+                    url_prefix='{}{}'.format(API_ROOT, bp.url_prefix) if blueprint_name != 'root' else bp.url_prefix
+                )