import flask as F
import requests as R



def zfa(proxy_url):
    m=F.request.method
    a=F.request.args
    ps={}
    for i in a:
        v=a.get(i)
        ps[i]=v
    data=F.request.get_data()
    h=F.request.headers
    h2={}
    for k,v in h:
        h2[k]=v
    url=proxy_url+F.request.path
    r=R.request(url=url,method=m,data=data,headers=h2,params=ps)
    r:R.Response
    res=F.Response(response=r.content,headers=dict(r.headers))
    res.status_code=r.status_code
    return res






class HttpProxy:
    def __init__(self,proxy_url) -> None:
        self.proxy_url=proxy_url
        self.flask=F.Flask(__name__)
        self.flask.before_request(self.handle)
    def run(self,*args,**ks):
        return self.flask.run(*args,**ks)
    def on_error(self,e):
        return repr(e)
    
    def pre_handle(self):
        pass
    def post_handle(self,res:F.Response):
        return res
    def handle(self):
        try:
            r=self.pre_handle()
            if r:
                return r
            res=zfa(self.proxy_url)
            res=self.post_handle(res)
            return res
        except Exception as e:
            print(e)
            return self.on_error(e)
   

app=HttpProxy('http://127.0.0.1:8000')
app.run(port=80)