如何下载文件并将其传递给函数

时间:2015-12-19 15:01:24

标签: python

我有一个需要base64编码文件的函数。 我可以使用

下载并保存此文件
req = urllib2.Request(url, headers={'User-Agent' : "Magic Browser"}) 
f = urllib2.urlopen(req)
with open('file.torrent', "wb") as code:
        code.write(f.read())

但我不想保存它,我想把它传递给一个函数。

1 个答案:

答案 0 :(得分:1)

如果需要传递像object这样的文件,可以将base64编码的字符串写入io.BytesIO对象:

import requests
from io import BytesIO
f = BytesIO()
r = requests.get("http://stackoverflow.com/questions/34372102/how-to-download-a-file-and-pass-it-to-a-function")

f.write(r.content.encode("base_64"))

f.seek(0)

print(f.read())

输出:

PCFET0NUWVBFIGh0bWw+DQo8aHRtbCBpdGVtc2NvcGUgaXRlbXR5cGU9Imh0dHA6Ly9zY2hlbWEu
b3JnL1FBUGFnZSI+DQo8aGVhZD4NCg0KPHRpdGxlPnB5dGhvbiAtIEhvdyB0byBkb3dubG9hZCBh
IGZpbGUgYW5kIHBhc3MgaXQgdG8gYSBmdW5jdGlvbiAtIFN0YWNrIE92ZXJmbG93PC90aXRsZT4N
CiAgICA8bGluayByZWw9InNob3J0Y3V0IGljb24iIGhyZWY9Ii8vY2RuLnNzdGF0aWMubmV0L3N0
YWNrb3ZlcmZsb3cvaW1nL2Zhdmljb24uaWNvP3Y9NGYzMmVjYzhmNDNkIj4NCiAgICA8bGluayBy
ZWw9ImFwcGxlLXRvdWNoLWljb24gaW1hZ2Vfc3JjIiBocmVmPSIvL2Nkbi5zc3RhdGljLm5ldC9z
dGFja292ZXJmbG93L2ltZy9hcHBsZS10b3VjaC1pY29uLnBuZz92PWM3OGJkNDU3NTc1YSI+DQog
ICAgPGxpbmsgcmVsPSJzZWFyY2giIHR5cGU9ImFwcGxpY2F0aW9uL29wZW5zZWFyY2hkZXNjcmlw
dGlvbit4bWwiIHRpdGxlPSJTdGFjayBPdmVyZmxvdyIgaHJlZj0iL29wZW5zZWFyY2gueG1sIj4N
 ..................................................

f直接传递给您的函数。如果您只需传递已编码的字符串,只需使用r.content.encode("base_64")传递f.read().encode("base_64")urllib2

如果需要.torrent扩展名,您可以写信至NamedTemporaryFile并使用f.name提取名称:

from tempfile import NamedTemporaryFile
f = NamedTemporaryFile("wb+",suffix=".torrent")
r = requests.get("http://stackoverflow.com/questions/34372102/how-to-download-a-file-and-pass-it-to-a-function")

print(f.name)
f.write(r.content.encode("base_64"))
f.seek(0)

/tmp/tmpzedDPm.torrent

您可以将文件对象或f.name传递给您的函数,因为该函数只能使用f.name文件。

with open(f.name) as f:
    print(f.read())