leancloud.engine.https_redirect_middleware 源代码

# coding: utf-8

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
from __future__ import unicode_literals

import os

from werkzeug.wrappers import Request
from werkzeug.utils import redirect

__author__ = "asaka <lan@leancloud.rocks>"

is_prod = True if os.environ.get("LEANCLOUD_APP_ENV") == "production" else False

[文档]class HttpsRedirectMiddleware(object): def __init__(self, wsgi_app): self.origin_app = wsgi_app def __call__(self, environ, start_response): request = Request(environ) engine_health = "/1.1/functions/_ops/metadatas" if ( is_prod and request.path != engine_health and request.headers.get("X-Forwarded-Proto") != "https" ): url = "https://{0}{1}".format(request.host, request.full_path) return redirect(url)(environ, start_response) return self.origin_app(environ, start_response)