python 接口自动化测试框架设计（二）

三、base
封装post/get/put请求（run_method.py）

# -*- coding:utf-8 -*-

import  requests
import json
import urllib3
urllib3.disable_warnings()

class RunMethod:

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

    def post_main(self,url,data=None,headers=None,cookie=None):
        res = None
        if (headers != None) and (cookie == None):
            res = requests.post(url=url,data=data,headers=headers)
        elif (headers != None) and (cookie !=None):
            res = requests.post(url=url,data=data,headers=headers,cookie=cookie,verify=False)
        else:
            res = requests.post(url=url,data=data,verify=False)
        return res

    def put_main(self,url,data=None,headers=None):
        res = None
        if headers != None:
            res = requests.put(url=url,data=data,headers=headers)
        else:
            res = requests.put(url=url,data=data)
        return res

    def run_main(self,method,url,data=None,headers=None,cookie=None):
        res = None
        if method == 'post':
            res = self.post_main(url,data,headers,cookie)
        elif method == 'get':
            res = self.get_main(url,data,headers)
        else:
            res = self.put_main(url,data,headers)

        if len(res.text) != 0:
            return res.text
        else:
            with open('D:\\PycharmProjects\\testapi\\dataconfig\\cookies.json','w') as fp:
                fp.write(json.dumps(requests.utils.dict_from_cookiejar(res.cookies)))
            return str(requests.utils.dict_from_cookiejar(res.cookies))

if __name__ == '__main__':
    run_method = RunMethod()