【Python接口測試】10_接口請求方法封裝

接口請求方法千千萬，哈哈沒那麼誇張！為了相容多個不同請求方法的接口，統一接口請求格式，将接口多種請求方法（多用post和get請求方式）統一封裝處理。

import requests
import json

class RunMethod:

	# requests 單獨的 post 接口方法請求
    def post_main(self, url, data, header=None):
        res = None
        if header is not None:
            res = requests.post(url=url, data=data, headers=header).json()
        else:
            res = requests.post(url=url, data=data).json()
        return res

	# requests 單獨的 get 接口方法請求
    def get_main(self, url, data=None, header=None):
        res = None
        if header is not None:
            res = requests.get(url=url, data=data, headers=header)
        else:
            res = requests.get(url=url, data=data)
        return res.json()

	# 根據請求方式，選擇不同的請求方法
    def run_main(self, method, url, data=None, header=None):
        res = None
        if method == "post":
            res = self.post_main(url, data, header)
        else:
            res = self.get_main(url, data, header)
        return json.dumps(res, ensure_ascii=False, indent=4, sort_keys=True)