download_picture

download_pic.py

#-*-coding:utf-8-*-
from BeautifulSoup import BeautifulSoup
import requests
from PIL import Image
from StringIO import StringIO

r = requests.get('http://www.xiaomi.com')
assert(r.status_code == 200)
soup = BeautifulSoup(r.text)
urls = soup.findAll('img')
for url in urls:
    src = None
    if url.has_key('src'):
        src = url['src']
    if url.has_key('src2'):
        src = url['src2']
    r = requests.get(src)
    try:
        image_name = src.split('/')[-1].strip()
        fd = open(image_name,'w')
        fd.write(r.content)
    except:
        print src