【无标题】

读取CSV文件,获取文件中的url地址,并对url地址进行解析,将图片以原名称下载保存到本地文件夹

import csv
import os
import requests
import tqdm

# 该脚本主要实现,对算法平台的调用记录生成的csv文件进行读取,获取文件中的url地址,并对url地址进行解析,将图片以原名称下载保存到本地文件夹中

def download_url(source,save_path):
    with open(source) as f:
        f_csv = csv.reader(f)
        headers = next(f_csv)
        for row in f_csv:
            dict1 = row[-2]
            # 排除返回值为空的情况
            if len(dict1)==0:
                continue

            # 为str类型,先转为dict类型
            dict1 = eval(dict1)
            # 读取key对应的value值
            url_path = dict1['url']
            print(url_path)
            response = requests.get(url_path)
            file_name = url_path.split('/')[-1]
            # 如果图片名称中带":",命名方式要加以修改
            if ":" in file_name:
                file_name0 = url_path.split('/')[-1].split(':')[-3]
                file_name1 = url_path.split('/')[-1].split(':')[-2]
                file_name2 = url_path.split('/')[-1].split(':')[-1]
                file_name = file_name0 + '-' + file_name1 + '-' + file_name2

            file_path = save_path + '/' + file_name

            with open(file_path, 'wb') as f:  # �Զ�����д���ļ�����
                f.write(response.content)

        print("图片保存完成")

首次更新,希望能对你有所帮助!

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值