1. 程式人生 > 實用技巧 >requests二次封裝

requests二次封裝

做api自動化的時候,可以將requests進行二次封裝,減少程式碼冗餘度

import requests




class req_method(object):
    def __init__(self):
        pass
    def try_do(self,json_ret=None):
        try:
            ret=json_ret.json()
        except:
            ret=json_ret.text
        return ret


    def req_get(self,url=None,datas=None,header=None):
        if url and datas:
            ret=requests.get(url=url,params=datas,headers=header)
            json_ret=self.try_do(ret)
            return json_ret
        else:
            return False
    def req_post(self,url=None,datas=None,header=None):
        if url and datas:
            ret=requests.post(url=url,json=datas,headers=header)
            json_ret=self.try_do(ret)
            return json_ret
        else:
            return False
    def req_put(self,url=None,datas=None,header=None):
        if url and datas:
            ret=requests.put(url=url,json=datas,headers=header)
            json_ret=self.try_do(ret)
            return json_ret
        else:
            return False