使用Flask搭建代理服务器

使用Flask搭建代理转发restful请求

开发过程当中常常须要测试接口,有些接口须要须要认证信息(token). 为了不在每一个请求中重复加入token. 能够简单搭建一个代理服务统一拦截全部请求.

Flask提供了完善的生命周期函数,咱们能够在before_request中拦截请求,加上token后转发出去,而后将Response返回.

from flask import Flask, request, jsonify, Response
import requests


def create_app():
    app = Flask(__name__)
    app.config['JSON_AS_ASCII'] = False
    return app


app = create_app()


@app.before_request
def proxy():
    headers = {h[0]: h[1] for h in request.headers}
    url = request.url
    headers['x-token'] = '***'
    # 一些本身的逻辑...
    return requests.request(request.method, url, data=request.json, headers=headers).content

if __name__ == "__main__":
    app.run()