在B站爬取坤坤的视频,并获取视频基本信息,我们用seleniu


在B站有许多坤坤的视频,作为一名ikun,让我们写个爬虫研究一下视频的视频的名字、链接、观看次数、弹幕、发布时间以及作者。我们用selenium来实现这个爬虫,由于要获取的数据比较多,我们写几个函数来实现这个爬虫。

先倒入需要用到的库,包括selenium, time ,BeautifulSoup ,ChromeDriverManager。

打开及搜索函数:在这段代码中,我们导入了 `webdriver` 和 `ChromeDriverManager` 模块,以便使用 ChromeDriver 控制 Chrome 浏览器,以及自动下载和安装最新版本的 ChromeDriver。同时,我们还导入了 `time` 模块,以便在代码中添加延迟,以便页面加载完成。最后,我们还导入了 `BeautifulSoup` 模块,以便从网页中提取信息。

    from selenium import webdriver
    from webdriver_manager.chrome import ChromeDriverManager
    import time
    from bs4 import BeautifulSoup

先创建一个 ChromeDriver 实例,打开B站,在这里我们使用ChromeDriverManager().install()方法,他可以自动下载对应版本的chrome,防止因为Chrome版本不正确而报错。(PS:直接用webdriver.Chrome()会使用已下载Chrome游览器,他的版本可能与webdriver需要的版本不匹配这样打开chrome的时候会出现闪退报错)

    url="哔哩哔哩 (゜-゜)つロ 干杯~-bilibili"
    browser = webdriver.Chrome(ChromeDriverManager().install()) #创建一个 ChromeDriver 实例
    browser.set_window_size(1400, 900) #设定页面大小
    browser.get(url) #访问B站

我们可以用xpath方法来获取目标操作的位置,chrome浏览器能够自动提取xpath链接。我们使用 `browser.window_handles` 方法获取当前窗口中的所有句柄,并切换到第二个窗口,以便获取搜索结果页面。

在打开B站首页的时候,会出现一个提示登陆的页面干扰我们,因此我们最好点击一下首页,刷新一下。

    first_page = browser.find_element_by_xpath("//*[@id='i_cecream']/div[2]/div[1]/div[1]/ul[1]/li[1]/a/span") #取得首页按钮的的位置
    first_page.click() #点击首页进行刷新 

然后,我们模拟用户在网站上搜索蔡徐坤篮球相关的视频;点击搜索栏,输入"蔡徐坤 篮球",并进行搜索;最后定位到新页面。

    input = browser.find_element_by_xpath("//*[@id='nav-searchform']/div[1]/input") #取得搜索栏的位置
    input.send_keys('蔡徐坤 篮球') #输入内容
    submit = browser.find_element_by_xpath("//*[@id='nav-searchform']/div[2]") #取得搜索按钮的位置  
    submit.click() #点击搜索
    all_h = browser.window_handles #取得新页面的代码
    browser.switch_to.window(all_h[1]) #取得新页面的代码

打开及搜索函数的完整代码:

def search(url="哔哩哔哩 (゜-゜)つロ 干杯~-bilibili"):
    try:
        browser = webdriver.Chrome(ChromeDriverManager().install())
        browser.set_window_size(1400, 900)
        browser.get(url)
        first_page = browser.find_element_by_xpath("//*[@id='i_cecream']/div[2]/div[1]/div[1]/ul[1]/li[1]/a/span")
        first_page.click()
        input = browser.find_element_by_xpath("//*[@id='nav-searchform']/div[1]/input")
        input.send_keys('蔡徐坤 篮球')
        submit = browser.find_element_by_xpath("//*[@id='nav-searchform']/div[2]")
        submit.click()
        all_h = browser.window_handles
        browser.switch_to.window(all_h[1])
        return browser
    except TimeoutError:
        return search(url)

翻页函数:该函数使用 `find_element_by_xpath` 方法查找“下一页”按钮,并使用 `click` 方法单击该按钮。在单击“下一页”按钮之后,该函数使用 `browser.window_handles` 方法获取所有窗口句柄,并使用`browser.switch_to.window` 方法将焦点切换到搜索结果页面,以便我们可以继续提取视频信息,最后返回页面代码信息。如果没有找到“下一页”按钮,该函数会递归调用自身来查找“下一页”按钮,直到找到为止。

这里有一点需要注意,在取得下一页按钮的时候,最好利用“下一页”的文本进行匹配,如果根据浏览器提供的位置信息匹配,在某些页面下会出现位置不匹配而导致程序异常。

`next = browser.find_element_by_xpath("//button[text()='下一页']")

这段代码较为简单,直接给出完整代码。

    def next_page(browser): #获取下一页
      try:
          next = browser.find_element_by_xpath("//button[text()='下一页']") #下一页按钮
          next.click() #点击按钮
          time.sleep(1) #听一秒等待刷新
          all_h1 = browser.window_handles #取得新页面的代码
          browser.switch_to.window(all_h1[1]) #取得新页面的代码
          return browser 
      except:
          next_page(browser)

在这个函数中,我们首先使用 `browser.page_source` 方法获取当前页面的 HTML 代码,然后使用 `BeautifulSoup` 类解析 HTML 代码,以提取视频信息。接着,我们使用 `xlwt` 库将视频信息写入到 Excel 文件中。

我们还需要一个函数来提取页面信息,

先用BeautifulSoup解析页面,用find方法找到\<class\_='video-list'>标签,这里包含我们需要的所有视频,然后用find\_all方法找到每个视频的标签\<class\_='bili-video-card'>,同时find\_all会帮助我们形成一个列表存储数据,也就是我用到的videos:

    html = browser.page_source
    soup = BeautifulSoup(html, "lxml")
    videos = soup.find(class_='video-list').find_all(class_='bili-video-card')

遍历视频信息列表,提取信息(视频的名字、链接、观看次数、弹幕、发布时间以及作者),这里建议多尝试,多打印几次确保信息准确:

            for item in videos:
                item_readers = item.find("span", class_="bili-video-card__stats--item").span.string
                item_danmu = item.find_all("span", class_="bili-video-card__stats--item")[1].span.string
                item_time = item.find("span", class_="bili-video-card__stats__duration").string
                item_title = item.find("h3").get("title")
                item_author = item.find("span", class_="bili-video-card__info--author").string
                item_link = item.find("a").get("href")[2:]
                print(item_title)

最后一步是储存信息:

sheet.write(n, 0, n)

sheet.write(n, 1, item_title)

sheet.write(n, 2, item_link)

sheet.write(n, 3, item_readers)

sheet.write(n, 4, item_danmu)

sheet.write(n, 5, item_time)

sheet.write(n, 6, item_author)

获取信息和储存函数的完整代码:

    def save_to_excel(browser,sheet,n,wb):
        try:
            html = browser.page_source
            soup = BeautifulSoup(html, "lxml")
            videos = soup.find(class_='video-list').find_all(class_='bili-video-card')

            for item in videos:
                item_readers = item.find("span", class_="bili-video-card__stats--item").span.string
                item_danmu = item.find_all("span", class_="bili-video-card__stats--item")[1].span.string
                item_time = item.find("span", class_="bili-video-card__stats__duration").string
                item_title = item.find("h3").get("title")
                item_author = item.find("span", class_="bili-video-card__info--author").string
                item_link = item.find("a").get("href")[2:]
                print(item_title)

                sheet.write(n, 0, n)
                sheet.write(n, 1, item_title)
                sheet.write(n, 2, item_link)
                sheet.write(n, 3, item_readers)
                sheet.write(n, 4, item_danmu)
                sheet.write(n, 5, item_time)
                sheet.write(n, 6, item_author)
                n += 1
            wb.save("cxk1.xls")
            print(n)
            return n
        except TimeoutError:
            save_to_excel(browser,sheet,n,wb) #存储函数

最后写一个主程序调用三个函数,在主程序中,我们要记得先设置一下存储数据用到的EXCEL;

import xlwt

#设置excel
wb = xlwt.Workbook(encoding="utf-8",style_compression=0)
sheet = wb.add_sheet("b站爬去",cell_overwrite_ok=True)
sheet.write(0,0,"序号")
sheet.write(0,1,"名称")
sheet.write(0,2,"链接")
sheet.write(0,3,"观看次数")
sheet.write(0,4,"弹幕")
sheet.write(0,5,"时间")
sheet.write(0,6,"作者")

主程序完整代码:

在调用主程序时,当我们进入收索信息的第一页时,可以先爬取第一信息,然后利用一个for循环取得其他页面信息。

#设置excel
wb = xlwt.Workbook(encoding="utf-8",style_compression=0)
sheet = wb.add_sheet("b站爬去",cell_overwrite_ok=True)
sheet.write(0,0,"序号")
sheet.write(0,1,"名称")
sheet.write(0,2,"链接")
sheet.write(0,3,"观看次数")
sheet.write(0,4,"弹幕")
sheet.write(0,5,"时间")
sheet.write(0,6,"作者")

n = 1
browser = search("哔哩哔哩 (゜-゜)つロ 干杯~-bilibili") #进入B站,并完成搜索
all_h = browser.window_handles
browser.switch_to.window(all_h[1]) #获取页面代码

total = browser.find_element_by_xpath("//*[@id='i_cecream']/div/div[2]/div[2]/div/div/div/div[2]/div/div/button[9]").text #获取总页数,给for循环做准备
# n = save_to_excel(browser,sheet,n,wb) #调用函数存储页面数据
# browser = next_page(browser) #翻页
for i in range(1,int(total)+1):

    time.sleep(3) #等待加载
    n = save_to_excel(browser,sheet,n,wb)  # 调用函数存储页面数据
    browser = next_page(browser)  # 翻页

完整代码地址:github.com/zhangaynami…

评论关闭