下载并保存图片Python2.7

#!/usr/bin/env python
# -*- encoding: utf-8 -*-
import urllib2
import os

def save_img(img_url,file_name,file_path='img'):
#保存图片到磁盘文件夹 file_path中,默认为当前脚本运行目录下的 book\img文件夹
try:
if not os.path.exists(file_path):
print '文件夹',file_path,'不存在,重新建立'
#os.mkdir(file_path)
os.makedirs(file_path)
#获得图片后缀
if 'jpeg' in img_url:
file_suffix = '.jpeg'
elif 'jpg' in img_url:
file_suffix = '.jpg'
elif 'png' in img_url:
file_suffix = '.png'
else:
file_suffix = '.jpeg'
#拼接图片名(包含路径)
filename = '{}{}{}{}'.format(file_path,os.path.sep,file_name,file_suffix)
#下载图片,并保存到文件夹中
response = urllib2.urlopen(img_url)
cat_img = response.read()
with open(filename, 'wb') as f:
f.write(cat_img)
except IOError as e:
print '文件操作失败',e
except Exception as e:
print '错误 :',e
save_img(img_url, file_name)
posted @ 2018-03-01 16:36  失落的黎明  阅读(1852)  评论(0编辑  收藏  举报