如何实现烧瓶重定向装饰器

时间:2019-07-02 18:57:08

标签: flask python-decorators flask-restful flask-login

我有一个Flask应用程序实例在端口80上运行,同一应用程序的另一个实例在端口8080上运行。我希望所有用户对所有URL使用端口80,但一个名为/parallel_url的URL除外。因此,当他们访问https://server:80/parallel_url时,它会重定向到https://server:8080/parallel_url。我也不希望用户能够访问端口8080上的任何其他URL(不包括/parallel_url),并且希望他们被重定向到发布80 因此,如果他们访问https://server:8080/*,则需要重定向到https://server:80/*

我打算编写一个装饰器,以查看当前URL是否为/parallel_url之外的其他任何东西,并将其重定向到80。因此,如果用户正在端口8080中访问此URL,则需要将其重定向到80 < / p>


def parallel_redirect(func):

    @functools.wraps(func)
    def wrapper():
        """
           What to do here to redirect to 80
        """
        if request.url_rule == "/parallel_url":
            value = func()
        return value
    return wrapper

@mod.route('/parallel_url')
@parallel_redirect
@login_required
def parallel_url():
    template = tpl_env.get_template('my_index.html')
    x = "some data from db"
    y = "some other data from db"
    return template.render(x=x,y=y)

更新: 我找到了一个变量,该变量告诉我我在哪个端口上。请参考下面的代码


def parallel_redirect(func):

    @functools.wraps(func)
    def wrapper():
        if os.environ.curr_port == 8080:
            if request.url_rule != "/parallel_url":
               # code to redirect to "https://server:80/"+request.url_rule
        elif os.environ.curr_port == 80:
            if request.url_rule == "/parallel_url":
               # code to redirect to "https://server:8080/"+request.url_rule
        else:
             value = func()

        return value
    return wrapper

@mod.route('/parallel_url')
@parallel_redirect
@login_required
def parallel_url():
    template = tpl_env.get_template('my_index.html')
    x = "some data from db"
    y = "some other data from db"
    return template.render(x=x,y=y)

1 个答案:

答案 0 :(得分:1)

我只会捕获所有可能的组合,并使用redirect函数来调用其他服务器。您只需要请求中的当前端口即可,例如See annotated plot here

from flask import redirect

def parallel_redirect(func):
  @functools.wraps(func):
    request_port = request.environ.get('REMOTE_PORT')
    if request.url_rule == "/parallel_url":
      if request_port == 80:
        # /parallel_url, but on port 80, redirect
        return redirect("https://...:8080/parallel_url", code=302)
      else:
        return func()
    elif request_port == 8080:
      # not /parallel_url, but on port 8080, also redirect
      return redirect("https://...:80" + request.url_rule, code=302)
    # finally, not /parallel_url, and on port 80, so we're all good
    return func()