python爬虫入门 实战(六)---用webdriver实现微

作者: 王雨城 | 来源:发表于2017-06-29 00:02 被阅读0次

    需求:
    博主之前有一段时间突然不想玩微博了,然后正好表弟想玩,就给他用了,手机绑定也换成了他的号码。近期突然又想要玩,就重新开了个号。新号微博空空的,也没有什么关注。于是就产生了两个需求,正好可以借这个机会学习一下自动化测试工具webdriver的基本使用:
    1、将原微博的博文搬到新账号
    2、用新账号关注原微博的所有关注

    说明:
    这一篇是python爬虫入门 实战(五)---用webdriver实现批量自动发微博的姊妹篇,实现的是第二个需求,我们继续来学习webdriver的使用。

    涉及:
    使用webdriver实现:
    1、寻找单个元素(element)
    2、寻找一组具有相同属性的元素
    3、点击按钮
    4、获取元素的属性
    5、获取元素包含的文本

    本篇目录:
    1.思路
    2.登录微博
    3.进入目标用户关注列表,获取未关注用户id。
    4.利用获取到的id进入目标博主主页,进行关注操作
    5.完整代码
    6.效果演示动图
    7.参考


    思路:

    1、登录微博
    2、进入目标用户(博主原微博)关注列表,获取未关注用户id
    3、利用获取到的id进入目标博主主页,进行关注操作

    登录微博

    在上一篇教程python爬虫入门 实战(五)---用webdriver实现微博“搬家”(1)里的第三部分【用webdriver启动浏览器登录微博】有思路和代码的详细说明。此篇不再赘述。

    进入目标用户关注列表,获取未关注用户id。

    1、先手动登录微博,进入到目标用户的关注列表,观察url链接的构造:

    图1.目标用户关注列表
    http://weibo.com/p/1005052622535523/follow?from=page_100505&wvr=6&mod=headfollow#place
    

    1、?后的参数并不重要可以直接删去,重新加上 page 参数可以指定页码。
    2、p/后的那串数字是100505+‘用户 id ’

    所以我们可以登录以后,用一个循环遍历所有的页码数构造 url 链接,用
    driver.get 方法访问,即可访问所有的关注列表。

    2、接下来看如何获取一页列表中所有的未关注用户的 id 。

    定位到【关注】按钮,发现可以找到我们要的用户 id (即图中红框里的 “uid” )。所以,我们只要找到这个元素获取它的 “action-data” 属性,用字符串切割和切片即可得到用户 id 字符串。

    另外,我们只需关注未关注的用户,所以已关注用户的 id 我们是不需要的。再定位到【已关注】按钮,对比两个元素的属性差异。如下两个截图:

    图2.定位到关注按钮 图3.定位到已关注按钮

    我们可以发现它们都有 “action-type” 这个属性,而且不同。所以我们可以用这个属性来区分未关注用户和已关注用户。思路总结成一句话就是,找到本页中所有
    action-type="follow" 的 a 标签,获取 “action-data” 属性并截取其中的 uid 。

    因为符号条件的元素有多个(每页有多个用户的关注按钮),所以用driver.find_elements(注意,比上篇教程中的方法 driver.find_element 多一个 “s”,返回的是 list )来查找一组目标元素。用 element.get_attribute 来获取目标属性。

    获取用户 id 的代码如下:

    # 获取某用户的关注列表中 登录用户未关注的用户
    def get_followlist_unf(self,uid,page):
        url='http://weibo.com/p/100505'+uid+'/follow'+'?page='+str(page)# 关注列表页
        self.driver.get(url)
        a_list=self.driver.find_elements_by_xpath('//a[@action-type="follow"]')# 找到所有的关注按钮上的 a 标签元素
        
        uid_list=[]
        for a in a_list:
            action_data=a.get_attribute('action-data')# 获取 action-data 属性
            uid=action_data.split('&')[-3].split('=')[1]# 获取uid的值
            uid_list.append(uid)# 将uid加入到list中
        return uid_list # 返回本页所有的uid
    

    action_data 属性例子 :

    action-data="refer_sort=followlist&refer_flag=followlist&uid=1785997537&fnick=黄思琦-&f=1"
    

    获取 uid 的那一行,是将 action_data 属性的字符串值从‘&’符号分割,[-3]索引到分割结果的倒数第三个,获得字符串 uid="用户id",再从“=”分割,[1]索引到分割结果的第二个,也就是 uid 的值。

    利用获取到的id进入目标博主主页,进行关注操作

    **1、手动进入一个未关注用户的主页,查看 url 结构。如下图: **

    图4.未关注用户主页
    http://weibo.com/u/2392717877?refer_flag=1005050006_&is_hot=1
    

    1、同样的,?后面的参数没有用,可以直接删除不用。
    2、u/后面就是用户的id
    3、有些用户可能有个性域名,地址与普通用户不一样,但是没关系,用 id 构造的 url (如以上)一样可以访问到用户主页。

    2、找到【关注】按钮,进行点击。

    我们定位到【关注】按钮,对应的元素是一个 div 下的 a 标签,这个 div 具有属性 node-type="focusLink",所以我们可以用 xpath语法定位到这个元素。回顾xpath语法参考XPath 教程

    图5.定位关注按钮

    关注操作的代码如下:

    driver.get('http://weibo.com/u/'+uid)
    focuslink=driver.find_element_by_xpath('//div[@node-type="focusLink"]/a')
    focuslink.click()
    

    像上一篇教程说的一样,需要用 WebDriverWait(driver,10) 方法等待加载(10是超时时间,可以替换成其他值)。具体使用可以参考上一篇教程,或者Python selenium 三种等待方式详解WebDriverWait等设置等待时间和超时时间这两篇。

    所以第二行改为:

    focuslink=WebDriverWait(driver,10).until(lambda x:x.find_element_by_xpath('//div[@node-type="focusLink"]/a'))
    

    3、点击了关注以后,有两种情况需要处理。

    第一种:关注成功

    图6.关注成功

    第二种:请输入验证码

    图7.请输入验证码

    这两种情况可以通过定位弹出浮窗的标题头来区别,定位到标题头:

    图8.定位到标题头

    所以我们只需要用driver.find_element_by_class_name方法来定位到该元素,再用element.get_attribute("innerHTML")来获取元素的包含文本。

    进一步根据这个文本进行判断即可,如果成功,可以什么都不做,或者输出成功提示,如果需要输入验证码,因为对于初学者过于复杂,我们不进行破解,等待两分钟后再次调用关注该用户的方法进行重试。(经过尝试,一般等待2~5分钟再关注即不会弹出验证码,下次碰到验证码,同样进行等待。)

    代码如下:

    def follow(self,uid):
        self.driver.get('http://weibo.com/u/'+uid)
        focuslink=WebDriverWait(driver,10).until(lambda x:x.find_element_by_xpath('//div[@node-type="focusLink"]/a'))
        focuslink.click()
        # 点完关注后,检查页面是否有需要输入验证码,如果有的话,等待2分钟重新关注该用户
        try:
            title=WebDriverWait(driver,10).until(lambda x:x.find_element_by_class_name('W_layer_title'))
            title_txt=title.get_attribute('innerHTML')
            if title_txt==u'关注成功':
                print u'关注用户:'+uid+u' 成功!'
            elif title_txt==u'请输入验证码':
                print u'需要输入验证码,等待120s……'
                time.sleep(120)# 休息两分钟再试一次
                print u'等待结束,重试'
                self.follow(uid)# 再次调用本方法
        except: 
            pass
    

    这里加了一个 try 和 except,是为了应对什么浮窗都不弹出,找不到标题元素的情况,如果找不到即视为已成功关注,什么都不做,pass掉即可。

    由于有些用户的关注列表里有城市的主页,这些城市主页的 uid 比较特殊,拼接成之前的用户主页 url 后会跳转到主页面 home ,为了应对这种情况,我们还需要一个 try 和 except 来防止找不到【关注】按钮的情况,同样的,找不到关注按钮(即异常),我们什么都不做,直接pass掉,不关注城市主页。

    修改后的代码见完整代码部分。

    图9.城市
    以上是关注一个用户的操作,批量关注即遍历之前获取到的用户id的list,循环调用这个方法即可。

    完整代码:

    相关代码已上传github:py_study
    weibo_op_driver模块在项目中的weibo文件夹中。

    1、weibo_op_driver模块中的爬取关注列表的方法:

        # 获取某用户的关注列表中 登录用户未关注的用户
        def get_followlist_unf(self,uid,page):
            url='http://weibo.com/p/100505'+uid+'/follow'+'?page='+str(page)# 关注列表页
            self.driver.get(url)
            a_list=self.driver.find_elements_by_xpath('//a[@action-type="follow"]')# 找到所有的关注按钮上的 a 标签元素
            
            uid_list=[]
            for a in a_list:
                action_data=a.get_attribute('action-data')# 获取 action-data 属性
                uid=action_data.split('&')[-3].split('=')[1]# 获取uid的值
                uid_list.append(uid)# 将uid加入到list中
            return uid_list # 返回本页所有的uid
    

    2、weibo_op_driver模块中的关注与批量关注方法:

    # 关注用户
    def follow(self,uid):
        self.driver.get('http://weibo.com/u/'+uid)
        try:
            focuslink=self.wait.until(lambda x:x.find_element_by_xpath('//div[@node-type="focusLink"]/a'))
            focuslink.click()
            # 点完关注后,检查页面是否有需要输入验证码,如果有的话,等待2分钟重新关注该用户
            try:
                title=self.wait.until(lambda x:x.find_element_by_class_name('W_layer_title'))
                title_txt=title.get_attribute('innerHTML')
                if title_txt==u'关注成功':
                    print u'关注用户:'+uid+u' 成功!'
                elif title_txt==u'请输入验证码':
                    print u'需要输入验证码,等待120s……'
                    time.sleep(120)# 休息两分钟再试一次
                    print u'等待结束,重试'
                    self.follow(uid)
            except:
                pass
        except:
            pass
    
    # 批量关注(参数可能包含自身id)
    def follow_uidlist(self,uid_list):# 参数为uid的一个list
        for uid in uid_list:
            if uid==self.uid_login:# 如果是自己的id,跳过
                continue
            
            # 是未关注用户,则关注
            self.follow(uid)
    
    

    3、主方法代码:

    operator=WBoperator()
    operator.login('account', 'password')# 填入你的微博账号密码
    for i in range(41):# 博主原微博有41页的关注列表
        page=i+1# 页码
         
        uid_list_unf=operator.get_followlist_unf('2622535523', page)
        operator.follow_uidlist(uid_list_unf)
        print u'第'+str(page)+u'页关注完毕!'
    

    效果演示动图

    图10.批量自动关注动图展示

    参考

    XPath 教程
    Python selenium 三种等待方式详解
    WebDriverWait等设置等待时间和超时时间

    相关文章

      网友评论

        本文标题:python爬虫入门 实战(六)---用webdriver实现微

        本文链接:https://www.haomeiwen.com/subject/lczucxtx.html