上一篇文章通过selenium工具自动搜索爬取哔哩哔哩上面的视频相关信息,今天我们接着上一篇文章,保存视频的图片到本地。
首先找到要爬取的网页数据所在的位置,如下图
并且,右键点击该网址,可以选择在新标签页中打开,在新标签页中,除了显示图片不会显示其他东西。如下
到这就是原本的图片数据了,如果跳转过后依然不是这样,就需要继续往下寻找到最终的图片数据地址。
在上一篇文章的基础上就添加了一个保存图片的函数,上代码。
def img_save(soup):
img_url_list = soup.find(class_='video-list clearfix').find_all_next(class_='img-anchor')
index = 0
for url in img_url_list:
img_url = url.find('img').get('src')
print(img_url)
//i2.hdslb.com/bfs/archive/176fdf5ba89c5eff3c0f300aaf53967dcddff923.jpg@320w_200h.webp
获取图片链接地址,得到如上文本
if img_url != '':
img_request = ('https:'+ img_url).replace('webp', 'jpg')
print(img_request)
img_resp = requests.get(img_request)
https://i2.hdslb.com/bfs/archive/176fdf5ba89c5eff3c0f300aaf53967dcddff923.jpg@320w_200h.jpg
将得到的网址进行拼接,并替换最后的‘webp’为‘jpg’,如果不修改直接保存到本地,图片会无法正常显示
if not os.path.exists("cxk_video_img"):
os.mkdir('cxk_video_img')
with open('cxk_video_img/%d.jpg' %index, 'wb') as f:
f.write(img_resp.content)
index += 1
time.sleep(1)
最后就是新建文件夹,保存图片到文件夹中。
但是有一个问题,就是有的图片网址获取到的为空,不知道是否和网速也有关系,还是代码本身部分还有我没有发现的欠缺,欢迎各位大佬指正。
完整代码如下:
from selenium import webdriver
from selenium.common.exceptions import TimeoutException
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.keys import Keys
from bs4 import BeautifulSoup
import xlwt
import time
import re
import requests
import os
browser = webdriver.Chrome()
#browser = webdriver.PhantomJS()
WAIT = WebDriverWait(browser, 20)
browser.set_window_size(1400, 900)
book = xlwt.Workbook(encoding='utf-8', style_compression=0)
sheet = book.add_sheet('蔡徐坤篮球', 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, '发布时间')
n = 1
def search():
try:
print('开始访问b站....')
browser.get("https://www.bilibili.com/")
search = browser.find_element_by_xpath('//div[@class="nav-search"]/form/input')
search.send_keys("蔡徐坤 篮球")
search.send_keys(Keys.ENTER)
# 跳转到新的窗口
print('跳转到新窗口')
all_h = browser.window_handles
browser.switch_to.window(all_h[1])
# html = browser.page_source
# # print(html)
# soup = BeautifulSoup(html, 'html.parser')
# save_to_excel(soup)
# total_index = soup.find(class_='page-item last').find(class_='pagination-btn')
get_source()
total_index = WAIT.until(EC.presence_of_element_located((By.CSS_SELECTOR,
"li.page-item.last > button")))
# pattern = re.compile('.*?.*?(\d+).*? ', re.S)
# total_index = int(re.findall(pattern, html)[0])
return int(total_index.text)
except TimeoutException:
return search()
def next_page(page_num):
try:
print('获取第(%d)页数据' % page_num)
next_btn = WAIT.until(EC.element_to_be_clickable((By.CSS_SELECTOR,
'li.page-item.next > button')))
next_btn.click()
get_source()
except TimeoutException:
browser.refresh()
return next_page(page_num)
def save_to_excel(soup):
list = soup.find(class_='video-list clearfix').find_all_next(class_='info')
for item in list:
item_title = item.find('a').get('title')
item_link = item.find('a').get('href')
item_dec = item.find(class_='des hide').text
item_view = item.find(class_='so-icon watch-num').text
item_biubiu = item.find(class_='so-icon hide').text
item_date = item.find(class_='so-icon time').text
print('爬取:' + item_title)
global n
sheet.write(n, 0, item_title)
sheet.write(n, 1, item_link)
sheet.write(n, 2, item_dec)
sheet.write(n, 3, item_view)
sheet.write(n, 4, item_biubiu)
sheet.write(n, 5, item_date)
n = n + 1
def img_save(soup):
img_url_list = soup.find(class_='video-list clearfix').find_all_next(class_='img-anchor')
index = 0
for url in img_url_list:
img_url = url.find('img').get('src')
print(img_url)
if img_url != '':
img_request = ('https:'+ img_url).replace('webp', 'jpg')
print(img_request)
img_resp = requests.get(img_request)
if not os.path.exists("cxk_video_img"):
os.mkdir('cxk_video_img')
with open('cxk_video_img/%d.jpg' %index, 'wb') as f:
f.write(img_resp.content)
index += 1
time.sleep(1)
def get_source():
WAIT.until(EC.presence_of_element_located(
(By.CSS_SELECTOR, 'ul.video-list.clearfix')))
# browser.refresh()
html = browser.page_source
# print(html)
soup = BeautifulSoup(html, 'html.parser')
save_to_excel(soup)
img_save(soup)
def main():
try:
total = search()
# print(total)
for i in range(2, int(total)+1):
next_page(i)
finally:
browser.close()
browser.quit()
if __name__ == '__main__':
main()
book.save(u'蔡徐坤篮球.xls')
本来还想爬取保存视频到本地,结果没法得到视频的最终地址,所以只能暂且搁置,保存视频到本地,也可以采用保存图片相类似的方式。