接口测试第三步 --》 封装请求方法

#！/user/bin/python3
#coding=utf-8
#2019/10/20 --13:17
#author :hui

import json,requests
#封装 get和post方法
class Method:
    #post 方法
    def post(self,url,body,headers=None):
        res=None
        #如果有headers，执行post请求，传入url 和body
        if headers!=None:
            res=requests.post(url=url,json=body,headers=headers)
        else:
            res=requests.post(url=url,body=body)
        return res.json() #以json串格式返回
    #get方法
    def get(self,url,body,headers=None):
        res=None
        if headers!=None:
            res = requests.get(url=url, json=body, headers=headers)
        else:
            res=requests.get(url=url,body=body)
        return res.json()

    # 判断是get还是post请求
    def choice_method(self,method,url,body=None,headers=None):
        res = None
        if method =='post':
            res=self.post(url,body,headers)
        else:
            res=self.get(url,body,headers)
        return json.dumps(res,indent=2)#保留2行缩进