Spaces:
Runtime error
Runtime error
File size: 1,355 Bytes
836a3ed 5585ad5 836a3ed e2cf778 879fdec 836a3ed 879fdec 836a3ed 879fdec 836a3ed 879fdec 836a3ed 879fdec 836a3ed 879fdec |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 |
import asyncio
import os
from aiohttp import ClientSession
from flask import Flask, request, jsonify
from werkzeug.exceptions import HTTPException
API_ENDPOINT = os.environ.get("API_ENDPOINT", "")
WORKERS = int(os.environ.get("WORKERS", "1"))
app = Flask(__name__)
loop = asyncio.new_event_loop()
asyncio.set_event_loop(loop)
sem = asyncio.Semaphore(WORKERS)
@app.route('/api', methods=['GET', 'POST'])
def handle_request():
headers = {key: value for key, value in request.headers}
data = request.get_data().decode()
method = request.method
response = loop.run_until_complete(
forward_request(headers, data, method))
return jsonify(response), 200
async def forward_request(headers, data, method):
async with sem, ClientSession() as session:
if method == 'POST':
async with session.post(API_ENDPOINT, headers=headers, data=data) as resp:
return await resp.json()
elif method == 'GET':
async with session.get(API_ENDPOINT, headers=headers) as resp:
return await resp.json()
@app.errorhandler(HTTPException)
def handle_exception(e):
response = e.get_response()
response.data = jsonify({"code": e.code, "name": e.name, "description": e.description})
return response
if __name__ == "__main__":
app.run(host='0.0.0.0', port=5000)
|