3、request_method

request_method.py

import requests
import json

class RequestMethod:
    def post_main(self, url, data, cookies=None):
        if cookies != None:
            res = requests.post(url=url, data=data, cookies=cookies, verify=False)
            # print("请求的url：%s"%url+"请求的参数：%s"%str(data))
        else:
            res = requests.post(url=url, data=data, verify=False)
            # print("请求的url：%s" % url + "请求的参数：%s" % str(data))
        return res

    def get_main(self, url, data=None, cookies=None):
        if cookies != None:
            res = requests.get(url=url, data=data, cookies=cookies, verify=False)
        else:
            res = requests.get(url=url, data=data, verify=False)
        return res

    def json_main(self, url, json=None, cookies=None):
        if cookies != None:
            res = requests.post(url=url, json=json, cookies=cookies, verify=False)
        else:
            res = requests.post(url=url, json=json, verify=False)
        return res

    def upload_main(self, url, data=None, files=None, cookies=None):
        if cookies != None:
            res = requests.post(url=url, data=data, files=files, cookies=cookies, verify=False)
        else:
            res = requests.post(url=url, data=data, files=files, verify=False)
        return res

    def run_main(self, method, url, type, data=None, cookies=None):
        requests.packages.urllib3.disable_warnings()  # 防止警告报错
        if type == "json":
            if method == 'post':
                res = (self.post_main(url, data, cookies)).json()
            elif method == 'post_json':
                res = (self.json_main(url, data, cookies)).json()
            else:
                res = (self.get_main(url, data, cookies)).json()
            return json.dumps(res, ensure_ascii=False)
        elif type == "html":
            if method == 'post':
                res = (self.post_main(url, data, cookies)).text
            else:
                res = (self.get_main(url, data, cookies)).text
            return res


if __name__ == '__main__':
    r = RequestMethod()
    url = 'http://test/api/sysmgr/main/login'
    data = {
        "userName": "admin",
        "password": "ZqONDX8Dd3iEjTRKG4HDeg==",
        "secFactorCode": "1"
    }
    headers = {
        "Content-Type": "application/json"
    }
    aa = r.run_main('post_json', url, "json", data, headers)
    print(aa)