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

需求:
博主之前有一段时间突然不想玩微博了,然后正好表弟想玩,就给他用了,手机绑定也换成了他的号码。近期突然又想要玩,就重新开了个号。新号微博空空的,也没有什么关注。于是就产生了两个需求,正好可以借这个机会学习一下自动化测试工具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实现微博“搬家”(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.批量自动关注动图展示


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

猜你喜欢

转载自blog.csdn.net/qq_40925239/article/details/81222601