python - 如何从 header 中的 Content-Disposition 获取文件名

Question

我正在下载一个带有 Mechanize 的文件，并且在响应头中有一个字符串:

Content-Disposition: attachment; filename=myfilename.txt

是否有一种快速的标准方法来获取该文件名值？我现在想到的是:

filename = f[1]['Content-Disposition'].split('; ')[1].replace('filename=', '')

但它看起来像一个快速的'n'dirty解决方案。

Answer 1

首先使用mechanize获取header的值，然后使用内置cgi解析header模块。

演示:

>>> import mechanize
>>> browser = mechanize.Browser()
>>> response = browser.open('http://example.com/your/url')
>>> info = response.info()
>>> header = info.getheader('Content-Disposition')
>>> header
'attachment; filename=myfilename.txt'

然后可以解析标题值:

>>> import cgi               
>>> value, params = cgi.parse_header(header)
>>> value
'attachment'
>>> params
{'filename': 'myfilename.txt'}

params 是一个简单的字典，所以 params['filename'] 是你所需要的。文件名是否用引号括起来并不重要。