Proxingir a otro servicio web con el Flask

Question

Deseo enviar las solicitudes de proxy a mi aplicación Flask a otro servicio web que se ejecute localmente en la máquina. Prefiero usar Flask para esto que nuestra instancia de nginx de nivel superior para que podamos reutilizar nuestro sistema de autenticación existente integrado en nuestra aplicación. Mientras más podamos mantener este "inicio de sesión único", mejor.

¿Existe un módulo u otro código para hacer esto? Tratar de conectar la aplicación Flask con algo como httplib o urllib está demostrando ser un dolor.

Answer 1

Tengo una implementación de un proxy usando httplib en una aplicación basada en Werkzeug (como en su caso, necesitaba usar la autenticación y autorización de la aplicación web).

Aunque los documentos del Flask no indican cómo acceder a los encabezados HTTP, puede usar request.headers (consulte Werkzeug documentation). Si no necesita modificar la respuesta, y los encabezados utilizados por la aplicación de proxy son predecibles, el proxy es continuo.

Tenga en cuenta que si no necesita modificar la respuesta, debe usar werkzeug.wsgi.wrap_file para ajustar la secuencia de respuesta de httplib. Eso permite pasar el descriptor de archivo de nivel de SO abierto al servidor HTTP para un rendimiento óptimo.

Answer 2

Mi plan original era que la URL pública fuera algo así como http://www.example.com/admin/myapp proxying a http://myapp.internal.example.com/. Por ese camino lleva la locura.

La mayoría de los webapps, especialmente los autohospedados, asumen que se ejecutarán en la raíz de un servidor HTTP y hacen cosas como referenciar otros archivos por ruta absoluta. Para evitar esto, debe volver a escribir las URL en cualquier lugar: encabezados de ubicación y archivos HTML, JavaScript y CSS.

Hice write a Flask proxy blueprint que hizo esto, y si bien funcionó lo suficientemente bien para la única aplicación web que realmente quería proxy, no era sostenible. Fue un gran lío de expresiones regulares.

Al final, configuré un nuevo host virtual en nginx y usé su propio proxying. Dado que ambos estaban en la raíz del host, la reescritura de URL fue en su mayoría innecesaria. (Y lo poco que fue necesario, se manejó el módulo de proxy de nginx). La aplicación de Internet a la que se hace el proxy realiza su propia autenticación, que es lo suficientemente buena por ahora.

Answer 3

Pasé mucho tiempo trabajando en esto mismo y, finalmente, encontré una solución utilizando la solicitudes de biblioteca que parece funcionar bien. Incluso maneja la configuración de múltiples cookies en una respuesta, lo que llevó un poco de investigación para resolver. Aquí está la función de vista del matraz:

from flask import request, Response 
import requests 

def _proxy(*args, **kwargs): 
    resp = requests.request(
     method=request.method, 
     url=request.url.replace('old-domain.com', 'new-domain.com'), 
     headers={key: value for (key, value) in request.headers if key != 'Host'}, 
     data=request.get_data(), 
     cookies=request.cookies, 
     allow_redirects=False) 

    excluded_headers = ['content-encoding', 'content-length', 'transfer-encoding', 'connection'] 
    headers = [(name, value) for (name, value) in resp.raw.headers.items() 
       if name.lower() not in excluded_headers] 

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