欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

Python用webdriver实现微博批量自动关注

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

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

 

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

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


思路:

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

Python用webdriver实现微博批量自动关注

登录微博

在上一篇教程里的第三部分【用webdriver启动浏览器登录微博】有思路和代码的详细说明。此篇不再赘述。

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

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

 
Python用webdriver实现微博批量自动关注
图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 我们是不需要的。再定位到【已关注】按钮,对比两个元素的属性差异。如下两个截图:

 
Python用webdriver实现微博批量自动关注
图2.定位到关注按钮
 
Python用webdriver实现微博批量自动关注
图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 结构。如下图: **

 
Python用webdriver实现微博批量自动关注
图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语法参考。

 
Python用webdriver实现微博批量自动关注
图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是超时时间,可以替换成其他值)。具体使用可以参考上一篇教程,或者 和 这两篇。

所以第二行改为:

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

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

第一种:关注成功

 
Python用webdriver实现微博批量自动关注
图6.关注成功

 

第二种:请输入验证码

 
Python用webdriver实现微博批量自动关注
图7.请输入验证码

 

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

 
Python用webdriver实现微博批量自动关注
图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掉,不关注城市主页。

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

 
Python用webdriver实现微博批量自动关注
图9.城市

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

 

完整代码:

相关代码已上传github:
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'页关注完毕!'

效果演示动图

 
Python用webdriver实现微博批量自动关注
图10.批量自动关注动图展示



作者:王雨城
链接:https://www.jianshu.com/p/37ef59c28a18