Python下载百度贴吧图片程序代码

#!/usr/bin/env python
#-*- coding: UTF-8 -*-
import urllib,re,os
def demo():
    print "-----------------------------------------＼n"
    print u"      Python下载百度贴吧图片＼n"
    print "-----------------------------------------"

def getCode(url): # 获取源码
    page = urllib.urlopen(url)
    code = page.read().decode('utf-8')
    return code

def getImg(code): # 获取图片url
    key = re.compile(r'src="(.*＼.jpg)" pic_ext')  # 匹配图片地址的正则表达式
    img_list = key.findall(code)  # 利用正则表达式找到图片地址
    saveDir = raw_input(unicode('请输入图片保存路径：','utf-8').encode('gbk'))
    m = 1
    print u"正在获取，请稍后..."
    print u"总共%s张图片" % len(img_list)
    for img_url in img_list:
        getDir = saveDir + os.sep + 'mk_%s.jpg' % m
        urllib.urlretrieve(img_url,getDir) #  下载图片
        m += 1
        if m == len(img_list):
            print u"获取完毕"
            print u"图片储存路径为：" + saveDir

if __name__ == '__main__':
    demo()
    url = raw_input(unicode('请输入百度贴吧地址：','utf-8').encode('gbk'))
    code = getCode(url)
    getImg(code)