python接口测试：使用类封装脚本（二）

#!/usr/bin/env python
# coding=utf-8

import requests


class RunMain:
    """
    根据传入的URL和请求方式来发送请求，并返回最终的URL和text
    """
    def __init__(self, url, method, params=None):
        self.url = url
        self.method = method
        self.params = params

    def send_post(self):
        r = requests.post(url=self.url, data=self.params)
        return r.url, r.text
        # requests.utils.dict_from_cookiejar(r.cookies)

    def send_get(self):
        r = requests.get(url=self.url, params=self.params)
        return r.url, r.text
        # requests.utils.dict_from_cookiejar(r.cookies)

    def run_main(self):
        if self.method == 'GET':
            print('This is a get request')
            res = self.send_get()
        else:
            print('This is a post request')
            res = self.send_post()
        return res


if __name__ == '__main__':
    url0 = 'http://httpbin.org/'
    params0 = {
        'key1': 'params1', 'key2': 'params2'
    }
    interface = RunMain(url=url0, method='POST', params=params0)
    print(interface.run_main())