python爬虫避免重复爬取,url管理,附带代码

        简介:在爬虫运行时容易产生反复爬取页面的情况,以下代码为了避免重复爬取生成一个url管理器作为一个模块使用。


class UrlManager():
    def __init__(self):  # 初始化函数
        self.new_urls = set()  # 定义一个新的待爬取的
        self.old_urls = set()  # 定义一个已经待爬取的url

    def add_new_url(self, url):  # 新增一个url
        if url is None or len(url) == 0:  # 判断一个url是否合格
            return
        if url in self.new_urls or url in self.old_urls:  # 判断这个url是否在新或老的容器里面
            return
        self.new_urls.add(url)

    def add_new_urls(self, urls):  # 批量添加url
        if urls is None or len(urls) == 0:  # 判断一个url是否合格
            return

        for url in urls:
            self.add_new_url(url)

    def get_url(self):  # 获取一个待爬取的url
        if self.has_new_url():  # 判断这个容器中有待爬取的rul
            url = self.new_urls.pop()
            self.old_urls.add(url)
            return
        else:
            return None

    def has_new_url(self):  # 判断一个容器里面有没有待爬取的url
        return len(self.new_urls) > 0


if __name__ == "__main__":
    url_manger = UrlManager()

    url_manger.add_new_url("url1")
    url_manger.add_new_urls(["url1", "url2"])
    print(url_manger.new_urls, url_manger.old_urls)

    print("#"*30)
    new_url = url_manger.get_url()
    print(url_manger.new_urls, url_manger.old_urls)

    print("#"*30)
    new_url = url_manger.get_url()
    print(url_manger.new_urls, url_manger.old_urls)

    print("#"*30)
    print(url_manger.has_new_url())
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值