python练习册 每天一个小程序 第0013题

# -*-coding:utf-8-*—
'''
    题目描述：
         用 Python 写一个爬图片的程序，爬 这个链接里的日本妹子图片 :-)
    地址：
        http://tieba.baidu.com/p/2166231880

    思路：
        用正则表达式匹配图片链接，然后进行下载

'''
'''
import re
import requests

def main():
    url = 'http://tieba.baidu.com/p/2166231880'
    response = requests.get(url)
    html = response.text
    match = re.compile('img .*?src=\"(.*?)\"')
    for i in  match.findall(html):
        if 'imgsrc' in i :
            print i

if __name__ == '__main__':
    main()
'''

import urllib2
import re
from os.path import basename
from urlparse import urlsplit

url = "http://tieba.baidu.com/p/2166231880"
def getPage(url):
    url=url+"?see_lz=1"
    urlContent = urllib2.urlopen(url).read()
    page='<span class="red">(.*?)</span>'
    thePage=re.findall(page,urlContent)
    return int(thePage[0])
def downImg(url):
    urlContent = urllib2.urlopen(url).read()
    spans='<cc>(.*?)</cc>'
    ss=re.findall(spans,urlContent)
    obImgs=','.join(ss)
    imgUrls = re.findall('img .*?src="(.*?)"', obImgs)
    for imgUrl in imgUrls:
        print imgUrl
        '''
        try:
            imgData = urllib2.urlopen(imgUrl).read()
            fileName = basename(urlsplit(imgUrl)[2])
            output = open(fileName,'wb')
            output.write(imgData)
            output.close()
        except:
            print "Er.."
        '''
def downLoad(url):
    numb=getPage(url)
    cont=0
    print "There are "+str(numb)+" pages."
    while cont<numb:
        cont+=1
        print "Downloading "+url+"?see_lz=1&pn="+str(cont)+"..."
        downImg(url+"?see_lz=1&pn="+str(cont))
    print 'Completed!'

downImg(url)