python 接口自动化http请求方法封装

# coding=utf-8
import requests
import json


class RunMain:

    def send_post(self, url, data,headers):  # 定义一个方法，传入需要的参数url和data
        result = requests.request("POST", url, data=data, headers=headers) 
        return result
        # print res

    def send_get(self, url, data):
        result = requests.get(url=url, data=data)
        res = json.dumps(result,ensure_ascii=False,sort_keys=True,indent=2)
        return res

    def run_main(self, method, url=None, data=None,headers=None):
        result = None
        if method == 'post':
            result = self.send_post(url, data,headers)
        elif method == 'get':
            result = self.send_get(url, data)
        else:
            print("错误")
        return result

 

if __name__ == '__main__':
    url = "http://..../..../api/govern/center/getMdColData"
    data = '{"mouth": "","year": ""}'
    data = {
        "mouth": "",
        "year": ""
    }
    data_str = json.dumps(data)
    headers = {
        'Content-Type': "application/json",
        'cache-control': "no-cache",
        'Postman-Token': "da47772b-8a0f-4483-a2c9-b94f93d08ede"
    }
    # 实例化RunMain类
    run = RunMain()
    res = run.run_main("post",url,data_str,headers)
    print(res)