需求:
博主之前有一段时间突然不想玩微博了,然后正好表弟想玩,就给他用了,手机绑定也换成了他的号码。近期突然又想要玩,就重新开了个号。新号微博空空的,也没有什么关注。于是就产生了两个需求,正好可以借这个机会学习一下自动化测试工具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、点击了关注以后,有两种情况需要处理。
第一种:关注成功
第二种:请输入验证码
这两种情况可以通过定位弹出浮窗的标题头来区别,定位到标题头:
图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掉,不关注城市主页。
修改后的代码见完整代码部分。
以上是关注一个用户的操作,批量关注即遍历之前获取到的用户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'页关注完毕!'
网友评论