使用 Flask 代理到另一个 Web 服务

Question

我想将对 Flask 应用程序发出的请求代理到计算机上本地运行的另一个 Web 服务。我宁愿使用 Flask 而不是更高级别的 nginx 实例，这样我们就可以重用应用程序中内置的现有身份验证系统。我们越能保持这种“单点登录”越好。

是否有现有的模块或其他代码可以执行此操作？尝试将 Flask 应用程序桥接到 httplib 或 urllib 之类的东西被证明是一件痛苦的事情。

Answer 1

我花了很多时间研究同样的事情，并最终找到了一个使用 requests 库的解决方案，该解决方案似乎运行良好。它甚至可以在一个响应中设置多个 cookie，这需要一些调查才能弄清楚。这是 Flask 视图函数：

from dotenv import load_dotenv  # pip package python-dotenv
import os
#
from flask import request, Response
import requests  # pip package requests


load_dotenv()
API_HOST = os.environ.get('API_HOST'); assert API_HOST, 'Envvar API_HOST is required'

@api.route('/', defaults={'path': ''}, methods=["GET", "POST"])  # ref. https://medium.com/@zwork101/making-a-flask-proxy-server-online-in-10-lines-of-code-44b8721bca6
@api.route('/<path>', methods=["GET", "POST"])  # NOTE: better to specify which methods to be accepted. Otherwise, only GET will be accepted. Ref: https://flask.palletsprojects.com/en/3.0.x/quickstart/#http-methods
def redirect_to_API_HOST(path):  #NOTE var :path will be unused as all path we need will be read from :request ie from flask import request
    res = requests.request(  # ref. https://stackoverflow.com/a/36601467/248616
        method          = request.method,
        url             = request.url.replace(request.host_url, f'{API_HOST}/'),
        headers         = {k:v for k,v in request.headers if k.lower() != 'host'}, # exclude 'host' header
        data            = request.get_data(),
        cookies         = request.cookies,
        allow_redirects = False,
    )

    #region exlcude some keys in :res response
    excluded_headers = ['content-encoding', 'content-length', 'transfer-encoding', 'connection']  #NOTE we here exclude all "hop-by-hop headers" defined by RFC 2616 section 13.5.1 ref. https://www.rfc-editor.org/rfc/rfc2616#section-13.5.1
    headers          = [
        (k,v) for k,v in res.raw.headers.items()
        if k.lower() not in excluded_headers
    ]
    #endregion exlcude some keys in :res response

    response = Response(res.content, res.status_code, headers)
    return response

2021 年 4 月更新：excluded_headers 可能应该包含 RFC 2616 第 13.5.1 节。

Answer 2

我在基于 Werkzeug 的应用程序中使用 httplib 实现了代理（就像您的情况一样，我需要使用 web 应用程序的身份验证和授权）。

尽管 Flask 文档没有说明如何访问 HTTP 标头，但您可以使用 request.headers （请参阅 Werkzeug 文档）。如果您不需要修改响应，并且代理应用程序使用的标头是可预测的，则代理是直接的。

请注意，如果您不需要修改响应，则应使用 werkzeug.wsgi.wrap_file 来包装 httplib 的响应流。这允许将打开的操作系统级文件描述符传递到 HTTP 服务器以获得最佳性能。

Answer 3

我最初的计划是面向公众的 URL 类似于 http://www.example.com/admin/myapp 代理到 http://myapp.internal.example.com /。沿着这条路走下去会导致疯狂。

大多数网络应用程序，特别是自托管的网络应用程序，假设它们将在 HTTP 服务器的根目录下运行，并执行诸如通过绝对路径引用其他文件之类的操作。要解决此问题，您必须重写所有位置的 URL：位置标头以及 HTML、JavaScript 和 CSS 文件。

我编写了一个 Flask 代理蓝图，它做到了这一点，虽然它对于我真正想要代理的一个 Web 应用程序来说工作得足​​够好，但它是不可持续的。这是一团乱七八糟的正则表达式。

最后，我在 nginx 中设置了一个新的虚拟主机并使用了它自己的代理。由于两者都位于主机的根目录，因此几乎不需要重写 URL。 （nginx 的代理模块处理了几乎不需要的事情。）被代理的 web 应用程序会进行自己的身份验证，目前这已经足够好了。