Python HTTP客户端手动插入Cookie的自定义Handler实现

Python HTTP客户端手动插入Cookie的自定义Handler实现

大多数脚本语言都提供了便捷的HTTP客户端处理功能，Python也不例外。利用urllib和urllib2模块，可以轻松地执行HTTP GET和POST等多种操作。此外，系统允许以类似插件的方式添加handler，从而定制request和response的处理逻辑，例如代理支持和cookie支持便是通过这种方式实现的。具体而言，可以通过以下方式构造一个opener：

opener = urllib2.build_opener(urllib2.HTTPCookieProcessor())

随后，该opener即可处理cookie，操作相当简便且具备高度可定制性。然而，当前需求是在客户端手动插入特定的cookie值，但无论是HTTPCookieProcessor还是cookielib中的CookieJar，均未提供直接实现此功能的方法。

显然，此类需求并非个例。在搜索解决方案时，发现有人曾向Python提交过相关Patch以添加此功能，但该Patch似乎尚未被接受。对标准库进行此类修改的可移植性较差，因此需要寻找其他解决方案。实际上，方法很简单：通过研究HTTPCookieProcessor的实现代码，可以发现可以编写类似的handler，将所需的cookie值强制写入request对象的header中。

查阅Python文档后发现，关于handler接口的描述极少，因此参考HTTPCookieProcessor的实现进行编写。该handler应置于常规cookie处理handler之后，检查已存在的cookie header并进行合并。值得注意的是，Python文档中未提供Request对象获取已存在header值的方法（如get_header），这令人困惑。通过查阅源代码，确认了该方法确实存在。尽管Python文档通常被认为质量较高，尤其是其末尾的Examples部分，但此次遗漏函数文档的情况也揭示了人工维护文档的局限性。相比之下，Ruby文档从代码注释自动生成，而Python采用独立的人工编写文档系统，尽管后者通常更优，但仍存在疏漏。不过，重点在于实现，具体的handler代码如下：

class SimpleCookieHandler(urllib2.BaseHandler):
 def http_request(self, req):
  simple_cookie = 'cc98Simple=1'
  if not req.has_header('Cookie'):
   req.add_unredirected_header('Cookie', simple_cookie)
  else:
   cookie = req.get_header('Cookie')
   req.add_unredirected_header('Cookie', simple_cookie + '; ' + cookie)
  return req

在构造opener时，加入此handler即可：

opener = urllib2.build_opener(urllib2.HTTPCookieProcessor(), 
SimpleCookieHandler())

这终究是一种变通方案，期待该Patch能被纳入标准库。

感谢阅读，希望能帮助到大家，谢谢大家对本站的支持！