python实现下载指定网址所有图片的方法

#coding=utf-8#download pictures of the url#useage: python downpicture.py www.baidu.comimport osimport sysfrom html.parser import HTMLParserfrom urllib.request import urlopenfrom urllib.parse import urlparsedef getpicname(path):  '''  retrive filename of url    '''  if os.path.splitext(path)[1] == '':    return None  pr=urlparse(path)  path='http://'+pr[1]+pr[2]  return os.path.split(path)[1]def saveimgto(path, urls):  '''  save img of url to local path  '''  if not os.path.isdir(path):    print('path is invalid')    sys.exit()  else:    for url in urls:      of=open(os.path.join(path, getpicname(url)), 'w+b')      q=urlopen(url)      of.write(q.read())      q.close()      of.close()class myhtmlparser(HTMLParser):  '''put all src of img into urls'''  def __init__(self):    HTMLParser.__init__(self)    self.urls=list()    self.num=0  def handle_starttag(self, tag, attr):    if tag.lower() == 'img':      srcs=[u[1] for u in attr if u[0].lower() == 'src']      self.urls.extend(srcs)      self.num = self.num+1if __name__ == '__main__':  url=sys.argv[1]  if not url.startswith('http://'):    url='http://' + sys.argv[1]  parseresult=urlparse(url)  domain='http://' + parseresult[1]  q=urlopen(url)  content=q.read().decode('utf-8', 'ignore')  q.close()  myparser=myhtmlparser()  myparser.feed(content)  for u in myparser.urls:    if (u.startswith('//')):      myparser.urls[myparser.urls.index(u)]= 'http:'+u    elif u.startswith('/'):      myparser.urls[myparser.urls.index(u)]= domain+u  saveimgto(r'D:/python/song', myparser.urls)  print('num of download pictures is {}'.format(myparser.num))