本文共 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.win32com1 #将滚动条滑到底,最多滑动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()
注:转载需注明出处及作者。