如何下载 1MB <文件？

Question

我正在尝试从给定路径下载 1MB 的 pdf 文件，但我的代码存在问题。 下载超过 1MB . 我该如何解决？

try:
response = requests.get(href, stream=True)
total = response.headers.get('content-length')
if len(total) > self.MAX_SIZE:
print "maximum size (%d kbs)" % (self.MAX_SIZE/1024)
else:
if total is None:
pass
else:
os.system('wget -P %s %s'%(PATH, href))
except Exception as e:
pass

Answer 1

total = response.headers.get('content-length')是字符串，表示在比较之前应转换为数字的内容长度。

if int(total) > self.MAX_SIZE:

您的代码还有另一个问题是response = requests.get(href, stream=True)对性能不利 - 您在比较之前下载整个文件，如果这是您唯一需要的东西，应该使用response = requests.head(href)来检索Content-Length。