python爬虫 -- 截图网页中的图片(先Mark下,待改善)

# -*- coding: utf-8 -*-

from selenium import webdriver
from PIL import Image

if __name__ == '__main__':
    codePrefix = u'https://kyfw.12306.cn/passport/captcha/captcha-image'
    wbe = webdriver.PhantomJS('/usr/bin/phantomjs')
    wbe.get("https://kyfw.12306.cn/otn/login/init")
    # 此处有问题,还待解决,学习爬虫中,,,,,,
    #element = wbe.find_element_by_xpath('//*[@id="loginForm"]/div/ul[1]/li[4]/div/div/div[2]/img')
    element = wbe.find_element_by_xpath('//*[contains(@class_name="touclick-image")]')
    left = element.location['x']
    top = element.location['y']
    right = element.location['x'] + element.size['width']
    bottom = element.location['y'] + element.size['height']
    im = Image.open(r'登录页.png')# 全页面截屏
    im = im.crop((left, top, right, bottom))
    im.save('验证码.png')

r'''
Message: 'phantomjs' executable needs to be in PATH.
	geckodriver 要自行下载
	phantomjs 自行下载
'''

def get_image():
	pass