Python下载图片

# -*- encoding:UTF-8 -*-
import re,urllib2,time,urllib
def getPage(url):
    '''下载文件html代码，找出一楼的核心代码'''
    opener = urllib2.build_opener()
    #不加头信息则出现403错误和乱码
    opener.addheaders = [('User-agent', 'Mozilla/5.0')];
    time.sleep(10)
    htmlAll = opener.open( url ).read()
    #文件保存编码和文件编辑编码都是utf-8，所以decode一次，不然会出现乱码，但是不影响结果。
    return htmlAll.decode('utf-8')

def getImg(url):
    '''从核心代码中照图图片地址，并且下载保存、命名'''
    regImg = 'src="(.*?)"'
    dir = 'F:\\downtemp'
    pageHtml = getPage(url)
    #找到所有图片地址
    imglist = re.findall(regImg,pageHtml)
    for index in xrange(1,len(imglist)+1):
        #得到文件名
        strFileNamelist = imglist[index-1].split('/')
        strFileNameListLen = len(strFileNamelist)
        fileName = strFileNamelist[strFileNameListLen-1]
        #得到文件名后缀
        suffixNameList = fileName.split('.')
        suffixNameLen = len(suffixNameList)
        suffixName = suffixNameList[suffixNameLen-1]
        #得到保存文件路径
        saveFileName = dir + '\\' + str(time.time()) + '_' + fileName
        #只保存jpeg和jpg文件
        if suffixName == 'jpeg' or suffixName == 'jpg' :
            urllib.urlretrieve(url+imglist[index-1], saveFileName)
            print saveFileName + ' ok!'

if __name__ == '__main__':
    getImg('http://www.huawei.com')