堆糖图片爬虫升级版

一路吧

1. import os
   
2. import json
   
3. import requests
   
4. 
   
5. 
   
6. def get_img_content(img_url):
   
7.     web = requests.get(img_url)
   
8.     return web.content
   
9. 
   
10. def get_json(url,num,dir_path,number):
    
11.     web = requests.get(url)
    
12.     html = web.text
    
13.     my_dict = json.loads(html)
    
14.     #链接的最终结果：my_dict['data']['object_list'][0]['photo']['path']
    
15. 
    
16.     for i in my_dict['data']['object_list']:
    
17.         if number <= num:
    
18.             my_link = i['photo']['path']
    
19.             img_path = my_link.split('/')[-1]
    
20.             img_content = get_img_content(my_link)
    
21.             with open(dir_path + img_path,'wb') as f:
    
22.                 f.write(img_content)
    
23.             print('正在下载第%d张' % number)
    
24.             number += 1
    
25.         else:
    
26.             break
    
27.     return number
    
28. 
    
29. def main(keyword,i,number):
    
30.     dir_path = 'D://堆糖/' #后面有个斜杠
    
31. 
    
32.     if not os.path.exists(dir_path):
    
33.         os.mkdir(dir_path)
    
34. 
    
35.     url = 'https://www.duitang.com/napi/blog/list/by_search/?kw=%s&type=feed&start=%d' % (keyword, i)
    
36.     number = get_json(url,num,dir_path,number)
    
37.     return number
    
38. 
    
39. 
    
40. if __name__ == '__main__':
    
41.     keyword = input('请输入需要查找的关键字，如：fate：')
    
42.     num = int(input('请输入需要下载的张数，如：24张：'))
    
43.     temp_n = 0
    
44.     number = 1
    
45.     for i in range(-1,num//24):
    
46.         number = main(keyword,temp_n,number)
    
47.         temp_n += 24

复制代码