Ủy quyền cho một dịch vụ web khác bằng Flask

Question

Tôi muốn yêu cầu proxy được thực hiện cho ứng dụng Flask của tôi đến một dịch vụ web khác chạy cục bộ trên máy. Tôi muốn sử dụng Flask cho điều này hơn là phiên bản nginx cấp cao hơn của chúng tôi để chúng tôi có thể sử dụng lại hệ thống xác thực hiện có của chúng tôi được tích hợp trong ứng dụng của chúng tôi. Chúng ta càng có thể giữ "dấu hiệu đơn" này càng tốt.

Có mô-đun hiện tại hoặc mã khác để thực hiện việc này không? Đang cố gắng kết nối ứng dụng Flask với một cái gì đó như httplib hoặc urllib đang chứng tỏ là một cơn đau.

Answer 1

Tôi có triển khai proxy sử dụng httplib trong ứng dụng dựa trên Werkzeug (như trong trường hợp của bạn, tôi cần sử dụng xác thực và ủy quyền của webapp).

Mặc dù tài liệu Flask không nêu cách truy cập vào tiêu đề HTTP, bạn có thể sử dụng request.headers (xem Werkzeug documentation). Nếu bạn không cần phải sửa đổi phản hồi và các tiêu đề được ứng dụng proxy sử dụng có thể dự đoán được, proxy sẽ rất dễ hiểu.

Lưu ý rằng nếu bạn không cần sửa đổi phản hồi, bạn nên sử dụng werkzeug.wsgi.wrap_file để bao gồm luồng phản hồi của httplib. Điều đó cho phép truyền mô tả tệp cấp hệ điều hành mở cho máy chủ HTTP để có hiệu suất tối ưu.

Answer 2

Gói ban đầu của tôi là cho URL công khai giống như http://www.example.com/admin/myapp ủy quyền cho http://myapp.internal.example.com/. Trên con đường đó dẫn đến điên khùng.

Hầu hết các ứng dụng web, đặc biệt là các ứng dụng tự lưu trữ, giả định rằng chúng sẽ chạy ở gốc của máy chủ HTTP và thực hiện những việc như tham chiếu các tệp khác theo đường dẫn tuyệt đối. Để giải quyết vấn đề này, bạn phải viết lại URL tất cả các địa điểm: Tiêu đề vị trí và các tệp HTML, JavaScript và CSS.

Tôi đã làm write a Flask proxy blueprint đã làm điều này, và trong khi nó hoạt động tốt đủ cho một webapp tôi thực sự muốn proxy, nó không bền vững. Đó là một mớ hỗn độn lớn của các biểu thức thông thường.

Cuối cùng, tôi thiết lập một máy chủ ảo mới trong nginx và sử dụng proxy của riêng nó. Vì cả hai đều ở gốc của máy chủ, nên việc viết lại URL hầu như không cần thiết. (Và cái gì là cần thiết, mô-đun proxy của nginx được xử lý.) Webapp được ủy quyền để thực hiện xác thực riêng của nó đủ tốt cho đến bây giờ.

Answer 3

Tôi đã dành rất nhiều thời gian làm việc trên cùng một điều này và cuối cùng đã tìm thấy giải pháp sử dụng thư viện yêu cầu có vẻ hoạt động tốt. Nó thậm chí còn xử lý việc thiết lập nhiều cookie trong một câu trả lời, điều này mất một chút điều tra để tìm ra. Đây là chức năng xem bình:

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