lb20309 发表于 2018-8-6 11:54:53

使用Python通过xlrd处理excel实战

"""  
从excel获取url,下载指定图片保存到指定目录
  
"""
  

  
#xlrd模块需要手工安装,此为处理excel模块
  
import xlrd
  
import os
  
import urllib.request
  

  
def dir(dir_name="images"):
  
    #设定图片保存目录,基于当前程序运行目录
  
    if os.path.isdir(dir_name):
  
      pass
  
    else:
  
      os.mkdir(dir_name)
  

  
def saveImage(img, url):
  
    #保存图片方法,需要两个参数,一个为图片名,一个为下载的url
  
    respones = urllib.request.urlopen(url)
  
    image = respones.read()
  
    with open(img, "wb") as f:
  
      f.write(image)
  

  

  
if __name__ == "__main__":
  
    data = xlrd.open_workbook(r"D:\images.xls")   #打开指定文件
  
    table = data.sheet_by_name(u"Sheet 1")          #选择工作薄
  
    nrows = table.nrows                           #获取行数
  

  
    for i in range(1410,1425):
  
      """
  
      可通过if判断只下载特定行图片,或者修改以上括号数字为从n到n
  
      """
  
      rows = table.row_values(i)
  
      dir(dir_name = rows)               #创建各供应商目录
  
      imagesclass = rows + "\\" + rows#获取证件类型
  
      dir(dir_name = imagesclass)             #创建供应商目录下子目录,各证件类型
  
      url = r"http://" + rows
  
      imagename = rows + "\\" + rows + "\\" + url.split("/")[-1]    #指定图片保存的位置与名称
  
      saveImage(imagename, url)
页: [1]
查看完整版本: 使用Python通过xlrd处理excel实战