Python 3請求或urllib - 如何總是添加標頭？

Question

標題說明了一切：是否有一種'最佳'的方式來爲每個請求添加標題？我有一個內部工具，希望將請求ID發送給其他內部工具;我正在尋找一個有福的解決方案。我已經瀏覽了兩者的文檔，看起來這不是一個受歡迎的東西，因爲我無法找到一本食譜示例。

我想，一對夫婦的解決方案：在我自己的瘦包裝

1. 裹請求並使用它。需要教授開發人員記得不要import requests，但import myrequestswrapper as requests。
2. 猴子補丁請求。我不喜歡猴子補丁，但也許這只是一次......？我怕有什麼時候需要而不是給這個特定系統發送一個頭文件。

編輯：爲什麼我沒有考慮requests.Session：它存儲cookies，因爲它保持連接打開狀態，所以需要處理。

Answer 1

創建session object，這是advanced usage下所示的第1件事：

​​

，並使用會話執行請求。正如你說，你不希望在請求之間持續存在的cookie，你可以繼承的Session：

In [64]: from requests.adapters import HTTPAdapter 

In [65]: from requests.cookies import cookiejar_from_dict 

In [66]: class CookieMonsterSession(Session): 
    ...:  
    ...:  def __init__(self, *args, **kwgs): 
    ...:   super(CookieMonsterSession, self).__init__(*args, **kwgs) 
    ...:   # Override default adapters with 0-pooling adapters 
    ...:   self.mount('https://', HTTPAdapter(pool_connections=1, 
    ...:           pool_maxsize=0)) 
    ...:   self.mount('http://', HTTPAdapter(pool_connections=1, 
    ...:           pool_maxsize=0)) 
    ...:  @property 
    ...:  def cookies(self): 
    ...:   """ Freshly baked cookies, always!""" 
    ...:   return cookiejar_from_dict({}) 
    ...:  @cookies.setter 
    ...:  def cookies(self, newcookies): 
    ...:   """ OM NOM NOM NOM...""" 
    ...:   pass 
    ...:  

In [67]: s = CookieMonsterSession() 

In [69]: real_s = Session() 

In [70]: s.get('http://www.google.fi') 
Out[70]: <Response [200]> 

In [71]: s.cookies 
Out[71]: <RequestsCookieJar[]> 

In [72]: real_s.get('http://www.google.fi') 
Out[72]: <Response [200]> 

In [73]: real_s.cookies 
Out[73]: <RequestsCookieJar[Cookie(version=0, name='NID', value='86=14qy...Rurx', port=None, port_specified=False, domain='.google.fi', domain_specified=True, domain_initial_dot=True, path='/', path_specified=True, secure=False, expires=1489744358, discard=False, comment=None, comment_url=None, rest={'HttpOnly': None}, rfc2109=False)]> 

不幸的是，Session是由設計難以擴展和配置，所以「禁用」餅乾，並修改像這樣的池是一個破解，並容易打破，如果當Session更新絲毫。此外，我們僅針對持久性標頭禁用了Session的2個主要功能。

結束語基本API方法可能是更清潔和更安全的方法：

# customrequests.py 
from functools import wraps 
from requests import api as requests_api 

custom_headers = {} 


def _header_wrapper(f): 
    @wraps(f) 
    def wrapper(*args, **kwgs): 
     headers = kwgs.pop('headers', None) or {} 
     headers.update(custom_headers) 
     return f(*args, headers=headers, **kwgs) 

    return wrapper 

request = _header_wrapper(requests_api.request) 
get = _header_wrapper(requests_api.get) 
options = _header_wrapper(requests_api.options) 
head = _header_wrapper(requests_api.head) 
post = _header_wrapper(requests_api.post) 
put = _header_wrapper(requests_api.put) 
patch = _header_wrapper(requests_api.patch) 
delete = _header_wrapper(requests_api.delete) 

在行動：

In [1]: import customrequests as requests 

In [2]: print(requests.get('http://httpbin.org/headers').text) 
{ 
    "headers": { 
    "Accept": "*/*", 
    "Accept-Encoding": "gzip, deflate", 
    "Host": "httpbin.org", 
    "User-Agent": "python-requests/2.11.1" 
    } 
} 


In [3]: requests.custom_headers['X-Test'] = "I'm always here" 

In [4]: print(requests.get('http://httpbin.org/headers').text) 
{ 
    "headers": { 
    "Accept": "*/*", 
    "Accept-Encoding": "gzip, deflate", 
    "Host": "httpbin.org", 
    "User-Agent": "python-requests/2.11.1", 
    "X-Test": "I'm always here" 
    } 
}