爬虫时保存图片到本地(路径为当前工作目录),并进行图片名称自动生成的操作

from hashlib import md5


def download_image(url): #下载该图片的二进制内容
    print('正在下载', url)
    try:
        response = requests.get(url)
        if response.status_code == 200:
            save_image(response.content)
        return None
    except ConnectionError:
        print('下载图片错误', url)
        return None


def save_image(content): # 把二进制内容放到.jpg文件中,即得到图片
    file_path = '{0}/{1}.{2}'.format(os.getcwd(), md5(content).hexdigest(), 'jpg') # 对内容进行哈希后得到图片路径和名称。os.getcwd()为当前目录
    print(file_path)
    if not os.path.exists(file_path):
        with open(file_path, 'wb') as f:
            f.write(content)
            f.close()

猜你喜欢

转载自blog.csdn.net/xiongzaiabc/article/details/80890048