import requests as RE
import flask as FL




class Proxy(FL.Flask):
    def __init__(self,host):
       
        super().__init__(__name__)
        self.host=host
        self.before_request(self._handle)
    
    def _handle(self):
        req=FL.request
        ps={}
        for k in req.args:
            v=req.args[k]
            ps[k]=v
        hd={}
        for k,v in req.headers:
            hd[k]=v
        r=RE.request(method=req.method,url=self.host+req.path,data=self._send(req.stream))
        out=self._output(r)
        res=FL.Response(response=out)
        res.status_code=r.status_code
        for k in r.headers:
            v=r.headers[k]
            res.headers[k]=v
        return res
            
    def _send(self,f):
        while 1:
            r=f.read(1024)
            if not r:
                break
            yield r

    def _output(self,r:RE.Response):
        for i in r.iter_content(chunk_size=1024):
            yield i
   
        


app=Proxy('http://127.0.0.1:8000')
app.run()