python-3.x - 带有自定义 header 的 urllib.urlretrieve

Question

我正在尝试使用 urlretrieve 检索文件，同时添加自定义标题。

在查看 urllib.request 的代码源时我意识到urlopen可以拍Request参数中的对象而不仅仅是一个字符串，允许放置我想要的标题。
但如果我尝试对 urlretrieve 做同样的事情, 我得到一个 TypeError: expected string or bytes-like object正如其他帖子中提到的那样。

我最终做的是重写我自己的 urlretrieve，删除抛出错误的行(该行与我的用例无关)。

它工作正常但我想知道是否有 更好/更清洁这样做的方式，而不是重写我自己的urlretrieve .如果可以将自定义 header 传递给 urlopen ，感觉应该可以对 urlretrieve 做同样的事情?

Answer 1

我找到了一种方法，您只需添加几行额外的代码...

import urllib.request

opener = urllib.request.build_opener()
opener.addheaders = [('User-agent', 'Mozilla/5.0')]
urllib.request.install_opener(opener)
urllib.request.urlretrieve("type URL here", "path/file_name")

如果您想了解详细信息，可以引用 python 文档: https://docs.python.org/3/library/urllib.request.html