python selenium爬取kuk

# -*- coding:utf-8 -*-
# crawl kuku漫画
__author__='fengzhankui'
from selenium import webdriver
from selenium.webdriver.common.desired_capabilities import DesiredCapabilities
import os
import urllib2
import chrom
class getManhua(object):
    def __init__(self):
        self.num=5
        self.starturl='http://comic.kukudm.com/comiclist/2154/51850/1.htm'
        self.browser=self.getBrowser()
        self.getPic(self.browser)
    def getBrowser(self):
        dcap = dict(DesiredCapabilities.PHANTOMJS)
        dcap["phantomjs.page.settings.userAgent"] = ("Mozilla/5.0 (Windows NT 6.1; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/59.0.3071.115 Safari/537.36")
        browser=webdriver.PhantomJS(desired_capabilities=dcap)
        try:
            browser.get(self.starturl)
        except:
            print 'open url fail'
        browser.implicitly_wait(20)
        return browser
    def getPic(self,browser):
        cartoonTitle = browser.title.split('_')[0]
        self.createDir(cartoonTitle)
        os.chdir(cartoonTitle)
        for i in range(1,self.num):
            i=str(i)
            imgurl = browser.find_element_by_tag_name('img').get_attribute('src')
            print imgurl
            with open('page'+i+'.jpg','wb') as fp:
                agent = chrom.pcUserAgent.get('Firefox 4.0.1 - Windows')
                request=urllib2.Request(imgurl)
                request.add_header(agent.split(':',1)[0],agent.split(':',1)[0])
                response=urllib2.urlopen(request)
                fp.write(response.read())
                print 'page'+i+'success'
            NextTag = browser.find_elements_by_tag_name('a')[-1].get_attribute('href')
            browser.get(NextTag)
            browser.implicitly_wait(20)
    def createDir(self,cartoonTitle):
        if os.path.exists(cartoonTitle):
            print 'exists'
        else:
            os.mkdir(cartoonTitle)
if __name__=='__main__':
    getManhua()