Pyhton接口自动化—封装get、post方法

#coding:utf-8
import requests
#执行get、post方法封装
class RunMethod:
    def __init__(self,method,url,data=None,header=None):
        self.res=self.run_main(method,url,data,header)
    def post_main(self,url,data,header=None):
        res=None
        if header!=None:
            res=requests.post(url=url, data=data, headers=header).json()
        else:
            res=requests.post(url=url,data=data).json()
        return res
    def get_main(self,url,data=None,header=None):
        res = None
        if header != None:
            res = requests.get(url=url, data=data, headers=header).json()
        else:
            res = requests.get(url=url, data=data).json()
        return res
    def run_main(self,method,url,data,header=None):
        res=None
        if method=='post':
            res = self.post_main(url,data,header)
        else:
            res =self.get_main(url,data,header)
        return res
if __name__ == '__main__':
    url="XXXX/api/sysmgr/sso/login"
    method='post'
    data={
        "account": "XXX",
        "password": "XXXX",
        "domainNumber" :"XXXX"
    }
    run=RunMethod(method,url,data)