2012-07-19 132 views
6

我需要找到一个HTML代码类似于此的图像:Python的下载图像

... 
<a href="/example/1"> 
    <img id="img" src="http://example.net/example.jpg" alt="Example" /> 
</a> 
... 

我使用LXML和要求。

下面是代码:

import lxml 
from lxml import html 
import requests 

url = 'http://www.example.com' 

r = requests.get(url) 
tree = lxml.html.fromstring(r.content) 

img = tree.get_element_by_id("img") 
f = open("image.jpg",'wb') 
f.write(requests.get(img['src']).content) 

但我得到一个错误:

Traceback (most recent call last): 
    File "/Users/Name/Documents/Python/Example/Script.py", line 13, in <module> 
    s = requests.get(img['src']) 
    File "/Library/Python/2.6/site-packages/lxml/lxml.etree.pyx", line 1052, in lxml.etree._Element.__getitem__ (src/lxml/lxml.etree.c:38272) 
TypeError: 'str' object cannot be interpreted as an index 

建议?

+2

建议:阅读文档,并请固定的HTML。 – dav1d 2012-07-19 18:24:34

回答

4

尝试f.write(requests.get(img.attrib['src']).content)

1
import lxml.html 
import requests 

url = 'http://www.example.com/' 
tree = lxml.html.parse(url) 
img = tree.get_element_by_id('img') 
img_url = img.attrib['src'] 

with open('image.jpg', 'wb') as outf: 
    data = requests.get(img_url).content 
    outf.write(data) 
+0

IMG = tree.get_element_by_id( 'IMG') 它不工作这一次,它说: 回溯(最近通话最后一个): 文件“/Users/Example/Documents/Python/Example/Script.py ”,第6行,在 IMG = tree.get_element_by_id( 'IMG') AttributeError的: 'lxml.etree._ElementTree' 对象没有属性 'get_element_by_id' 我试图替代树= lxml.html.parse(URL )与 tree = lxml.html.fromstring(requests.get(url).content) 现在它的工作原理,感谢您的帮助! – Jiloc 2012-07-20 00:26:57