Python PhatomJS 和Selenium动态加载页面 获取图片内容

如果您觉得感兴趣的话，可以添加我的微信公众号：一步一步学Python

随笔中的代码拷贝自

http://www.cnblogs.com/Albert-Lee/p/6275146.html

感谢阿里波特，通过学习自己做了一些注释，作为总结和记录。

from selenium import webdriver
import requests
from bs4 import BeautifulSoup
import os
import time
#定义函数时，尽量写上默认参数，字符=''，数字=0，特殊对象也要定义默认值，这样就可以简单的通过.来得到参数具有的性质等。
class BeautifulPicture():
    def __init__(self):
        # 给请求指定一个请求头来模拟chrome浏览器
        self.headers={'User-Agent': 'Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.1 (KHTML, like Gecko) Chrome/22.0.1207.1 Safari/537.1'}
        self.web_url='https://unsplash.com'
        self.folder_path=r'E:Python Code	est'

    def request(self,url):#返回网页的response
        r=requests.get(url)#获得目标网址的response对象
        return r

    def mkdir(self,path=''):#创建文件夹
        path=path.strip()#删除空白字符
        isExists=os.path.exists(path)#判断路径是否存在
        if not isExists:#如果不存在，则创建文件夹
            print('创建名字叫做',path,'的文件夹')
            os.makedirs(path)#创建目标文件夹
            print('创建成功')
            return True#文件夹不存在返回True
        else:#如果存在，返回
            print(path,'文件夹已经存在了，不再创建了')
            return False

    def scroll_down(self,driver=webdriver.PhantomJS(),times=1):#模拟下拉操作
        for i in range(times):
            print('开始执行第',str(i+1),'次下拉操作')
            driver.execute_script('window.scrollTo(0, document.body.scrollHeight);')#执行js代码操作，页面滚动到指定位置（x，y）
            print('第',str(i+1),'次下拉操作执行完毕')
            print("第", str(i + 1), "次等待网页加载......")
            time.sleep(20)

    def get_files(self,path=''):
        pic_names=os.listdir(path)#获取目录中的内容
        return pic_names

    def save_img(self,url,file_name):
        print('开始请求图片地址，过程比较长。。。')
        img=self.request(url)#img是个response对象
        print('开始保存图片')
        f=open(file_name,'ab')

        # rU或Ua以只读方式打开
        # w 以写方式打开
        # a 以追加模式打开
        # r+ 以读写方式打开
        # w+ 读写
        # a+ 读写
        # rb 二进制读
        # wb 二进制写 ab 二进制追加 rb+ 二进制读写 wb+二进制读写 ab+二进制读写
        #
        f.write(img.content)
        #如果想获取文本，可以通过response.text，如果想获取图片、文件，可以通过response.content。
        # resp.text返回的是Unicode型的数据。
        # resp.content返回的是bytes型也就是二进制的数据。
        print(file_name,'图片保存成功')
        f.close()


    def get_pic(self):
        print('开始网页请求')
        #使用selenium通过PhantomJS来进行网络请求
        driver=webdriver.PhantomJS()
        driver.get(self.web_url)#这里就不再通过urllib进行网络请求了
        self.scroll_down(driver=driver,times=5)#下拉5次，可以自己设定
        print('开始获取所有a标签')
        #获取网页中class为cV68d的所有a标签
        all_a=BeautifulSoup(driver.page_source,'lxml').find_all('a',class_='cV68d')
        #按照css类名搜索tag的功能非常实用，但标识css类名的关键字class在Python中是保留字，实用class做参数会导致语法错误。同BS4.1.1版本开始，通过class_参数搜索。
        print('开始创建文件夹')
        is_new_folder=self.mkdir(self.folder_path)
        print('开始切换文件夹')
        os.chdir(self.folder_path)#改变当前工作路径，.gwd()为获取当前工作目录

        print('a标签的数量是：',len(all_a))
        file_names=self.get_files(self.folder_path)#获取当前工作目录下的文件名？干吗用？

        for a in all_a:
            img_str=a['style']
            print('a标签的style内容是：',img_str)
            first_pos=img_str.index('(')+1
            second_pos=img_str.index(')')
            img_url=img_str[first_pos:second_pos]

            # 注：为了尽快看到下拉加载的效果，截取高度和宽度部分暂时注释掉，因为图片较大，请求时间较长。
            # 获取高度和宽度的字符在字符串中的位置
            # width_pos = img_url.index('&w=')
            # height_pos = img_url.index('&q=')
            # width_height_str = img_url[width_pos : height_pos] #使用切片功能截取高度和宽度参数，后面用来将该参数替换掉
            # print('高度和宽度数据字符串是：', width_height_str)
            # img_url_final = img_url.replace(width_height_str, '')  #把高度和宽度的字符串替换成空字符
            # print('截取后的图片的url是：', img_url_final)

            # 截取url中参数前面、网址后面的字符串为图片名
            name_start_pos=img_url.index('.com/')+5
            name_end_pos=img_url.index('?')
            img_name=img_url[name_start_pos:name_end_pos]+'.jpg'
            img_name=img_name.replace('/','')

            if is_new_folder:#如果文件夹不存在，那么肯定没有重复文件，则直接保存文件。
                self.save_img(img_url,img_name)
            else:
                if img_name not in file_names:#如果文件夹存在，需要判断是否有重复文件
                    self.save_img(img_url, img_name)  # 调用save_img方法来保存图片
                else:
                    print("该图片已经存在：", img_name, "，不再重新下载。")


beauty = BeautifulPicture()  #创建类的实例
beauty.get_pic()  #执行类中的方法