python 爬虫下载网站图片

#coding:utf-8
import os
import urllib
import re    #正则表达式模块

def getHtml(url):
page = urllib.urlopen(url)    #打开一个URL地址
html = page.read()         #读取URL上的数据
return html

def getImgs(html):
reg = r'src="(.+?\.jpg)" pic_ext'  #写一个正则表达式
imgre = re.compile(reg)        #编译正则表达式
imglist = re.findall(imgre,html)  #读取html中所有包含正则表达式的数据,返回一个list
return imglist

def download(urllist):
x = 1
filepath = "/home/swz/imgdownload/"
#if os.path.exists(filepath) is True:
#filepath = filepath + "2"
os.mkdir(filepath)

for imgurl in urllist:
#print filepath+'%s.jpg' % x
urllib.urlretrieve(imgurl,filepath+'%s.jpg' % x) #将远程数据下载到本地,并对图片进行重命名
x = x + 1
return

if __name__ == "__main__":
html = getHtml("http://tieba.baidu.com/p/2460150866")
urllist = getImgs(html)
#print urllist
download(urllist)