博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
python批量下载图片的三种方法
阅读量:6614 次
发布时间:2019-06-24

本文共 6442 字,大约阅读时间需要 21 分钟。

  一是用微软提供的扩展库win32com来操作IE:

win32com可以获得类似js里面的document对象,但貌似是只读的(文档都没找到)。

  二是用selenium的webdriver:

selenium则提供了Chrome,IE,FireFox等的支持,每种浏览器都有execute_script和find_element_by_xx方法,可以方便的执行js脚本(包括修改元素)和读取html里面的元素。不足是selenium只提供对python2.6和2.7的支持。

  三是用python自带的HTMLParser解析:

HTMLParser则是需要自己写个类继承基类,重写解析元素的方法。

1.win32com

1 #将滚动条滑到底,最多滑动20000像素 2 #模拟键盘右键,查看多张图片 3 import sys 4 import win32com.client,win32api 5 import urllib.request 6 import time 7 import os 8 def main(): 9     #获取参数10     url=sys.argv[1]11     #操作IE12     ie=win32com.client.Dispatch("InternetExplorer.Application")13     ie.Navigate(url)14     ie.Visible=True15     last_url=''16     dir_name=''17     while last_url!=url:18         print('\nThe URL is:',url,'\n')19         while ie.ReadyState != 4:    20             time.sleep(1)21         while ie.Document.readyState != "complete": 22             time.sleep(1)23         #滑动滚动条24         win=ie.Document.parentWindow25         lastY=-1;26         for i in range(40):27             win.scrollTo(0,500*i)28             nowY=win.pageYOffset29             if(nowY==lastY):30                 break31             lastY=nowY32             time.sleep(0.4)33         print('Document load state:',ie.Document.readyState)34         doc=ie.Document35         #第一次需要创建目录36         if(dir_name==''):37             root_dir='E:\\img'38             dir_name=root_dir+'\\'+doc.title39             dir_name=dir_name.replace('|','-')40             if(os.path.exists(root_dir)!=True):41                 os.mkdir(root_dir)42             if(os.path.exists(dir_name)!=True):43                 os.mkdir(dir_name)44         all_image=doc.images45         print('共有',all_image.length,'张图片')46         count=0;47         for img in all_image:48             if(img.id=='b_img'):49                 count=count+150                 print(count,img.src)51                 time.sleep(1)52                 img_file=urllib.request.urlopen(img.src)53                 byte=img_file.read()54                 print(count,'donwload complete!','-'*10,'size:','{:.3}'.format(byte.__len__()/1024),'KB')55                 if(byte.__len__()>7000):56                     file_name=img.src.replace('/','_')57                     file_name=file_name.replace(':','_')58                     end=file_name.__len__()59                     if(file_name.rfind('!')!=-1):60                         end=file_name.rfind('!')61                     if(file_name.rfind('?')!=-1):62                         end=file_name.rfind('?')63                     file_name=file_name[:end]64                     write_file=open(dir_name+'\\'+file_name,'wb')65                     write_file.write(byte)66                     write_file.close()67                     print(count,file_name,'complete!')68         #下一张69         last_url=url70         win32api.keybd_event(39,0)71         time.sleep(1)72         url=ie.Document.url73         print(last_url,url)74     #ie.Quit()75 if __name__ == '__main__':76     main()

2.selenium

1 # -*- coding: cp936 -*- 2 import sys 3 import urllib 4 import time 5 import os 6 from selenium import webdriver 7 def main(): 8     #获取参数 9     url=sys.argv[1]10     #操作IE11     driver=webdriver.Chrome()12     driver.get(url)13     driver.execute_script("window.scrollTo(0, document.body.scrollHeight);")14     #创建目录15     dir_name=driver.find_element_by_tag_name('title').text16     print dir_name17     root_dir='E:\\img'18     dir_name=root_dir+'\\'+dir_name19     dir_name=dir_name.replace('|','-')20     if(os.path.exists(root_dir)!=True):21         os.mkdir(root_dir)22     if(os.path.exists(dir_name)!=True):23         os.mkdir(dir_name)24     images=driver.find_elements_by_tag_name('img')25     count=026     for image in images:27         count=count+128         image_url=str(image.get_attribute('src'))29         img_file=urllib.urlopen(image_url)30         byte=img_file.read()31         print count,'donwload complete!','-'*10,'size:',byte.__len__()/1024,'KB'32         if(byte.__len__()>7000):33             file_name=image_url.replace('/','_')34             file_name=file_name.replace(':','_')35             end=file_name.__len__()36             if(file_name.rfind('!')!=-1):37                 end=file_name.rfind('!')38             if(file_name.rfind('?')!=-1):39                 end=file_name.rfind('?')40             file_name=file_name[:end]41             write_file=open(dir_name+'\\'+file_name,'wb')42             write_file.write(byte)43             write_file.close()44             print count,file_name,'complete!'45     driver.quit()46 if __name__ == '__main__':47     main()

3.HTMLParser:

1 # import modules used here -- sys is a very standard one 2 import sys 3 import urllib.request 4 # Gather our code in a main() function 5 from html.parser import HTMLParser 6 class MyHTMLParser(HTMLParser): 7     def handle_starttag(self,tag,attrs): 8         if(tag=='img'): 9             for attr in attrs:10                 if(attr[0]=='src'):11                     img_file=urllib.request.urlopen(attr[1])12                     byte=img_file.read()13                     #文件大于1000b则生成文件,添加计数,下载多少图片,显示html代码14                     if(byte.__len__()>1000):15                         file_name=attr[1].replace('/','_')16                         file_name=file_name.replace(':','_')17                         end=file_name.__len__()18                         if(file_name.rfind('!')!=-1):19                             end=file_name.rfind('!')20                         if(file_name.rfind('?')!=-1):21                             end=file_name.rfind('?')22                         file_name=file_name[:end]23 ##                        print(file_name)24                         write_file=open('E:\\img\\'+file_name,'wb')25                         write_file.write(byte)26                         write_file.close()27 def main():28     #获取参数29     url=sys.argv[1]30     print('\nThe URL is:',url,'\n')31     #读取url所指向的资源32     html_file=urllib.request.urlopen(url)33     byte_content=html_file.read()34     #将html网页保存起来35     url_file=open('E:\\img\\html\\result.htm','wb')36     url_file.write(byte_content)37     url_file.close()38     #从字节转换为字符串39     s=str(byte_content, encoding = "utf-8")40     #print(s)41     #bytes.decode(html_file.read())42     parser=MyHTMLParser(strict=False)43     parser.feed(s)44 # Standard boilerplate to call the main() function to begin45 # the program.46 if __name__ == '__main__':47     main()

 

注:转载需注明出处及作者。

      

你可能感兴趣的文章
[logstash-input-file]插件使用详解
查看>>
植物大战僵尸
查看>>
原创文章
查看>>
理解JavaScript私有作用域
查看>>
BZOJ 1012: [JSOI2008]最大数maxnumber【线段树单点更新求最值,单调队列,多解】
查看>>
Drupal 7模板(主题钩子)的建议
查看>>
nginx配置文件中location说明
查看>>
连载-第1章绪论 1.1嵌入式系统概述
查看>>
UltraVNC
查看>>
详解synchronized
查看>>
Spring Cloud第二篇 创建一个Eureka Server
查看>>
初探数据双向绑定
查看>>
Webpack4 不深不浅的实践教程
查看>>
nginx1.9+做TCP代理(端口转发)
查看>>
HTML元素的默认CSS设置介绍
查看>>
CSS-图片不变形设置
查看>>
Git异常:fatal: could not create work tree dir 'XXX': No such file or directory
查看>>
GNU make manual 翻译(八十二)
查看>>
python批量下载图片的三种方法
查看>>
/bin/bash^M: bad interpreter: 没有那个文件或目录
查看>>