做爬虫项目时,我们需要考虑一个爬虫在爬取时会遇到各种情况(网站验证,ip封禁),导致爬虫程序中断,这时我们已经爬取过一些数据,再次爬取时这些数据就可以忽略,所以我们需要在爬虫项目中设置一个中断重连的功能,使其在重新运行时从之前断掉的位置重新爬取数据。
实现该功能有很多种做法,我自己就有好几种思路,但是真要自己写出来就要费很大的功夫,下面我就把自己好不容易拼凑出来的代码展示出来吧。
首先是来介绍代码的思路:
将要爬取的网站连接存在一个数组new_urls中,爬取一个网址就将它移入另一个数组old_urls中,爬取网站时,就看它是在哪一个数组中,然后再决定要不要爬取。
下面展示代码(从别处抄的):
class UrlManager(object):
def __init__(self): #定义两个数组
self.new_urls=set()
self.old_urls=set()
def add_new_url(self, url): #将一个url加入到new_urls数组中
if url is None:
return
if url not in self.new_urls and url not in self.old_urls:
self.new_urls.add(url)
def add_new_urls(self, urls): #将多个url加入到new_urls数组中
if urls is None or len(urls)==0:
return
for url in urls :
self.add_new_url(url)