如何从python子进程中获取输出

时间:2012-12-28 06:45:06

标签: python linux subprocess wget

我正在使用此

从命令行执行python脚本

python myscript.py

这是我的剧本

if item['image_urls']:
            for image_url in item['image_urls']:
            subprocess.call(['wget','-nH', image_url, '-P  images/'])

现在,当我在屏幕上运行时,我看到这样的输出

HTTP request sent, awaiting response... 200 OK
Length: 4159 (4.1K) [image/png]

现在我想要的是终端上应该没有输出。

我想获取输出并从那里找到图片扩展名,即[image/png]抓住png并将文件重新设置为something.png

这可能吗

1 个答案:

答案 0 :(得分:3)

如果您只想使用wget下载内容,为什么不在标准python库中尝试urllib.urlretrieve

import os
import urllib
image_url = "https://www.google.com/images/srpr/logo3w.png"
image_filename = os.path.basename(image_url)
urllib.urlretrieve(image_url, image_filename)

编辑:如果图片是由脚本动态重定向的,您可以尝试requests包来处理重定向。

import requests
r = requests.get(image_url)
# here r.url will return the redirected true image url
image_filename = os.path.basename(r.url)
f = open(image_filename, 'wb')
f.write(r.content)
f.close()

我没有测试代码,因为我找不到合适的测试用例。 requests的一大优势是它还可以处理authorization

EDIT2 :如果图片是由脚本动态提供的,例如gravatar图片,您通常可以在响应标头的content-disposition字段中找到文件名。

import urllib2
url = "http://www.gravatar.com/avatar/92fb4563ddc5ceeaa8b19b60a7a172f4"
req = urllib2.Request(url)
r = urllib2.urlopen(req)
# you can check the returned header and find where the filename is loacated
print r.headers.dict
s = r.headers.getheader('content-disposition')
# just parse the filename
filename = s[s.index('"')+1:s.rindex('"')]
f = open(filename, 'wb')
f.write(r.read())
f.close()

EDIT3 :正如@Alex在评论中建议的那样,您可能需要清理返回标头中的编码filename,我认为只需获取基本名称即可。

import os
# this will remove the dir path in the filename
# so that `../../../etc/passwd` will become `passwd`
filename = os.path.basename(filename)