Python接口自動化淺析requests請求封裝原理

第Python接口自動化淺析requests請求封裝原理目錄以下主要介紹如何封裝請求將常用的get、post請求封裝起來get請求源碼：post請求源碼：再來研究下request源碼：直接調(diào)用request函數(shù)在上一篇Python接口自動化測試系列文章：Python接口自動化淺析Token應(yīng)用原理，介紹token基本概念、運(yùn)行原理及在自動化中接口如何攜帶token進(jìn)行訪問。

以下主要介紹如何封裝請求

還記得我們之前寫的get請求、post請求么？

大家應(yīng)該有體會，每個請求類型都寫成單獨(dú)的函數(shù)，代碼復(fù)用性不強(qiáng)。

接下來將請求類型都封裝起來，自動化用例都可以用這個封裝的請求類進(jìn)行請求

將常用的get、post請求封裝起來

importrequests

classRequestHandler:

defget(self,url,**kwargs):

"""封裝get方法"""

#獲取請求參數(shù)

params=kwargs.get("params")

headers=kwargs.get("headers")

try:

result=requests.get(url,params=params,headers=headers)

returnresult

exceptExceptionase:

print("get請求錯誤:%s"%e)

defpost(self,url,**kwargs):

"""封裝post方法"""

#獲取請求參數(shù)

params=kwargs.get("params")

data=kwargs.get("data")

json=kwargs.get("json")

try:

result=requests.post(url,params=params,data=data,json=json)

returnresult

exceptExceptionase:

print("post請求錯誤:%s"%e)

defrun_main(self,method,**kwargs):

判斷請求類型

:parammethod:請求接口類型

:paramkwargs:選填參數(shù)

:return:接口返回內(nèi)容

ifmethod=='get':

result=self.get(**kwargs)

returnresult

elifmethod=='post':

result=self.post(**kwargs)

returnresult

else:

print('請求接口類型錯誤')

if__name__=='__main__':

#以下是測試代碼

#get請求接口

url='https://api.apiopen.top/getJokepage=1count=2type=video'

res=RequestHandler().get(url)

#post請求接口

url2=':8000/user/login/'

payload={

"username":"vivi",

"password":"123456"

res2=RequestHandler().post(url2,json=payload)

print(res.json())

print(res2.json())

請求結(jié)果如下：

'message':'成功!',

'result':[{'sid':'31004305',

'text':'羊：師傅，理個發(fā)，稍微修一下就行',

'type':'video',

'thumbnail':'/picture/2025/0410/5e8fbf227c7f3_wpd.jpg',

'video':'/video/2025/0410/5e8fbf227c7f3_wpd.mp4',

'images':None,

'up':'95',

'down':'1',

'forward':'0',

'comment':'25',

'uid':'23189193',

'name':'青川小舟',

'header':'/profile/large/2025/12/24/5e01934bb01b5_mini.jpg',

'top_comments_content':None,

'top_comments_voiceuri':None,

'top_comments_uid':None,

'top_comments_name':None,

'top_comments_header':None,

'passtime':'2025-04-1201:43:02'},

{'sid':'30559863',

'text':'機(jī)器人女友，除了不能生孩子，其他的啥都會，價(jià)格239000元',

'type':'video',

'thumbnail':'/picture/2025/0306/5e61a41172a1b_wpd.jpg',

'video':'/video/2025/0306/5e61a41172a1b_wpd.mp4',

'images':None,'up':'80','down':'6',

'forward':'3',

'comment':'20',

'uid':'23131273',

'name':'水到渠成',

'header':'/profile/large/2025/07/04/5d1d90349cd1a_mini.jpg',

'top_comments_content':'為游戲做的秀',

'top_comments_voiceuri':'',

'top_comments_uid':'10250040',

'top_comments_name':'不得姐用戶',

'top_comments_header':'/profile',

'passtime':'2025-04-1120:43:49'}]}

{'token':'eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJ1c2VyX2lkIjoxLCJ1c2VybmFtZSI6InZpdmkiLCJleHAiOjE1ODY4NTc0MzcsImVtYWlsIjoidml2aUBxcS5jb20ifQ.k6y0dAfNU2o9Hd9LFfxEk1HKgczlQfUaKE-imPfTsm4',

'user_id':1,

'username':'vivi'}

這樣就完美了嗎，no,no,no。

以上代碼痛點(diǎn)如下：

代碼量大：只是封裝了get、post請求，加上其他請求類型，代碼量較大；

缺少會話管理：請求之間如何保持會話狀態(tài)。

我們再來回顧下get、post等請求源碼，看下是否有啥特點(diǎn)。

get請求源碼：

defget(url,params=None,**kwargs):

r"""SendsaGETrequest.

:paramurl:URLforthenew:class:`Request`object.

:paramparams:(optional)Dictionary,listoftuplesorbytestosend

inthequerystringforthe:class:`Request`.

:param\*\*kwargs:Optionalargumentsthat``request``takes.

:return::class:`ResponseResponse`object

:rtype:requests.Response

kwargs.setdefault('allow_redirects',True)

returnrequest('get',url,params=params,**kwargs)

post請求源碼：

defpost(url,data=None,json=None,**kwargs):

r"""SendsaPOSTrequest.

:paramurl:URLforthenew:class:`Request`object.

:paramdata:(optional)Dictionary,listoftuples,bytes,orfile-like

objecttosendinthebodyofthe:class:`Request`.

:paramjson:(optional)jsondatatosendinthebodyofthe:class:`Request`.

:param\*\*kwargs:Optionalargumentsthat``request``takes.

:return::class:`ResponseResponse`object

:rtype:requests.Response

returnrequest('post',url,data=data,json=json,**kwargs)

仔細(xì)研究下，發(fā)現(xiàn)get、post請求返回的都是request函數(shù)。

再來研究下request源碼：

defrequest(method,url,**kwargs):

"""Constructsandsendsa:class:`RequestRequest`.

:parammethod:methodforthenew:class:`Request`object.

:paramurl:URLforthenew:class:`Request`object.

:paramparams:(optional)Dictionary,listoftuplesorbytestosend

inthequerystringforthe:class:`Request`.

:paramdata:(optional)Dictionary,listoftuples,bytes,orfile-like

objecttosendinthebodyofthe:class:`Request`.

:paramjson:(optional)AJSONserializablePythonobjecttosendinthebodyofthe:class:`Request`.

:paramheaders:(optional)DictionaryofHTTPHeaderstosendwiththe:class:`Request`.

:paramcookies:(optional)DictorCookieJarobjecttosendwiththe:class:`Request`.

:paramfiles:(optional)Dictionaryof``'name':file-like-objects``(or``{'name':file-tuple}``)formultipartencodingupload.

``file-tuple``canbea2-tuple``('filename',fileobj)``,3-tuple``('filename',fileobj,'content_type')``

ora4-tuple``('filename',fileobj,'content_type',custom_headers)``,where``'content-type'``isastring

definingthecontenttypeofthegivenfileand``custom_headers``adict-likeobjectcontainingadditionalheaders

toaddforthefile.

:paramauth:(optional)AuthtupletoenableBasic/Digest/CustomHTTPAuth.

:paramtimeout:(optional)Howmanysecondstowaitfortheservertosenddata

beforegivingup,asafloat,ora:ref:`(connecttimeout,read

timeout)timeouts`tuple.

:typetimeout:floatortuple

:paramallow_redirects:(optional)Boolean.Enable/disableGET/OPTIONS/POST/PUT/PATCH/DELETE/HEADredirection.Defaultsto``True``.

:typeallow_redirects:bool

:paramproxies:(optional)DictionarymappingprotocoltotheURLoftheproxy.

:paramverify:(optional)Eitheraboolean,inwhichcaseitcontrolswhetherweverify

theserver'sTLScertificate,orastring,inwhichcaseitmustbeapath

toaCAbundletouse.Defaultsto``True``.

:paramstream:(optional)if``False``,theresponsecontentwillbeimmediatelydownloaded.

:paramcert:(optional)ifString,pathtosslclientcertfile(.pem).IfTuple,('cert','key')pair.

:return::class:`ResponseResponse`object

:rtype:requests.Response

Usage::

importrequests

req=requests.request('GET','/get')

Response[200]

#Byusingthe'with'statementwearesurethesessionisclosed,thuswe

#avoidleavingsocketsopenwhichcantriggeraResourceWarninginsome

#cases,andlooklikeamemoryleakinothers.

withsessions.Session()assession:

returnsession.request(method=method,url=url,**kwargs)

源碼看起來很長，其實(shí)只有三行，大部分是代碼注釋。

從源碼中可以看出，不管是get還是post亦或其他請求類型，最終都是調(diào)用request函數(shù)。

既然這樣，我們可以不像之前那樣，在類內(nèi)定義get方法、post方法，而是定義一個通用的方法

直接調(diào)用request函數(shù)

看起來有點(diǎn)繞，用代碼實(shí)現(xiàn)就清晰了。

importrequests

classRequestHandler:

def__init__(self):

"""session管理器"""

self.session=requests.session()

defvisit(self,method,url,params=None,data=None,json=None,headers