python3 用requests 保存網頁以及BeautifulSoup保存圖片,而且在本地能夠正常顯示文章的內容和圖片

用requests 模塊作了個簡單的爬蟲小程序,將博客的一篇文章以及圖片保存到本地,文章格式存爲'.html'。當文章保存到本地後,圖片的鏈接多是目標站點的絕對或者相對路徑,因此要是想在本地也顯示圖片,須要將保存下來圖片的本地路徑替換到本地的html文件裏。html

保存網頁用的時requests模塊,保存圖片用的時BeautifulSoup, 這兩個都是第三方模塊,須要安裝,使用時須要手動導入。python

**安裝方式:小程序

pip install requstsapp

在python3 可能用 pip install beautifulsoup 會報錯,能夠直接pip install bs4, 這樣時能夠成功安裝的。dom

由於其實beautifulsoup 在bs4安裝包中,使用的時候採用:from bs4 import  beautifulsoupurl

具體的代碼以下:spa

 1 from bs4 import BeautifulSoup
 2 import requests,os
 3 targetDir = os.path.join(os.path.dirname(os.path.abspath(__file__)),'imgs1')#圖片保存的路徑,eg,向前文件夾爲'D:\Coding', 即圖片保存在'D:\Coding\imgs1\'
 4 if not os.path.isdir(targetDir):#不存在建立路徑
 5     os.mkdir(targetDir)
 6 url = 'http://www.cnblogs.com/nancyzhu/p/8146408.html'
 7 domain = 'http://www.cnblogs.com'
 8 #保存頁面到本地
 9 def save_html():
10     r_page = requests.get(url)
11     f = open('page.html','wb')
12     f.write(r_page.content)#save to page.html
13     f.close()
14     return r_page
15 #修改文件,將圖片路徑改成本地的路徑
16 def update_file(old,new):
17     with open('page.html', encoding='utf-8') as f, open('page_bak.html', 'w',
18                                                    encoding='utf-8') as fw:  # 打開兩個文件,原始文件用來讀,另外一個文件將修改的內容寫入
19         for line in f:  # 遍歷每行,取出來的是字符串,所以能夠用replace 方法替換
20             new_line = line.replace(old, new)  # 逐行替換
21             fw.write(new_line)  # 寫入新文件
22     os.remove('page.html')  # 刪除原始文件
23     os.rename('page_bak.html', 'page.html')  # 修改新文件名, old -> new
24 #保存圖片到本地
25 def save_file_to_local():
26     obj = BeautifulSoup(save_html().content, 'lxml')  # 後面是指定使用lxml解析,lxml解析速度比較快,容錯高。
27     imgs = obj.find_all('img')
28     #將頁面上圖片的連接加入list
29     urls = []
30     for img in imgs:
31         if 'data-src' in str(img):
32             urls.append(img['data-src'])
33         else:
34             urls.append(img['src'])
35     #遍歷全部圖片連接,將圖片保存到本地指定文件夾,圖片名字用0,1,2...
36     i = 0
37     for url in urls:#看下文章的圖片有哪些格式,一一處理
38         if url.startswith('//'):
39             new_url = 'http:' + url
40             r = requests.get(new_url)
41         elif url.startswith('/') and url.endswith('gif'):
42             new_url = domain + url
43             r = requests.get(new_url)
44         elif url.endswith('.png') or url.endswith('jpg') or url.endswith('gif'):
45             r = requests.get(url)
46         t = os.path.join(targetDir, str(i) + '.jpg')#指定目錄
47         fw = open(t,'wb')  # 指定絕對路徑
48         fw.write(r.content)#保存圖片到本地指定目錄
49         i += 1
50         update_file(url,t)#將老的連接(有多是相對連接)修改成本地的連接,這樣本地打開整個html就能訪問圖片
51         fw.close()
52 
53 if __name__ == '__main__':
54     save_html()
55     save_file_to_local()
相關文章
相關標籤/搜索