Python請求鉤子返回一個值以引起異常

Question

對於Python請求模塊Documentation對於鉤子說：「如果回調函數返回一個值，則假定它將替換傳入的數據。如果函數doesn不返回任何東西，沒有其他的影響「

現在我想從我的鉤子函數返回一個值（int在我的情況下），它會引發異常。這在所有情況下都是有效的，當返回值是DOESNOT具有爲其定義的raw（）方法的對象時。

下面是一些代碼

def hook(resp,**kwargs): 
    print resp.url 
    return 1 

def main() 
    s = requests.Session() 
    s.hooks = {"response":hook} 
    r = s.get("http://localhost/index.html") 

這裏是例外：

http://localhost/index.html 
Traceback (most recent call last): 
File "/home/talha/ws/test.py", line 85, in <module> 
    main() 
File "/home/talha/ws/test.py", line 72, in main 
    r = s.get("http://localhost/index.html") 
File "/usr/lib/python2.7/site-packages/requests/sessions.py", line 347, in get 
    return self.request('GET', url, **kwargs) 
File "/usr/lib/python2.7/site-packages/requests/sessions.py", line 335, in request 
    resp = self.send(prep, **send_kwargs) 
File "/usr/lib/python2.7/site-packages/requests/sessions.py", line 446, in send 
    extract_cookies_to_jar(self.cookies, request, r.raw) 
AttributeError: 'int' object has no attribute 'raw' 

在sessions.py @line 446的代碼嘗試dispatch_hook..From源

後提取餅乾

# Response manipulation hooks 
    r = dispatch_hook('response', hooks, r, **kwargs) 

    # Persist cookies 
    extract_cookies_to_jar(self.cookies, request, r.raw) 

文檔需要更改或需要重新處理處理。處理這個問題的最好方法是什麼？

[更新]

基於我試圖返回基地response對象的註釋。原來它也不能以這種方式使用，因爲它的一些字段被初始化爲None。

較新的代碼：

def hook(resp, **kwargs): 
    obj = requests.Response() 
    return obj 

異常拋出現在：

Traceback (most recent call last): 
File "/home/talha/ws/test.py", line 88, in <module> 
    main() 
File "/home/talha/ws/test.py", line 75, in main 
    r = s.get("http://localhost/index.html") 
File "/usr/lib/python2.7/site-packages/requests/sessions.py", line 347, in get 
    return self.request('GET', url, **kwargs) 
File "/usr/lib/python2.7/site-packages/requests/sessions.py", line 335, in request 
    resp = self.send(prep, **send_kwargs) 
File "/usr/lib/python2.7/site-packages/requests/sessions.py", line 446, in send 
    extract_cookies_to_jar(self.cookies, request, r.raw) 
File "/usr/lib/python2.7/site-packages/requests/cookies.py", line 108, in extract_cookies_to_jar 
    res = MockResponse(response._original_response.msg) 
AttributeError: 'NoneType' object has no attribute '_original_response' 

什麼似乎是我將要實現一個完整的僞響應？

Answer 1

如果回調函數返回一個值，則假定它將替換傳入的數據。如果該函數沒有返回任何內容，則不會執行任何操作。

這意味着無論您返回什麼，預計會取代您通過的響應對象。

文檔中沒有說明您可以返回什麼東西。你預期會發生什麼？

如果您想要返回具有不同數據的響應，請返回仍像響應一樣的內容。這意味着，要麼你需要繼承requests響應對象，或實現的東西，提供了相同的API：

from requests.moduls import Response 

class MyIntResponse(Response): 
    def __init__(self, integer): 
     super(MyIntResponse, self).__init__() 
     self._content_consumed = True 
     self._content = integer 

def hook(resp,**kwargs): 
    print resp.url 
    newresp = MyIntResponse(1) 
    newresp.raw = resp.raw # copy across original HTTP response object 

您可能需要對一些從原來的響應的其他屬性的複製;檢查對象具有哪些屬性Response的文檔。