python-sdk/leancloud/engine/https_redirect_middleware.py at master · leancloud/python-sdk

35 lines (24 loc) · 979 Bytes

# 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)

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Expand file tree

Search code, repositories, users, issues, pull requests...

Uh oh!

FilesExpand file tree

https_redirect_middleware.py

Latest commit

History

https_redirect_middleware.py

File metadata and controls

Expand file tree