python自动化框架搭建(一)

对请求方法进行封装

使用requests库,对请求方法进行封装

import json
import requests

class RunMain:
    def send_post(self,url, data): 
        header=""
        res = requests.post(url= url, json = data)
        # return json.dumps(res, indent=2, sort_keys=True)
        return res.text

    def send_get(self,url,data):
        res = requests.get(url = url, json = data)
        return res.text

对post、get基类进行封装

import  requests
class RunMethod:
    def post_main(self,url,data,header=None):
        res = None
        if header != None:
            res=requests.post(url = url, json = data, headers = header).text
        else:
            res=requests.post(url = url, json = data).text
        return res

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

    def run_main(self,method,url,data = None,header = None):
        res = None
        if method == 'post':
            res = self.post_main(url,data,header)
            return res
        else:
            res = self.get_main(url,data,header)
            return res

猜你喜欢

转载自blog.csdn.net/Krystal_RonghuiLi/article/details/108003466
今日推荐