Python实现请求代理

前端开发调试的时候，通常是localhost，而接口不在本机，会有跨域问题。通过http.server 实现的请求代理，解决了跨域问题并且实现登录。

class MyRequestHandler(SimpleHTTPRequestHandler):

    def addHeader(self):
        self.send_header("Access-Control-Allow-Origin", "*")
        self.send_header("Access-Control-Allow-Headers", "*")
        self.send_header("Access-Control-Allow-Methods", "GET, POST, PUT, DELETE")
        self.send_header("Content-type", "text/html")

    def do_GET(self):
        print(\'Get请求：\'+self.path)
        rep_get = requests.get(serverurl + self.path, cookies=cookie_jar)
        if rep_get.status_code == 200:
            self.send_response(200)
            self.addHeader()
            self.end_headers()
            self.wfile.write(rep_get.text.encode("utf-8"))
        else:
            self.Login()
            rep_get = requests.get(serverurl + self.path, cookies=cookie_jar)
            self.send_response(200)
            self.addHeader()
            self.end_headers()
            self.wfile.write(rep_get.text.encode("utf-8"))

    def do_POST(self):
        print(\'Post请求：\'+self.path)
        request_contentType = str(self.headers["Content-Type"])
        if request_contentType.find(\'application/x-www-form-urlencoded\')>-1:
            data = str(self.rfile.read(int(self.headers["content-length"])),\'UTF-8\')

            #data = str(self.rfile.readlines(), \'UTF-8\')  # 先解码
            p1 = urllib.parse.parse_qs(urllib.parse.unquote(data))  # 解释参数
            rel_par={}
            for it in p1:
                rel_par[it]=p1[it][0]

            rep_post = requests.post(serverurl + self.path, cookies=cookie_jar,data=rel_par)
            if rep_post.status_code==200:

                self.send_response(200)
                self.addHeader()
                self.end_headers()
                self.wfile.write(rep_post.text.encode("utf-8"))
            else:
                self.Login()
                rep_post = requests.post(serverurl + self.path, cookies=cookie_jar, data=rel_par)
                self.send_response(200)
                self.addHeader()
                self.end_headers()
                self.wfile.write(rep_post.text.encode("utf-8"))

        else:
            req = {"success": "false", "message": "暂时只支持Content-Type=application/x-www-form-urlencoded"}
            self.send_response(200)
            self.addHeader()
            self.end_headers()
            rspstr = json.dumps(req)
            self.wfile.write(rspstr.encode("utf-8"))

    def Login(self):
        isLogin =True
        req_login1 = requests.post(serverurl + \'/SUP/Login/WebLogin\',data={
            \'ocode\': \'\',
            \'UserID\': user,
            \'UserPwd\': pwd,
            \'DataBase\': ucode,
            \'Language\': "zh-CN",
            \'verifyCode\': ""
        })
        if req_login1.status_code == 200:
            if req_login1.text.find(\'success: true\') > 0:
                globals()[\'cookie_jar\']=req_login1.cookies
            else:
                i1 = req_login1.text.find(\'msg\')
                km = json.loads(req_login1.text[i1 + 5:-2])
                if km != None:
                    print(\'开始踢人\')
                    req_kill = requests.post(serverurl + \'/SUP/Login/KillOnlineUser\', data={
                        \'IpAddress\': km[\'IpAddress\'],
                        \'Devicetype\': km[\'Devicetype\'],
                        \'UserId\': km[\'UserId\'],
                        \'SessionID\': km[\'SessionID\']
                    })
                    if req_kill.status_code == 200:
                        print(\'踢人成功，重新登录\')
                        req_login2 = requests.post(serverurl + \'/SUP/Login/WebLogin\', data={
                            \'ocode\': \'\',
                            \'UserID\': user,
                            \'UserPwd\': pwd,
                            \'DataBase\': ucode,
                            \'Language\': "zh-CN",
                            \'verifyCode\': "",
                            \'IsOnlineCheck\': \'1\'
                        })
                        if req_login2.status_code == 200:
                            globals()[\'cookie_jar\'] = req_login2.cookies
                        else:
                            isLogin = False
                            print(\'踢人以后登录返回错误，请检查服务器是否可访问\')
                    else:
                        isLogin = False
                        print(\'踢人接口错误，请检查服务器是否可访问\')
                else:
                    isLogin = False
                    print(\'登录返回错误，请检查服务器是否可访问\')
        else:
            isLogin = False
            print(\'服务器登录返回错误，请检查服务器是否可访问\')
        return isLogin

使用：

try:
    config = configparser.ConfigParser()
    isread = config.read(\'ngproxy.ini\')
    if len(isread) == 0:
        serverurl = input("请输入服务器url: ")
        ucode = input("请输入服务器账套: ")
        user = input("请输入服务器用户: ")
        pwd = input("请输入服务器密码: ")
        port = input("请输入代理端口: ")

        config.add_section(\'userinfo\')
        config.set(\'userinfo\', \'serverurl\', serverurl)
        config.set(\'userinfo\', \'username\', user)
        config.set(\'userinfo\', \'password\', pwd)
        config.set(\'userinfo\', \'ucode\', ucode)
        config.set(\'userinfo\', \'port\', port)
        f = open(\'ngproxy.ini\', "w")
        config.write(f)  # 写进文件
        f.close()
    else:
        serverurl = config[\'userinfo\'][\'serverurl\']
        user = config[\'userinfo\'][\'username\']
        pwd = config[\'userinfo\'][\'password\']
        ucode = config[\'userinfo\'][\'ucode\']
        port = config[\'userinfo\'][\'port\']
    bytes_user=user.encode()
    user = str(base64.b64encode(bytes_user), \'utf-8\')

    bytes_pwd = pwd.encode()
    pwd = base64.b64encode(bytes_pwd)
    pwd = str(pwd, \'utf-8\')

    cookie_jar = RequestsCookieJar()

    server = HTTPServer(("", int(port)), MyRequestHandler)
    print("代理服务器启动成功, serving at http://localhost:" + port)
    server.serve_forever()